123 lines
3.1 KiB
Go
123 lines
3.1 KiB
Go
|
package kingpin
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io/ioutil"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestParserExpandFromFile(t *testing.T) {
|
||
|
|
f, err := ioutil.TempFile("", "")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
defer os.Remove(f.Name())
|
||
|
|
f.WriteString("hello\nworld\n")
|
||
|
|
f.Close()
|
||
|
|
|
||
|
|
app := New("test", "")
|
||
|
|
arg0 := app.Arg("arg0", "").String()
|
||
|
|
arg1 := app.Arg("arg1", "").String()
|
||
|
|
|
||
|
|
_, err = app.Parse([]string{"@" + f.Name()})
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, "hello", *arg0)
|
||
|
|
assert.Equal(t, "world", *arg1)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestParserExpandFromFileLeadingArg(t *testing.T) {
|
||
|
|
f, err := ioutil.TempFile("", "")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
defer os.Remove(f.Name())
|
||
|
|
f.WriteString("hello\nworld\n")
|
||
|
|
f.Close()
|
||
|
|
|
||
|
|
app := New("test", "")
|
||
|
|
arg0 := app.Arg("arg0", "").String()
|
||
|
|
arg1 := app.Arg("arg1", "").String()
|
||
|
|
arg2 := app.Arg("arg2", "").String()
|
||
|
|
|
||
|
|
_, err = app.Parse([]string{"prefix", "@" + f.Name()})
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, "prefix", *arg0)
|
||
|
|
assert.Equal(t, "hello", *arg1)
|
||
|
|
assert.Equal(t, "world", *arg2)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestParserExpandFromFileTrailingArg(t *testing.T) {
|
||
|
|
f, err := ioutil.TempFile("", "")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
defer os.Remove(f.Name())
|
||
|
|
f.WriteString("hello\nworld\n")
|
||
|
|
f.Close()
|
||
|
|
|
||
|
|
app := New("test", "")
|
||
|
|
arg0 := app.Arg("arg0", "").String()
|
||
|
|
arg1 := app.Arg("arg1", "").String()
|
||
|
|
arg2 := app.Arg("arg2", "").String()
|
||
|
|
|
||
|
|
_, err = app.Parse([]string{"@" + f.Name(), "suffix"})
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, "hello", *arg0)
|
||
|
|
assert.Equal(t, "world", *arg1)
|
||
|
|
assert.Equal(t, "suffix", *arg2)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestParserExpandFromFileMultipleSurroundingArgs(t *testing.T) {
|
||
|
|
f, err := ioutil.TempFile("", "")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
defer os.Remove(f.Name())
|
||
|
|
f.WriteString("hello\nworld\n")
|
||
|
|
f.Close()
|
||
|
|
|
||
|
|
app := New("test", "")
|
||
|
|
arg0 := app.Arg("arg0", "").String()
|
||
|
|
arg1 := app.Arg("arg1", "").String()
|
||
|
|
arg2 := app.Arg("arg2", "").String()
|
||
|
|
arg3 := app.Arg("arg3", "").String()
|
||
|
|
|
||
|
|
_, err = app.Parse([]string{"prefix", "@" + f.Name(), "suffix"})
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, "prefix", *arg0)
|
||
|
|
assert.Equal(t, "hello", *arg1)
|
||
|
|
assert.Equal(t, "world", *arg2)
|
||
|
|
assert.Equal(t, "suffix", *arg3)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestParserExpandFromFileMultipleFlags(t *testing.T) {
|
||
|
|
f, err := ioutil.TempFile("", "")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
defer os.Remove(f.Name())
|
||
|
|
f.WriteString("--flag1=f1\n--flag2=f2\n")
|
||
|
|
f.Close()
|
||
|
|
|
||
|
|
app := New("test", "")
|
||
|
|
flag0 := app.Flag("flag0", "").String()
|
||
|
|
flag1 := app.Flag("flag1", "").String()
|
||
|
|
flag2 := app.Flag("flag2", "").String()
|
||
|
|
flag3 := app.Flag("flag3", "").String()
|
||
|
|
|
||
|
|
_, err = app.Parse([]string{"--flag0=f0", "@" + f.Name(), "--flag3=f3"})
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, "f0", *flag0)
|
||
|
|
assert.Equal(t, "f1", *flag1)
|
||
|
|
assert.Equal(t, "f2", *flag2)
|
||
|
|
assert.Equal(t, "f3", *flag3)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestParseContextPush(t *testing.T) {
|
||
|
|
app := New("test", "")
|
||
|
|
app.Command("foo", "").Command("bar", "")
|
||
|
|
c := tokenize([]string{"foo", "bar"}, false)
|
||
|
|
a := c.Next()
|
||
|
|
assert.Equal(t, TokenArg, a.Type)
|
||
|
|
b := c.Next()
|
||
|
|
assert.Equal(t, TokenArg, b.Type)
|
||
|
|
c.Push(b)
|
||
|
|
c.Push(a)
|
||
|
|
a = c.Next()
|
||
|
|
assert.Equal(t, "foo", a.Value)
|
||
|
|
b = c.Next()
|
||
|
|
assert.Equal(t, "bar", b.Value)
|
||
|
|
}
|