98 lines
2.4 KiB
Go
98 lines
2.4 KiB
Go
package kingpin
|
|
|
|
import (
|
|
"net"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"testing"
|
|
)
|
|
|
|
func TestAccumulatorStrings(t *testing.T) {
|
|
target := []string{}
|
|
acc := newAccumulator(&target, func(v interface{}) Value { return newStringValue(v.(*string)) })
|
|
acc.Set("a")
|
|
assert.Equal(t, []string{"a"}, target)
|
|
acc.Set("b")
|
|
assert.Equal(t, []string{"a", "b"}, target)
|
|
}
|
|
|
|
func TestStrings(t *testing.T) {
|
|
app := New("", "")
|
|
app.Arg("a", "").Required().String()
|
|
app.Arg("b", "").Required().String()
|
|
c := app.Arg("c", "").Required().Strings()
|
|
app.Parse([]string{"a", "b", "a", "b"})
|
|
assert.Equal(t, []string{"a", "b"}, *c)
|
|
}
|
|
|
|
func TestEnum(t *testing.T) {
|
|
app := New("", "")
|
|
a := app.Arg("a", "").Enum("one", "two", "three")
|
|
_, err := app.Parse([]string{"moo"})
|
|
assert.Error(t, err)
|
|
_, err = app.Parse([]string{"one"})
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "one", *a)
|
|
}
|
|
|
|
func TestEnumVar(t *testing.T) {
|
|
app := New("", "")
|
|
var a string
|
|
app.Arg("a", "").EnumVar(&a, "one", "two", "three")
|
|
_, err := app.Parse([]string{"moo"})
|
|
assert.Error(t, err)
|
|
_, err = app.Parse([]string{"one"})
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "one", a)
|
|
}
|
|
|
|
func TestCounter(t *testing.T) {
|
|
app := New("", "")
|
|
c := app.Flag("f", "").Counter()
|
|
_, err := app.Parse([]string{"--f", "--f", "--f"})
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, 3, *c)
|
|
}
|
|
|
|
func TestIPv4Addr(t *testing.T) {
|
|
app := newTestApp()
|
|
flag := app.Flag("addr", "").ResolvedIP()
|
|
_, err := app.Parse([]string{"--addr", net.IPv4(1, 2, 3, 4).String()})
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, *flag)
|
|
assert.Equal(t, net.IPv4(1, 2, 3, 4), *flag)
|
|
}
|
|
|
|
func TestInvalidIPv4Addr(t *testing.T) {
|
|
app := newTestApp()
|
|
app.Flag("addr", "").ResolvedIP()
|
|
_, err := app.Parse([]string{"--addr", "1.2.3.256"})
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestIPv6Addr(t *testing.T) {
|
|
app := newTestApp()
|
|
flag := app.Flag("addr", "").ResolvedIP()
|
|
_, err := app.Parse([]string{"--addr", net.IPv6interfacelocalallnodes.String()})
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, *flag)
|
|
assert.Equal(t, net.IPv6interfacelocalallnodes, *flag)
|
|
}
|
|
|
|
func TestHexBytes(t *testing.T) {
|
|
app := newTestApp()
|
|
actual := app.Arg("bytes", "").HexBytes()
|
|
_, err := app.Parse([]string{"01020aff"})
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, []byte{0x01, 0x02, 0x0a, 0xff}, *actual)
|
|
}
|
|
|
|
func TestSetValueDoesNotReset(t *testing.T) {
|
|
app := newTestApp()
|
|
mapping := map[string]string{
|
|
"key": "value",
|
|
}
|
|
app.Flag("set", "").StringMapVar(&mapping)
|
|
assert.NotEmpty(t, mapping)
|
|
}
|