diff options
Diffstat (limited to 'parse_test.go')
| -rw-r--r-- | parse_test.go | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/parse_test.go b/parse_test.go index d201b2f..b72563c 100644 --- a/parse_test.go +++ b/parse_test.go @@ -646,6 +646,16 @@ func (f *textUnmarshaler) UnmarshalText(b []byte) error { func TestTextUnmarshaler(t *testing.T) { // fields that implement TextUnmarshaler should be parsed using that interface var args struct { + Foo textUnmarshaler + } + err := parse("--foo abc", &args) + require.NoError(t, err) + assert.Equal(t, 3, args.Foo.val) +} + +func TestPtrToTextUnmarshaler(t *testing.T) { + // fields that implement TextUnmarshaler should be parsed using that interface + var args struct { Foo *textUnmarshaler } err := parse("--foo abc", &args) @@ -656,6 +666,19 @@ func TestTextUnmarshaler(t *testing.T) { func TestRepeatedTextUnmarshaler(t *testing.T) { // fields that implement TextUnmarshaler should be parsed using that interface var args struct { + Foo []textUnmarshaler + } + err := parse("--foo abc d ef", &args) + require.NoError(t, err) + require.Len(t, args.Foo, 3) + assert.Equal(t, 3, args.Foo[0].val) + assert.Equal(t, 1, args.Foo[1].val) + assert.Equal(t, 2, args.Foo[2].val) +} + +func TestRepeatedPtrToTextUnmarshaler(t *testing.T) { + // fields that implement TextUnmarshaler should be parsed using that interface + var args struct { Foo []*textUnmarshaler } err := parse("--foo abc d ef", &args) @@ -669,6 +692,19 @@ func TestRepeatedTextUnmarshaler(t *testing.T) { func TestPositionalTextUnmarshaler(t *testing.T) { // fields that implement TextUnmarshaler should be parsed using that interface var args struct { + Foo []textUnmarshaler `arg:"positional"` + } + err := parse("abc d ef", &args) + require.NoError(t, err) + require.Len(t, args.Foo, 3) + assert.Equal(t, 3, args.Foo[0].val) + assert.Equal(t, 1, args.Foo[1].val) + assert.Equal(t, 2, args.Foo[2].val) +} + +func TestPositionalPtrToTextUnmarshaler(t *testing.T) { + // fields that implement TextUnmarshaler should be parsed using that interface + var args struct { Foo []*textUnmarshaler `arg:"positional"` } err := parse("abc d ef", &args) |
