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

Create TestSetContext() for testing passing context between hooks

Browse source
This commit is contained in:
Unknown 2020-05-19 17:07:26 -03:00
parent a081152dc5
commit d0e2899afb
1 changed files with 32 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
command_test.go
View file

@ -148,6 +148,38 @@ func TestSubcommandExecuteC(t *testing.T) {
}
}
func TestSetContext(t *testing.T) {
ctx := context.TODO()
aKey := "akey"
aVal := "aval"
parentRun := func(cmd *Command, args []string) {
ctx := context.WithValue(cmd.Context(), aKey, aVal)
cmd.SetContext(ctx)
}
childRun := func(cmd *Command, args []string) {
if val := cmd.Context().Value(aKey); val != aVal {
t.Errorf(`Context attribute not found in child command. Expected: "%+v". Have: "%+v"`, aVal, val)
}
}
rootCmd := &Command{Use: "root", Run: parentRun, PersistentPreRun: parentRun}
childCmd := &Command{Use: "child", Run: childRun}
rootCmd.AddCommand(childCmd)
if _, err := executeCommandWithContext(ctx, rootCmd, ""); err != nil {
t.Errorf("Root command must not fail: %+v", err)
}
if _, err := executeCommandWithContext(ctx, rootCmd, "child"); err != nil {
t.Errorf("Subcommand must not fail: %+v", err)
}
}
func TestExecuteContext(t *testing.T) {
ctx := context.TODO()