From d806ebd41f7d8807dd5d2d28327a71f7f07b6e5d Mon Sep 17 00:00:00 2001
From: Luca Osti <luca.d.osti@gmail.com>
Date: Fri, 22 Mar 2019 11:36:20 +0100
Subject: [PATCH] Add test

---
 golden_test.go | 60 ++++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 58 insertions(+), 2 deletions(-)

diff --git a/golden_test.go b/golden_test.go
index 021a05d..2fb1808 100644
--- a/golden_test.go
+++ b/golden_test.go
@@ -50,11 +50,15 @@ var goldenJSONAndSQL = []Golden{
 }
 
 var goldenTrimPrefix = []Golden{
-	{"trimprefix", trimPrefixIn, dayOut},
+	{"trim prefix", trimPrefixIn, dayOut},
 }
 
 var goldenWithPrefix = []Golden{
-	{"withprefix", dayIn, prefixedDayOut},
+	{"with prefix", dayIn, prefixedDayOut},
+}
+
+var goldenTrimAndAddPrefix = []Golden{
+	{"trim and add prefix", trimPrefixIn, trimmedPrefixedDayOut},
 }
 
 // Each example starts with "type XXX [u]int", with a single space separating them.
@@ -170,6 +174,55 @@ func (i Day) IsADay() bool {
 }
 `
 
+const trimmedPrefixedDayOut = `
+const _DayName = "NightMondayNightTuesdayNightWednesdayNightThursdayNightFridayNightSaturdayNightSunday"
+
+var _DayIndex = [...]uint8{0, 11, 23, 37, 50, 61, 74, 85}
+
+func (i Day) String() string {
+	if i < 0 || i >= Day(len(_DayIndex)-1) {
+		return fmt.Sprintf("Day(%d)", i)
+	}
+	return _DayName[_DayIndex[i]:_DayIndex[i+1]]
+}
+
+var _DayValues = []Day{0, 1, 2, 3, 4, 5, 6}
+
+var _DayNameToValueMap = map[string]Day{
+	_DayName[0:11]:  0,
+	_DayName[11:23]: 1,
+	_DayName[23:37]: 2,
+	_DayName[37:50]: 3,
+	_DayName[50:61]: 4,
+	_DayName[61:74]: 5,
+	_DayName[74:85]: 6,
+}
+
+// DayString retrieves an enum value from the enum constants string name.
+// Throws an error if the param is not part of the enum.
+func DayString(s string) (Day, error) {
+	if val, ok := _DayNameToValueMap[s]; ok {
+		return val, nil
+	}
+	return 0, fmt.Errorf("%s does not belong to Day values", s)
+}
+
+// DayValues returns all values of the enum
+func DayValues() []Day {
+	return _DayValues
+}
+
+// IsADay returns "true" if the value is listed in the enum definition. "false" otherwise
+func (i Day) IsADay() bool {
+	for _, v := range _DayValues {
+		if i == v {
+			return true
+		}
+	}
+	return false
+}
+`
+
 // Enumeration with an offset.
 // Also includes a duplicate.
 const offsetIn = `type Number int
@@ -1100,6 +1153,9 @@ func TestGolden(t *testing.T) {
 	for _, test := range goldenWithPrefix {
 		runGoldenTest(t, test, false, false, false, false, "", "Day")
 	}
+	for _, test := range goldenTrimAndAddPrefix {
+		runGoldenTest(t, test, false, false, false, false, "Day", "Night")
+	}
 }
 
 func runGoldenTest(t *testing.T, test Golden, generateJSON, generateYAML, generateSQL, generateText bool, trimPrefix string, prefix string) {