forked from alecthomas/kong
-
Notifications
You must be signed in to change notification settings - Fork 0
/
model_test.go
72 lines (67 loc) · 2.5 KB
/
model_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
package kong_test
import (
"testing"
"github.com/stretchr/testify/require"
)
func TestModelApplicationCommands(t *testing.T) {
var cli struct {
One struct {
Two struct {
} `kong:"cmd"`
Three struct {
Four struct {
Four string `kong:"arg"`
} `kong:"arg"`
} `kong:"cmd"`
} `kong:"cmd"`
}
p := mustNew(t, &cli)
actual := []string{}
for _, cmd := range p.Model.Leaves(false) {
actual = append(actual, cmd.Path())
}
require.Equal(t, []string{"one two", "one three <four>"}, actual)
}
func TestFlagString(t *testing.T) {
var cli struct {
String string
DefaultInt int `default:"42"`
DefaultStr string `default:"hello"`
Placeholder string `placeholder:"world"`
DefaultPlaceholder string `default:"hello" placeholder:"world"`
SliceSep []string
SliceNoSep []string `sep:"none"`
SliceDefault []string `default:"hello"`
SlicePlaceholder []string `placeholder:"world"`
SliceDefaultPlaceholder []string `default:"hello" placeholder:"world"`
MapSep map[string]string
MapNoSep map[string]string `mapsep:"none"`
MapDefault map[string]string `mapsep:"none" default:"hello"`
MapPlaceholder map[string]string `mapsep:"none" placeholder:"world"`
Counter int `type:"counter"`
}
tests := map[string]string{
"help": "-h, --help",
"string": "--string=STRING",
"default-int": "--default-int=42",
"default-str": `--default-str="hello"`,
"placeholder": "--placeholder=world",
"default-placeholder": "--default-placeholder=world",
"slice-sep": "--slice-sep=SLICE-SEP,...",
"slice-no-sep": "--slice-no-sep=SLICE-NO-SEP",
"slice-default": "--slice-default=hello,...",
"slice-placeholder": "--slice-placeholder=world,...",
"slice-default-placeholder": "--slice-default-placeholder=world,...",
"map-sep": "--map-sep=KEY=VALUE;...",
"map-no-sep": "--map-no-sep=KEY=VALUE",
"map-default": "--map-default=hello",
"map-placeholder": "--map-placeholder=world",
"counter": "--counter",
}
p := mustNew(t, &cli)
for _, flag := range p.Model.Flags {
want, ok := tests[flag.Name]
require.Truef(t, ok, "unknown flag name: %s", flag.Name)
require.Equal(t, want, flag.String())
}
}