summaryrefslogtreecommitdiff
path: root/test/parse_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'test/parse_test.go')
-rw-r--r--test/parse_test.go88
1 files changed, 65 insertions, 23 deletions
diff --git a/test/parse_test.go b/test/parse_test.go
index d53b483..249cbf3 100644
--- a/test/parse_test.go
+++ b/test/parse_test.go
@@ -28,11 +28,11 @@ func parse(cmdline string, dest interface{}) error {
}
func pparse(cmdline string, dest interface{}) (*Parser, error) {
- return parseWithEnv(cmdline, nil, dest)
+ return parseWithEnv(Config{}, cmdline, nil, dest)
}
-func parseWithEnv(cmdline string, env []string, dest interface{}) (*Parser, error) {
- p, err := NewParser(Config{}, dest)
+func parseWithEnv(config Config, cmdline string, env []string, dest interface{}) (*Parser, error) {
+ p, err := NewParser(config, dest)
if err != nil {
return nil, err
}
@@ -231,7 +231,7 @@ func TestRequiredWithEnvOnly(t *testing.T) {
var args struct {
Foo string `arg:"required,--,-,env:FOO"`
}
- _, err := parseWithEnv("", []string{}, &args)
+ _, err := parseWithEnv(Config{}, "", []string{}, &args)
require.Error(t, err, "environment variable FOO is required")
}
@@ -609,6 +609,15 @@ func TestNoMoreOptionsBeforeHelp(t *testing.T) {
assert.NotEqual(t, ErrHelp, err)
}
+func TestNoMoreOptionsTwice(t *testing.T) {
+ var args struct {
+ X []string `arg:"positional"`
+ }
+ err := parse("-- --", &args)
+ require.NoError(t, err)
+ assert.Equal(t, []string{"--"}, args.X)
+}
+
func TestHelpFlag(t *testing.T) {
var args struct {
Foo string
@@ -692,11 +701,26 @@ func TestMustParse(t *testing.T) {
assert.NotNil(t, parser)
}
+func TestMustParseError(t *testing.T) {
+ var args struct {
+ Foo []string `default:""`
+ }
+ var exitCode int
+ var stdout bytes.Buffer
+ mustParseExit = func(code int) { exitCode = code }
+ mustParseOut = &stdout
+ os.Args = []string{"example"}
+ parser := MustParse(&args)
+ assert.Nil(t, parser)
+ assert.Equal(t, 2, exitCode)
+ assert.Contains(t, stdout.String(), "default values are not supported for slice or map fields")
+}
+
func TestEnvironmentVariable(t *testing.T) {
var args struct {
Foo string `arg:"env"`
}
- _, err := parseWithEnv("", []string{"FOO=bar"}, &args)
+ _, err := parseWithEnv(Config{}, "", []string{"FOO=bar"}, &args)
require.NoError(t, err)
assert.Equal(t, "bar", args.Foo)
}
@@ -705,7 +729,7 @@ func TestEnvironmentVariableNotPresent(t *testing.T) {
var args struct {
NotPresent string `arg:"env"`
}
- _, err := parseWithEnv("", nil, &args)
+ _, err := parseWithEnv(Config{}, "", nil, &args)
require.NoError(t, err)
assert.Equal(t, "", args.NotPresent)
}
@@ -714,7 +738,7 @@ func TestEnvironmentVariableOverrideName(t *testing.T) {
var args struct {
Foo string `arg:"env:BAZ"`
}
- _, err := parseWithEnv("", []string{"BAZ=bar"}, &args)
+ _, err := parseWithEnv(Config{}, "", []string{"BAZ=bar"}, &args)
require.NoError(t, err)
assert.Equal(t, "bar", args.Foo)
}
@@ -723,7 +747,7 @@ func TestEnvironmentVariableOverrideArgument(t *testing.T) {
var args struct {
Foo string `arg:"env"`
}
- _, err := parseWithEnv("--foo zzz", []string{"FOO=bar"}, &args)
+ _, err := parseWithEnv(Config{}, "--foo zzz", []string{"FOO=bar"}, &args)
require.NoError(t, err)
assert.Equal(t, "zzz", args.Foo)
}
@@ -732,7 +756,7 @@ func TestEnvironmentVariableError(t *testing.T) {
var args struct {
Foo int `arg:"env"`
}
- _, err := parseWithEnv("", []string{"FOO=bar"}, &args)
+ _, err := parseWithEnv(Config{}, "", []string{"FOO=bar"}, &args)
assert.Error(t, err)
}
@@ -740,7 +764,7 @@ func TestEnvironmentVariableRequired(t *testing.T) {
var args struct {
Foo string `arg:"env,required"`
}
- _, err := parseWithEnv("", []string{"FOO=bar"}, &args)
+ _, err := parseWithEnv(Config{}, "", []string{"FOO=bar"}, &args)
require.NoError(t, err)
assert.Equal(t, "bar", args.Foo)
}
@@ -749,7 +773,7 @@ func TestEnvironmentVariableSliceArgumentString(t *testing.T) {
var args struct {
Foo []string `arg:"env"`
}
- _, err := parseWithEnv("", []string{`FOO=bar,"baz, qux"`}, &args)
+ _, err := parseWithEnv(Config{}, "", []string{`FOO=bar,"baz, qux"`}, &args)
require.NoError(t, err)
assert.Equal(t, []string{"bar", "baz, qux"}, args.Foo)
}
@@ -758,7 +782,7 @@ func TestEnvironmentVariableSliceEmpty(t *testing.T) {
var args struct {
Foo []string `arg:"env"`
}
- _, err := parseWithEnv("", []string{`FOO=`}, &args)
+ _, err := parseWithEnv(Config{}, "", []string{`FOO=`}, &args)
require.NoError(t, err)
assert.Len(t, args.Foo, 0)
}
@@ -767,7 +791,7 @@ func TestEnvironmentVariableSliceArgumentInteger(t *testing.T) {
var args struct {
Foo []int `arg:"env"`
}
- _, err := parseWithEnv("", []string{`FOO=1,99`}, &args)
+ _, err := parseWithEnv(Config{}, "", []string{`FOO=1,99`}, &args)
require.NoError(t, err)
assert.Equal(t, []int{1, 99}, args.Foo)
}
@@ -776,7 +800,7 @@ func TestEnvironmentVariableSliceArgumentFloat(t *testing.T) {
var args struct {
Foo []float32 `arg:"env"`
}
- _, err := parseWithEnv("", []string{`FOO=1.1,99.9`}, &args)
+ _, err := parseWithEnv(Config{}, "", []string{`FOO=1.1,99.9`}, &args)
require.NoError(t, err)
assert.Equal(t, []float32{1.1, 99.9}, args.Foo)
}
@@ -785,7 +809,7 @@ func TestEnvironmentVariableSliceArgumentBool(t *testing.T) {
var args struct {
Foo []bool `arg:"env"`
}
- _, err := parseWithEnv("", []string{`FOO=true,false,0,1`}, &args)
+ _, err := parseWithEnv(Config{}, "", []string{`FOO=true,false,0,1`}, &args)
require.NoError(t, err)
assert.Equal(t, []bool{true, false, false, true}, args.Foo)
}
@@ -794,7 +818,7 @@ func TestEnvironmentVariableSliceArgumentWrongCsv(t *testing.T) {
var args struct {
Foo []int `arg:"env"`
}
- _, err := parseWithEnv("", []string{`FOO=1,99\"`}, &args)
+ _, err := parseWithEnv(Config{}, "", []string{`FOO=1,99\"`}, &args)
assert.Error(t, err)
}
@@ -802,7 +826,7 @@ func TestEnvironmentVariableSliceArgumentWrongType(t *testing.T) {
var args struct {
Foo []bool `arg:"env"`
}
- _, err := parseWithEnv("", []string{`FOO=one,two`}, &args)
+ _, err := parseWithEnv(Config{}, "", []string{`FOO=one,two`}, &args)
assert.Error(t, err)
}
@@ -810,7 +834,7 @@ func TestEnvironmentVariableMap(t *testing.T) {
var args struct {
Foo map[int]string `arg:"env"`
}
- _, err := parseWithEnv("", []string{`FOO=1=one,99=ninetynine`}, &args)
+ _, err := parseWithEnv(Config{}, "", []string{`FOO=1=one,99=ninetynine`}, &args)
require.NoError(t, err)
assert.Len(t, args.Foo, 2)
assert.Equal(t, "one", args.Foo[1])
@@ -821,11 +845,21 @@ func TestEnvironmentVariableEmptyMap(t *testing.T) {
var args struct {
Foo map[int]string `arg:"env"`
}
- _, err := parseWithEnv("", []string{`FOO=`}, &args)
+ _, err := parseWithEnv(Config{}, "", []string{`FOO=`}, &args)
require.NoError(t, err)
assert.Len(t, args.Foo, 0)
}
+func TestEnvironmentVariableWithPrefix(t *testing.T) {
+ var args struct {
+ Foo string `arg:"env"`
+ }
+
+ _, err := parseWithEnv(Config{EnvPrefix: "MYAPP_"}, "", []string{"MYAPP_FOO=bar"}, &args)
+ require.NoError(t, err)
+ assert.Equal(t, "bar", args.Foo)
+}
+
func TestEnvironmentVariableIgnored(t *testing.T) {
var args struct {
Foo string `arg:"env"`
@@ -858,7 +892,7 @@ func TestRequiredEnvironmentOnlyVariableIsMissing(t *testing.T) {
Foo string `arg:"required,--,env:FOO"`
}
- _, err := parseWithEnv("", []string{""}, &args)
+ _, err := parseWithEnv(Config{}, "", []string{""}, &args)
assert.Error(t, err)
}
@@ -867,7 +901,7 @@ func TestOptionalEnvironmentOnlyVariable(t *testing.T) {
Foo string `arg:"env:FOO"`
}
- _, err := parseWithEnv("", []string{}, &args)
+ _, err := parseWithEnv(Config{}, "", []string{}, &args)
assert.NoError(t, err)
}
@@ -906,7 +940,7 @@ func TestParserMustParse(t *testing.T) {
}{
{name: "help", args: struct{}{}, cmdLine: []string{"--help"}, code: 0, output: "display this help and exit"},
{name: "version", args: versioned{}, cmdLine: []string{"--version"}, code: 0, output: "example 3.2.1"},
- {name: "invalid", args: struct{}{}, cmdLine: []string{"invalid"}, code: -1, output: ""},
+ {name: "invalid", args: struct{}{}, cmdLine: []string{"invalid"}, code: 2, output: ""},
}
for _, tt := range tests {
@@ -1556,7 +1590,7 @@ func TestMustParseInvalidParser(t *testing.T) {
}
parser := mustParse(Config{Out: &stdout, Exit: exit}, &args)
assert.Nil(t, parser)
- assert.Equal(t, -1, exitCode)
+ assert.Equal(t, 2, exitCode)
}
func TestMustParsePrintsHelp(t *testing.T) {
@@ -1737,3 +1771,11 @@ func TestSubcommandGlobalFlag_InCommand_Strict_Inner(t *testing.T) {
require.NotNil(t, args.Sub)
assert.True(t, args.Sub.Guard)
}
+
+func TestExitFunctionAndOutStreamGetFilledIn(t *testing.T) {
+ var args struct{}
+ p, err := NewParser(Config{}, &args)
+ require.NoError(t, err)
+ assert.NotNil(t, p.config.Exit) // go prohibits function pointer comparison
+ assert.Equal(t, p.config.Out, os.Stdout)
+}