glenux-archive/spf13--cobra
1
0
Fork 0
mirror of https://github.com/spf13/cobra synced 2025-04-07 07:19:16 +00:00
Code Issues Projects Releases Packages Wiki Activity

Check specific context

Browse source
This commit is contained in:
Fraser Waters 2025-02-21 08:33:59 +00:00
parent 3912d67d76
commit d848cfcdf8
1 changed files with 6 additions and 3 deletions

9
command_test.go
View file

@ -2925,6 +2925,9 @@ func TestUnknownFlagShouldReturnSameErrorRegardlessOfArgPosition(t *testing.T) {
func TestHelpFuncExecuted(t *testing.T) { func TestHelpFuncExecuted(t *testing.T) {
helpText := "Long description" helpText := "Long description"
// Create a context that will be unique, not just the background context
executionCtx, _ := context.WithCancel(context.Background())
child := &Command{Use: "child", Run: emptyRun} child := &Command{Use: "child", Run: emptyRun}
child.SetHelpFunc(func(cmd *Command, args []string) { child.SetHelpFunc(func(cmd *Command, args []string) {
_, err := cmd.OutOrStdout().Write([]byte(helpText)) _, err := cmd.OutOrStdout().Write([]byte(helpText))
@ -2933,15 +2936,15 @@ func TestHelpFuncExecuted(t *testing.T) {
} }
// Test for https://github.com/spf13/cobra/issues/2240 // Test for https://github.com/spf13/cobra/issues/2240
if cmd.Context() == nil { if cmd.Context() != executionCtx {
t.Error("Context is nil") t.Error("Context doesn't equal the execution context")
} }
}) })
rootCmd := &Command{Use: "root", Run: emptyRun} rootCmd := &Command{Use: "root", Run: emptyRun}
rootCmd.AddCommand(child) rootCmd.AddCommand(child)
output, err := executeCommand(rootCmd, "help", "child") output, err := executeCommandWithContext(executionCtx, rootCmd, "help", "child")
if err != nil { if err != nil {
t.Errorf("Unexpected error: %v", err) t.Errorf("Unexpected error: %v", err)
} }