squash merge from dev branch

vendor/gopkg.in/alecthomas/kingpin.v2/args_test.go

@@ -0,0 +1,84 @@
+package kingpin
+
+import (
+	"io/ioutil"
+	"os"
+	"testing"
+
+	"github.com/alecthomas/assert"
+)
+
+func TestArgRemainder(t *testing.T) {
+	app := New("test", "")
+	v := app.Arg("test", "").Strings()
+	args := []string{"hello", "world"}
+	_, err := app.Parse(args)
+	assert.NoError(t, err)
+	assert.Equal(t, args, *v)
+}
+
+func TestArgRemainderErrorsWhenNotLast(t *testing.T) {
+	a := newArgGroup()
+	a.Arg("test", "").Strings()
+	a.Arg("test2", "").String()
+	assert.Error(t, a.init())
+}
+
+func TestArgMultipleRequired(t *testing.T) {
+	terminated := false
+	app := New("test", "")
+	app.Version("0.0.0").Writer(ioutil.Discard)
+	app.Arg("a", "").Required().String()
+	app.Arg("b", "").Required().String()
+	app.Terminate(func(int) { terminated = true })
+
+	_, err := app.Parse([]string{})
+	assert.Error(t, err)
+	_, err = app.Parse([]string{"A"})
+	assert.Error(t, err)
+	_, err = app.Parse([]string{"A", "B"})
+	assert.NoError(t, err)
+	_, err = app.Parse([]string{"--version"})
+	assert.True(t, terminated)
+}
+
+func TestInvalidArgsDefaultCanBeOverridden(t *testing.T) {
+	app := New("test", "")
+	app.Arg("a", "").Default("invalid").Bool()
+	_, err := app.Parse([]string{})
+	assert.Error(t, err)
+}
+
+func TestArgMultipleValuesDefault(t *testing.T) {
+	app := New("test", "")
+	a := app.Arg("a", "").Default("default1", "default2").Strings()
+	_, err := app.Parse([]string{})
+	assert.NoError(t, err)
+	assert.Equal(t, []string{"default1", "default2"}, *a)
+}
+
+func TestRequiredArgWithEnvarMissingErrors(t *testing.T) {
+	app := newTestApp()
+	app.Arg("t", "").Envar("TEST_ARG_ENVAR").Required().Int()
+	_, err := app.Parse([]string{})
+	assert.Error(t, err)
+}
+
+func TestArgRequiredWithEnvar(t *testing.T) {
+	os.Setenv("TEST_ARG_ENVAR", "123")
+	app := newTestApp()
+	flag := app.Arg("t", "").Envar("TEST_ARG_ENVAR").Required().Int()
+	_, err := app.Parse([]string{})
+	assert.NoError(t, err)
+	assert.Equal(t, 123, *flag)
+}
+
+func TestSubcommandArgRequiredWithEnvar(t *testing.T) {
+	os.Setenv("TEST_ARG_ENVAR", "123")
+	app := newTestApp()
+	cmd := app.Command("command", "")
+	flag := cmd.Arg("t", "").Envar("TEST_ARG_ENVAR").Required().Int()
+	_, err := app.Parse([]string{"command"})
+	assert.NoError(t, err)
+	assert.Equal(t, 123, *flag)
+}