package flag import ( "testing" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" ) func TestParse(t *testing.T) { testCases := []struct { desc string args []string element interface{} expected map[string]string }{ { desc: "no args", args: nil, expected: map[string]string{}, }, { desc: "bool value", args: []string{"--foo"}, element: &struct { Foo bool }{}, expected: map[string]string{ "traefik.foo": "true", }, }, { desc: "bool value capitalized", args: []string{"--Foo"}, element: &struct { Foo bool }{}, expected: map[string]string{ "traefik.Foo": "true", }, }, { desc: "equal", args: []string{"--foo=bar"}, element: &struct { Foo string }{}, expected: map[string]string{ "traefik.foo": "bar", }, }, { desc: "equal", args: []string{"--Foo=Bar"}, element: &struct { Foo string }{}, expected: map[string]string{ "traefik.Foo": "Bar", }, }, { desc: "space separated", args: []string{"--foo", "bar"}, element: &struct { Foo string }{}, expected: map[string]string{ "traefik.foo": "bar", }, }, { desc: "space separated capitalized", args: []string{"--Foo", "Bar"}, element: &struct { Foo string }{}, expected: map[string]string{ "traefik.Foo": "Bar", }, }, { desc: "space separated with end of parameter", args: []string{"--foo=bir", "--", "--bar"}, element: &struct { Foo string }{}, expected: map[string]string{ "traefik.foo": "bir", }, }, { desc: "multiple bool flags without value", args: []string{"--foo", "--bar"}, element: &struct { Foo bool Bar bool }{}, expected: map[string]string{ "traefik.foo": "true", "traefik.bar": "true", }, }, { desc: "slice with several flags", args: []string{"--foo=bar", "--foo=baz"}, element: &struct { Foo []string }{}, expected: map[string]string{ "traefik.foo": "bar,baz", }, }, { desc: "map string", args: []string{"--foo.name=bar"}, element: &struct { Foo map[string]string }{}, expected: map[string]string{ "traefik.foo.name": "bar", }, }, { desc: "map string capitalized", args: []string{"--foo.Name=Bar"}, element: &struct { Foo map[string]string }{}, expected: map[string]string{ "traefik.foo.Name": "Bar", }, }, { desc: "map struct", args: []string{"--foo.name.value=bar"}, element: &struct { Foo map[string]struct{ Value string } }{}, expected: map[string]string{ "traefik.foo.name.value": "bar", }, }, { desc: "map struct with sub-struct", args: []string{"--foo.name.bar.value=bar"}, element: &struct { Foo map[string]struct { Bar *struct{ Value string } } }{}, expected: map[string]string{ "traefik.foo.name.bar.value": "bar", }, }, { desc: "map struct with sub-map", args: []string{"--foo.name1.bar.name2.value=bar"}, element: &struct { Foo map[string]struct { Bar map[string]struct{ Value string } } }{}, expected: map[string]string{ "traefik.foo.name1.bar.name2.value": "bar", }, }, { desc: "slice with several flags 2", args: []string{"--foo", "bar", "--foo", "baz"}, element: &struct { Foo []string }{}, expected: map[string]string{ "traefik.foo": "bar,baz", }, }, { desc: "slice with several flags 3", args: []string{"--foo", "bar", "--foo=", "--baz"}, element: &struct { Foo []string Baz bool }{}, expected: map[string]string{ "traefik.foo": "bar,", "traefik.baz": "true", }, }, { desc: "slice with several flags 4", args: []string{"--foo", "bar", "--foo", "--baz"}, element: &struct { Foo []string Baz bool }{}, expected: map[string]string{ "traefik.foo": "bar,--baz", }, }, { desc: "multiple string flag", element: &struct { Foo string }{}, args: []string{"--foo=bar", "--foo=baz"}, expected: map[string]string{ "traefik.foo": "baz", }, }, { desc: "multiple string flag 2", element: &struct { Foo string }{}, args: []string{"--foo", "bar", "--foo", "baz"}, expected: map[string]string{ "traefik.foo": "baz", }, }, { desc: "string without value", element: &struct { Foo string Bar bool }{}, args: []string{"--foo", "--bar"}, expected: map[string]string{ "traefik.foo": "--bar", }, }, { desc: "struct pointer value", args: []string{"--foo"}, element: &struct { Foo *struct{ Field string } }{}, expected: map[string]string{ "traefik.foo": "true", }, }, { desc: "map string case sensitive", args: []string{"--foo.caseSensitiveName=barBoo"}, element: &struct { Foo map[string]string }{}, expected: map[string]string{ "traefik.foo.caseSensitiveName": "barBoo", }, }, { desc: "map struct with sub-map case senstitive", args: []string{"--foo.Name1.bar.name2.value=firstValue", "--foo.naMe1.bar.name2.value=secondValue"}, element: &struct { Foo map[string]struct { Bar map[string]struct{ Value string } } }{}, expected: map[string]string{ "traefik.foo.Name1.bar.name2.value": "secondValue", }, }, { desc: "map struct with sub-map and different case", args: []string{"--foo.Name1.bar.name2.value=firstValue", "--foo.naMe1.bar.name2.value=secondValue"}, element: &struct { Foo map[string]struct { Bar map[string]struct{ Value string } } }{}, expected: map[string]string{ "traefik.foo.Name1.bar.name2.value": "secondValue", }, }, { desc: "slice with several flags 2 and different cases.", args: []string{"--foo", "bar", "--Foo", "baz"}, element: &struct { Foo []string }{}, expected: map[string]string{ "traefik.foo": "bar,baz", }, }, } for _, test := range testCases { test := test t.Run(test.desc, func(t *testing.T) { t.Parallel() fl, err := Parse(test.args, test.element) require.NoError(t, err) assert.Equal(t, test.expected, fl) }) } } func TestParse_Errors(t *testing.T) { testCases := []struct { desc string args []string element interface{} }{ { desc: "triple hyphen", args: []string{"---foo"}, element: &struct { Foo bool }{}, }, { desc: "equal", args: []string{"--=foo"}, element: &struct { Foo bool }{}, }, { desc: "string without value", element: &struct { Foo string Bar bool }{}, args: []string{"--foo"}, }, } for _, test := range testCases { test := test t.Run(test.desc, func(t *testing.T) { t.Parallel() _, err := Parse(test.args, test.element) require.Error(t, err) }) } }