add support for enums

generate/util_test.go

@@ -0,0 +1,65 @@
+package generate
+
+import "testing"
+
+type test struct {
+	name string
+	in   string
+	out  string
+}
+
+func testStringFunc(t *testing.T, f func(string) string, tests []test) {
+	for _, test := range tests {
+		test := test
+		t.Run(test.name, func(t *testing.T) {
+			got := f(test.in)
+			if got != test.out {
+				t.Errorf("got %#v want %#v", got, test.out)
+			}
+		})
+	}
+}
+
+func TestLowerFirst(t *testing.T) {
+	tests := []test{
+		{"Empty", "", ""},
+		{"SingleLower", "l", "l"},
+		{"SingleUpper", "L", "l"},
+		{"SingleUnicodeLower", "ļ", "ļ"},
+		{"SingleUnicodeUpper", "Ļ", "ļ"},
+		{"LongerLower", "lasdf", "lasdf"},
+		{"LongerUpper", "Lasdf", "lasdf"},
+		{"LongerUnicodeLower", "ļasdf", "ļasdf"},
+		{"LongerUnicodeUpper", "Ļasdf", "ļasdf"},
+	}
+
+	testStringFunc(t, lowerFirst, tests)
+}
+
+func TestUpperFirst(t *testing.T) {
+	tests := []test{
+		{"Empty", "", ""},
+		{"SingleLower", "l", "L"},
+		{"SingleUpper", "L", "L"},
+		{"SingleUnicodeLower", "ļ", "Ļ"},
+		{"SingleUnicodeUpper", "Ļ", "Ļ"},
+		{"LongerLower", "lasdf", "Lasdf"},
+		{"LongerUpper", "Lasdf", "Lasdf"},
+		{"LongerUnicodeLower", "ļasdf", "Ļasdf"},
+		{"LongerUnicodeUpper", "Ļasdf", "Ļasdf"},
+	}
+
+	testStringFunc(t, upperFirst, tests)
+}
+
+func TestGoConstName(t *testing.T) {
+	tests := []test{
+		{"Empty", "", ""},
+		{"AllCaps", "ASDF", "asdf"},
+		{"AllCapsWithUnderscore", "ASDF_GH", "asdfGh"},
+		{"JustUnderscore", "_", "_"},
+		{"LeadingUnderscore", "_ASDF", "_asdf"},
+	}
+
+	testStringFunc(t, goConstName, tests)
+}