95 lines
2.5 KiB
Go
95 lines
2.5 KiB
Go
package flags
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
const (
|
|
CMD = "TEST"
|
|
SUMMARY = "SUMMARY"
|
|
)
|
|
|
|
type options struct {
|
|
Descriptor Arg
|
|
Tags Arg
|
|
Source String
|
|
Write Bool
|
|
Help Bool
|
|
}
|
|
|
|
func initOpts() options {
|
|
return options{
|
|
Arg{"", "midfix file descriptor"},
|
|
Arg{"", "tags to add to new file"},
|
|
String{"--source", "-s", "", "path to source list. reads from stdin if omitted."},
|
|
Bool{"--write", "-w", false, "write sorted list back to --key file"},
|
|
Bool{"--help", "-h", false, "show help"},
|
|
}
|
|
}
|
|
|
|
func TestParseArgsInternal(t *testing.T) {
|
|
flags := []Flag{
|
|
&Arg{"", "midfix file descriptor"},
|
|
&Arg{"", "tags to add to new file"},
|
|
&String{"--source", "-s", "", "path to source list. reads from stdin if omitted."},
|
|
&Bool{"--write", "-w", false, "write sorted list back to --key file"},
|
|
&Bool{"--help", "-h", false, "show help"},
|
|
}
|
|
args := []string{"--source", "foo"}
|
|
parseArgsInternal(CMD, SUMMARY, args, initOpts(), flags)
|
|
assert.False(t, flags[0].IsSet())
|
|
assert.False(t, flags[0].IsHelp())
|
|
assert.True(t, flags[2].IsSet())
|
|
assert.True(t, flags[4].IsHelp())
|
|
|
|
}
|
|
|
|
func TestParseArgsHelp(t *testing.T) {
|
|
args := []string{"--help"}
|
|
opts := initOpts()
|
|
err := ParseArgs(CMD, SUMMARY, args, &opts)
|
|
assert.ErrorContains(t, err, "um TEST [descriptor] [tags] [--source] [--write]\n\nSUMMARY")
|
|
// we set the Val true although we don't expect to use it:
|
|
assert.True(t, opts.Help.Val)
|
|
}
|
|
|
|
func TestParseArgsString(t *testing.T) {
|
|
args := []string{"--source", "foo"}
|
|
opts := initOpts()
|
|
err := ParseArgs(CMD, SUMMARY, args, &opts)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, opts.Source.Val, "foo")
|
|
}
|
|
|
|
func TestParseArgs(t *testing.T) {
|
|
opts := initOpts()
|
|
cases := []struct {
|
|
name string
|
|
args []string
|
|
stringVal *string
|
|
stringWant string
|
|
boolVal *bool
|
|
boolWant bool
|
|
}{
|
|
{"arg", []string{"bar"}, &opts.Descriptor.Val, "bar", nil, false},
|
|
{"arg arg", []string{"bar", "foo"}, &opts.Tags.Val, "foo", nil, false},
|
|
{"--bool", []string{"--write"}, nil, "", &opts.Write.Val, true},
|
|
{"arg --bool", []string{"foo", "--write"}, &opts.Descriptor.Val, "foo", &opts.Write.Val, true},
|
|
{"--flag val", []string{"--source", "foo"}, &opts.Source.Val, "foo", nil, false},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
err := ParseArgs(CMD, SUMMARY, tc.args, &opts)
|
|
assert.NoError(t, err)
|
|
if tc.boolVal != nil {
|
|
assert.True(t, *tc.boolVal)
|
|
}
|
|
if tc.stringVal != nil {
|
|
assert.Equal(t, *tc.stringVal, tc.stringWant)
|
|
}
|
|
})
|
|
}
|
|
}
|