From 6f2f3b4bf6c3fb40a0d2200ce6affee56cf781d8 Mon Sep 17 00:00:00 2001 From: Alex Flint Date: Wed, 18 Apr 2018 21:23:08 -0700 Subject: [PATCH] drop setScalar --- parse.go | 15 +++++---------- parse_test.go | 13 +++++++++++++ 2 files changed, 18 insertions(+), 10 deletions(-) diff --git a/parse.go b/parse.go index 10c6841..e272b59 100644 --- a/parse.go +++ b/parse.go @@ -159,7 +159,7 @@ func NewParser(config Config, dests ...interface{}) (*Parser, error) { } // Check whether this field is supported. It's good to do this here rather than - // wait until setScalar because it means that a program with invalid argument + // wait until ParseValue because it means that a program with invalid argument // fields will always fail regardless of whether the arguments it received // exercised those fields. var parseable bool @@ -275,7 +275,7 @@ func process(specs []*spec, args []string) error { } if spec.env != "" { if value, found := os.LookupEnv(spec.env); found { - err := setScalar(spec.dest, value) + err := scalar.ParseValue(spec.dest, value) if err != nil { return fmt.Errorf("error processing environment variable %s: %v", spec.env, err) } @@ -355,7 +355,7 @@ func process(specs []*spec, args []string) error { i++ } - err := setScalar(spec.dest, value) + err := scalar.ParseValue(spec.dest, value) if err != nil { return fmt.Errorf("error processing %s: %v", arg, err) } @@ -374,7 +374,7 @@ func process(specs []*spec, args []string) error { } positionals = nil } else if len(positionals) > 0 { - err := setScalar(spec.dest, positionals[0]) + err := scalar.ParseValue(spec.dest, positionals[0]) if err != nil { return fmt.Errorf("error processing %s: %v", spec.long, err) } @@ -438,7 +438,7 @@ func setSlice(dest reflect.Value, values []string, trunc bool) error { for _, s := range values { v := reflect.New(elem) - if err := setScalar(v.Elem(), s); err != nil { + if err := scalar.ParseValue(v.Elem(), s); err != nil { return err } if !ptr { @@ -500,8 +500,3 @@ func isScalar(t reflect.Type) (parseable, boolean bool) { return parseable, false } } - -// set a value from a string -func setScalar(v reflect.Value, s string) error { - return scalar.ParseValue(v, s) -} diff --git a/parse_test.go b/parse_test.go index 925a23e..ce3ad34 100644 --- a/parse_test.go +++ b/parse_test.go @@ -599,6 +599,19 @@ 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) +} + type boolUnmarshaler bool func (p *boolUnmarshaler) UnmarshalText(b []byte) error {