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

Change key to struct type

Browse source
This commit is contained in:
joshcarp 2022-03-16 17:26:58 +11:00
parent 9439d8696c
commit d969ce0e38
1 changed files with 18 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
command_test.go
View file

@ -2060,11 +2060,12 @@ func TestFParseErrWhitelistSiblingCommand(t *testing.T) {
}
func TestSetContext(t *testing.T) {
key, val := "foo", "bar"
type key struct{}
val := "val"
root := &Command{
Use: "root",
Run: func(cmd *Command, args []string) {
key := cmd.Context().Value(key)
key := cmd.Context().Value(key{})
got, ok := key.(string)
if !ok {
t.Error("key not found in context")
@ -2074,7 +2075,8 @@ func TestSetContext(t *testing.T) {
}
},
}
ctx := context.WithValue(context.Background(), key, val)
ctx := context.WithValue(context.Background(), key{}, val)
root.SetContext(ctx)
err := root.Execute()
if err != nil {
@ -2083,16 +2085,17 @@ func TestSetContext(t *testing.T) {
}
func TestSetContextPreRun(t *testing.T) {
key, val := "foo", "bar"
type key struct{}
val := "bar"
root := &Command{
Use: "root",
PreRun: func(cmd *Command, args []string) {
ctx := context.WithValue(cmd.Context(), key, val)
ctx := context.WithValue(cmd.Context(), key{}, val)
cmd.SetContext(ctx)
},
Run: func(cmd *Command, args []string) {
key := cmd.Context().Value(key)
got, ok := key.(string)
val := cmd.Context().Value(key{})
got, ok := val.(string)
if !ok {
t.Error("key not found in context")
}
@ -2108,18 +2111,19 @@ func TestSetContextPreRun(t *testing.T) {
}
func TestSetContextPreRunOverwrite(t *testing.T) {
key, val := "foo", "bar"
type key struct{}
val := "bar"
root := &Command{
Use: "root",
Run: func(cmd *Command, args []string) {
key := cmd.Context().Value(key)
key := cmd.Context().Value(key{})
_, ok := key.(string)
if ok {
t.Error("key found in context when not expected")
}
},
}
ctx := context.WithValue(context.Background(), key, val)
ctx := context.WithValue(context.Background(), key{}, val)
root.SetContext(ctx)
err := root.ExecuteContext(context.Background())
if err != nil {
@ -2128,18 +2132,19 @@ func TestSetContextPreRunOverwrite(t *testing.T) {
}
func TestSetContextPersistentPreRun(t *testing.T) {
key, val := "foo", "bar"
type key struct{}
val := "bar"
root := &Command{
Use: "root",
PersistentPreRun: func(cmd *Command, args []string) {
ctx := context.WithValue(cmd.Context(), key, val)
ctx := context.WithValue(cmd.Context(), key{}, val)
cmd.SetContext(ctx)
},
}
child := &Command{
Use: "child",
Run: func(cmd *Command, args []string) {
key := cmd.Context().Value(key)
key := cmd.Context().Value(key{})
got, ok := key.(string)
if !ok {
t.Error("key not found in context")