parse_test: add tests covering new TextUnamarshaler value support

Signed-off-by: Pavel Borzenkov <[email protected]>

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)