Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion string_to_int.go
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ func (s *stringToIntValue) String() string {
}

func stringToIntConv(val string) (interface{}, error) {
val = strings.Trim(val, "[]")
val = strings.TrimSuffix(strings.TrimPrefix(val, "["), "]")
// An empty string would cause an empty map
if len(val) == 0 {
return map[string]int{}, nil
Expand Down
2 changes: 1 addition & 1 deletion string_to_int64.go
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ func (s *stringToInt64Value) String() string {
}

func stringToInt64Conv(val string) (interface{}, error) {
val = strings.Trim(val, "[]")
val = strings.TrimSuffix(strings.TrimPrefix(val, "["), "]")
// An empty string would cause an empty map
if len(val) == 0 {
return map[string]int64{}, nil
Expand Down
19 changes: 19 additions & 0 deletions string_to_int64_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,25 @@ import (
"testing"
)

func TestGetStringToInt64PreservesBracketKeys(t *testing.T) {
for _, key := range []string{"[key]", "[[key", "]key", "[]", "key"} {
t.Run(key, func(t *testing.T) {
f := NewFlagSet("test", ContinueOnError)
value := f.StringToInt64("map", nil, "")
if err := f.Parse([]string{"--map=" + key + "=42"}); err != nil {
t.Fatal(err)
}
got, err := f.GetStringToInt64("map")
if err != nil {
t.Fatal(err)
}
if len(got) != 1 || got[key] != 42 || (*value)[key] != 42 {
t.Fatalf("getter = %v, bound value = %v; want key %q with value 42", got, *value, key)
}
})
}
}

func setUpS2I64FlagSet(s2ip *map[string]int64) *FlagSet {
f := NewFlagSet("test", ContinueOnError)
f.StringToInt64Var(s2ip, "s2i", map[string]int64{}, "Command separated ls2it!")
Expand Down
19 changes: 19 additions & 0 deletions string_to_int_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,25 @@ func TestEmptyS2I(t *testing.T) {
}
}

func TestGetStringToIntPreservesBracketKeys(t *testing.T) {
for _, key := range []string{"[key]", "[[key", "]key", "[]", "key"} {
t.Run(key, func(t *testing.T) {
f := NewFlagSet("test", ContinueOnError)
value := f.StringToInt("map", nil, "")
if err := f.Parse([]string{"--map=" + key + "=42"}); err != nil {
t.Fatal(err)
}
got, err := f.GetStringToInt("map")
if err != nil {
t.Fatal(err)
}
if len(got) != 1 || got[key] != 42 || (*value)[key] != 42 {
t.Fatalf("getter = %v, bound value = %v; want key %q with value 42", got, *value, key)
}
})
}
}

func TestS2I(t *testing.T) {
var s2i map[string]int
f := setUpS2IFlagSet(&s2i)
Expand Down
Loading