-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathfield_test.go
95 lines (89 loc) · 2.1 KB
/
field_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
package cli
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestFieldIgnoreMinusTag(t *testing.T) {
type Cfg struct {
Ignored string `cli:"-"`
}
fields, _, err := defaultCLI.getFieldsFromConfig(&Cfg{})
require.NoError(t, err)
assert.Len(t, fields, 0)
}
func TestFieldUnknownTagError(t *testing.T) {
type Cfg struct {
Foo string `cli:"asdfasdf"`
}
_, _, err := defaultCLI.getFieldsFromConfig(&Cfg{})
assert.Error(t, err)
}
func TestFieldEmbedded(t *testing.T) {
type EmbeddedCfg struct {
Bar string
}
type Cfg struct {
Foo string
EmbeddedCfg
}
fields, _, err := defaultCLI.getFieldsFromConfig(&Cfg{})
require.NoError(t, err)
assert.Len(t, fields, 2)
assert.Equal(t, "foo", fields[0].Name)
assert.Equal(t, "bar", fields[1].Name)
}
func TestFieldAppend(t *testing.T) {
getFieldSet := func(t *testing.T, cfg interface{}) func(s string) {
fields, _, err := defaultCLI.getFieldsFromConfig(cfg)
require.NoError(t, err)
require.Len(t, fields, 1)
flag := fields[0].value
return func(s string) {
err := flag.Set(s)
require.NoError(t, err)
}
}
t.Run("[]string", func(t *testing.T) {
cfg := struct {
Vars []string `cli:"append,short=v"`
}{}
set := getFieldSet(t, &cfg)
set("aaa")
set("bbb")
set("ccc")
assert.Equal(t, []string{"aaa", "bbb", "ccc"}, cfg.Vars)
})
t.Run("[]*string", func(t *testing.T) {
cfg := struct {
Vars []*string `cli:"append,short=v"`
}{}
set := getFieldSet(t, &cfg)
set("aaa")
set("bbb")
set("ccc")
s := func(v string) *string { return &v }
assert.EqualValues(t, []*string{s("aaa"), s("bbb"), s("ccc")}, cfg.Vars)
})
t.Run("[]int", func(t *testing.T) {
cfg := struct {
Vars []int `cli:"append,short=v"`
}{}
set := getFieldSet(t, &cfg)
set("1")
set("2")
set("3")
assert.Equal(t, []int{1, 2, 3}, cfg.Vars)
})
t.Run("[]*int", func(t *testing.T) {
cfg := struct {
Vars []*int `cli:"append,short=v"`
}{}
set := getFieldSet(t, &cfg)
set("1")
set("2")
set("3")
i := func(v int) *int { return &v }
assert.EqualValues(t, []*int{i(1), i(2), i(3)}, cfg.Vars)
})
}