aboutsummaryrefslogtreecommitdiff
path: root/flag_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'flag_test.go')
-rw-r--r--flag_test.go32
1 files changed, 32 insertions, 0 deletions
diff --git a/flag_test.go b/flag_test.go
index 3273775..411174a 100644
--- a/flag_test.go
+++ b/flag_test.go
@@ -203,3 +203,35 @@ func TestParseStringEnum(s *testing.T) {
t.AssertEqual("", val)
})
}
+
+func TestParseStringerEnum(s *testing.T) {
+ t := &core.T{T: s, Options: cmp.Options{fakeEnumComparer}}
+ parser := core.ParseStringerEnum(fakeEnumFoo, fakeEnumBar)
+
+ t.Run("Match", func(t *core.T) {
+ val, err := parser("FOO")
+ t.AssertErrorIs(nil, err)
+ t.AssertEqual(fakeEnumFoo, val)
+ })
+
+ t.Run("UnknownValue", func(t *core.T) {
+ val, err := parser("baz")
+ var exp core.UnknownEnumValueError[fakeEnum]
+ if t.AssertErrorAs(&exp, err) {
+ t.AssertEqual("baz", exp.Actual)
+ t.AssertEqual([]fakeEnum{fakeEnumFoo, fakeEnumBar}, exp.Expected)
+ }
+ t.AssertEqual(fakeEnum{}, val)
+ })
+}
+
+type fakeEnum struct{ string }
+
+var (
+ fakeEnumFoo = fakeEnum{"FOO"}
+ fakeEnumBar = fakeEnum{"BAR"}
+
+ fakeEnumComparer = cmp.Comparer(func(x, y fakeEnum) bool { return x == y })
+)
+
+func (e fakeEnum) String() string { return e.string }