Merge pull request #64 from alexflint/repeated-unmarshaltext

Fix repeated arguments implementing TextUnmarshaler

1 files changed, 26 insertions, 0 deletions

diff --git a/parse_test.go b/parse_test.go
index 925a23e..1461c02 100644
--- a/parse_test.go
+++ b/parse_test.go
@@ -599,6 +599,32 @@ func TestTextUnmarshaler(t *testing.T) {
 	assert.Equal(t, 3, args.Foo.val)
 }
 
+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 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)
+}
+
 type boolUnmarshaler bool
 
 func (p *boolUnmarshaler) UnmarshalText(b []byte) error {