Alec Thomas
98 lines
2.6 KiB
Go
98 lines
2.6 KiB
Go
package kong
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestValueMapper(t *testing.T) {
|
|
var cli struct {
|
|
Flag string
|
|
}
|
|
k := mustNew(t, &cli, ValueMapper(&cli.Flag, testMooMapper{}))
|
|
_, err := k.Parse(nil)
|
|
require.NoError(t, err)
|
|
require.Equal(t, "", cli.Flag)
|
|
_, err = k.Parse([]string{"--flag"})
|
|
require.NoError(t, err)
|
|
require.Equal(t, "MOO", cli.Flag)
|
|
}
|
|
|
|
func TestNamedMapper(t *testing.T) {
|
|
var cli struct {
|
|
Flag string `type:"moo"`
|
|
}
|
|
k := mustNew(t, &cli, NamedMapper("moo", testMooMapper{}))
|
|
_, err := k.Parse(nil)
|
|
require.NoError(t, err)
|
|
require.Equal(t, "", cli.Flag)
|
|
_, err = k.Parse([]string{"--flag"})
|
|
require.NoError(t, err)
|
|
require.Equal(t, "MOO", cli.Flag)
|
|
}
|
|
|
|
type testMooMapper struct {
|
|
text string
|
|
}
|
|
|
|
func (t testMooMapper) Decode(ctx *DecodeContext, target reflect.Value) error {
|
|
if t.text == "" {
|
|
target.SetString("MOO")
|
|
} else {
|
|
target.SetString(t.text)
|
|
}
|
|
return nil
|
|
}
|
|
func (testMooMapper) IsBool() bool { return true }
|
|
|
|
func TestTimeMapper(t *testing.T) {
|
|
var cli struct {
|
|
Flag time.Time `format:"2006"`
|
|
}
|
|
k := mustNew(t, &cli)
|
|
_, err := k.Parse([]string{"--flag=2008"})
|
|
require.NoError(t, err)
|
|
expected, err := time.Parse("2006", "2008")
|
|
require.NoError(t, err)
|
|
require.Equal(t, 2008, expected.Year())
|
|
require.Equal(t, expected, cli.Flag)
|
|
}
|
|
|
|
func TestDurationMapper(t *testing.T) {
|
|
var cli struct {
|
|
Flag time.Duration
|
|
}
|
|
k := mustNew(t, &cli)
|
|
_, err := k.Parse([]string{"--flag=5s"})
|
|
require.NoError(t, err)
|
|
require.Equal(t, time.Second*5, cli.Flag)
|
|
}
|
|
|
|
func TestSplitEscaped(t *testing.T) {
|
|
require.Equal(t, []string{"a", "b"}, SplitEscaped("a,b", ','))
|
|
require.Equal(t, []string{"a,b", "c"}, SplitEscaped(`a\,b,c`, ','))
|
|
}
|
|
|
|
func TestJoinEscaped(t *testing.T) {
|
|
require.Equal(t, `a,b`, JoinEscaped([]string{"a", "b"}, ','))
|
|
require.Equal(t, `a\,b,c`, JoinEscaped([]string{`a,b`, `c`}, ','))
|
|
require.Equal(t, JoinEscaped(SplitEscaped(`a\,b,c`, ','), ','), `a\,b,c`)
|
|
}
|
|
|
|
func TestMapWithNamedTypes(t *testing.T) {
|
|
var cli struct {
|
|
TypedValue map[string]string `type:":moo"`
|
|
TypedKey map[string]string `type:"upper:"`
|
|
}
|
|
k := mustNew(t, &cli, NamedMapper("moo", testMooMapper{}), NamedMapper("upper", testUppercaseMapper{}))
|
|
_, err := k.Parse([]string{"--typed-value", "first=5s", "--typed-value", "second=10s"})
|
|
require.NoError(t, err)
|
|
require.Equal(t, map[string]string{"first": "MOO", "second": "MOO"}, cli.TypedValue)
|
|
_, err = k.Parse([]string{"--typed-key", "first=5s", "--typed-key", "second=10s"})
|
|
require.NoError(t, err)
|
|
require.Equal(t, map[string]string{"FIRST": "5s", "SECOND": "10s"}, cli.TypedKey)
|
|
}
|