forked from mirror/enumer
1688 lines
38 KiB
Go
1688 lines
38 KiB
Go
// Copyright 2014 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
// This file contains simple golden tests for various examples.
|
|
// Besides validating the results when the implementation changes,
|
|
// it provides a way to look at the generated code without having
|
|
// to execute the print statements in one's head.
|
|
|
|
package main
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
// Golden represents a test case.
|
|
type Golden struct {
|
|
name string
|
|
input string // input; the package clause is provided when running the test.
|
|
output string // expected output.
|
|
}
|
|
|
|
var golden = []Golden{
|
|
{"day", dayIn, dayOut},
|
|
{"offset", offsetIn, offsetOut},
|
|
{"gap", gapIn, gapOut},
|
|
{"num", numIn, numOut},
|
|
{"unum", unumIn, unumOut},
|
|
{"prime", primeIn, primeOut},
|
|
}
|
|
|
|
var goldenJSON = []Golden{
|
|
{"prime", primeJsonIn, primeJsonOut},
|
|
}
|
|
var goldenText = []Golden{
|
|
{"prime", primeTextIn, primeTextOut},
|
|
}
|
|
|
|
var goldenYAML = []Golden{
|
|
{"prime", primeYamlIn, primeYamlOut},
|
|
}
|
|
|
|
var goldenSQL = []Golden{
|
|
{"prime", primeSqlIn, primeSqlOut},
|
|
}
|
|
|
|
var goldenJSONAndSQL = []Golden{
|
|
{"prime", primeJsonAndSqlIn, primeJsonAndSqlOut},
|
|
}
|
|
|
|
var goldenTrimPrefix = []Golden{
|
|
{"trim prefix", trimPrefixIn, dayOut},
|
|
}
|
|
|
|
var goldenTrimPrefixMultiple = []Golden{
|
|
{"trim multiple prefixes", trimPrefixMultipleIn, dayNightOut},
|
|
}
|
|
|
|
var goldenWithPrefix = []Golden{
|
|
{"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.
|
|
|
|
// Simple test: enumeration of type int starting at 0.
|
|
const dayIn = `type Day int
|
|
const (
|
|
Monday Day = iota
|
|
Tuesday
|
|
Wednesday
|
|
Thursday
|
|
Friday
|
|
Saturday
|
|
Sunday
|
|
)
|
|
`
|
|
|
|
const dayOut = `
|
|
const _DayName = "MondayTuesdayWednesdayThursdayFridaySaturdaySunday"
|
|
|
|
var _DayIndex = [...]uint8{0, 6, 13, 22, 30, 36, 44, 50}
|
|
|
|
const _DayLowerName = "mondaytuesdaywednesdaythursdayfridaysaturdaysunday"
|
|
|
|
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:6]: 0,
|
|
_DayLowerName[0:6]: 0,
|
|
_DayName[6:13]: 1,
|
|
_DayLowerName[6:13]: 1,
|
|
_DayName[13:22]: 2,
|
|
_DayLowerName[13:22]: 2,
|
|
_DayName[22:30]: 3,
|
|
_DayLowerName[22:30]: 3,
|
|
_DayName[30:36]: 4,
|
|
_DayLowerName[30:36]: 4,
|
|
_DayName[36:44]: 5,
|
|
_DayLowerName[36:44]: 5,
|
|
_DayName[44:50]: 6,
|
|
_DayLowerName[44:50]: 6,
|
|
}
|
|
|
|
var _DayNames = []string{
|
|
_DayName[0:6],
|
|
_DayName[6:13],
|
|
_DayName[13:22],
|
|
_DayName[22:30],
|
|
_DayName[30:36],
|
|
_DayName[36:44],
|
|
_DayName[44:50],
|
|
}
|
|
|
|
// 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
|
|
}
|
|
|
|
// DayStrings returns a slice of all String values of the enum
|
|
func DayStrings() []string {
|
|
strs := make([]string, len(_DayNames))
|
|
copy(strs, _DayNames)
|
|
return strs
|
|
}
|
|
|
|
// 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
|
|
}
|
|
`
|
|
|
|
const dayNightOut = `
|
|
const _DayName = "MondayTuesdayWednesdayThursdayFridaySaturdaySunday"
|
|
|
|
var _DayIndex = [...]uint8{0, 6, 13, 22, 30, 36, 44, 50}
|
|
|
|
const _DayLowerName = "mondaytuesdaywednesdaythursdayfridaysaturdaysunday"
|
|
|
|
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:6]: 0,
|
|
_DayLowerName[0:6]: 0,
|
|
_DayName[6:13]: 1,
|
|
_DayLowerName[6:13]: 1,
|
|
_DayName[13:22]: 2,
|
|
_DayLowerName[13:22]: 2,
|
|
_DayName[22:30]: 3,
|
|
_DayLowerName[22:30]: 3,
|
|
_DayName[30:36]: 4,
|
|
_DayLowerName[30:36]: 4,
|
|
_DayName[36:44]: 5,
|
|
_DayLowerName[36:44]: 5,
|
|
_DayName[44:50]: 6,
|
|
_DayLowerName[44:50]: 6,
|
|
}
|
|
|
|
var _DayNames = []string{
|
|
_DayName[0:6],
|
|
_DayName[6:13],
|
|
_DayName[13:22],
|
|
_DayName[22:30],
|
|
_DayName[30:36],
|
|
_DayName[36:44],
|
|
_DayName[44:50],
|
|
}
|
|
|
|
// 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
|
|
}
|
|
|
|
// DayStrings returns a slice of all String values of the enum
|
|
func DayStrings() []string {
|
|
strs := make([]string, len(_DayNames))
|
|
copy(strs, _DayNames)
|
|
return strs
|
|
}
|
|
|
|
// 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
|
|
}
|
|
`
|
|
|
|
const prefixedDayOut = `
|
|
const _DayName = "DayMondayDayTuesdayDayWednesdayDayThursdayDayFridayDaySaturdayDaySunday"
|
|
|
|
var _DayIndex = [...]uint8{0, 9, 19, 31, 42, 51, 62, 71}
|
|
|
|
const _DayLowerName = "daymondaydaytuesdaydaywednesdaydaythursdaydayfridaydaysaturdaydaysunday"
|
|
|
|
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:9]: 0,
|
|
_DayLowerName[0:9]: 0,
|
|
_DayName[9:19]: 1,
|
|
_DayLowerName[9:19]: 1,
|
|
_DayName[19:31]: 2,
|
|
_DayLowerName[19:31]: 2,
|
|
_DayName[31:42]: 3,
|
|
_DayLowerName[31:42]: 3,
|
|
_DayName[42:51]: 4,
|
|
_DayLowerName[42:51]: 4,
|
|
_DayName[51:62]: 5,
|
|
_DayLowerName[51:62]: 5,
|
|
_DayName[62:71]: 6,
|
|
_DayLowerName[62:71]: 6,
|
|
}
|
|
|
|
var _DayNames = []string{
|
|
_DayName[0:9],
|
|
_DayName[9:19],
|
|
_DayName[19:31],
|
|
_DayName[31:42],
|
|
_DayName[42:51],
|
|
_DayName[51:62],
|
|
_DayName[62:71],
|
|
}
|
|
|
|
// 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
|
|
}
|
|
|
|
// DayStrings returns a slice of all String values of the enum
|
|
func DayStrings() []string {
|
|
strs := make([]string, len(_DayNames))
|
|
copy(strs, _DayNames)
|
|
return strs
|
|
}
|
|
|
|
// 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
|
|
}
|
|
`
|
|
|
|
const trimmedPrefixedDayOut = `
|
|
const _DayName = "NightMondayNightTuesdayNightWednesdayNightThursdayNightFridayNightSaturdayNightSunday"
|
|
|
|
var _DayIndex = [...]uint8{0, 11, 23, 37, 50, 61, 74, 85}
|
|
|
|
const _DayLowerName = "nightmondaynighttuesdaynightwednesdaynightthursdaynightfridaynightsaturdaynightsunday"
|
|
|
|
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,
|
|
_DayLowerName[0:11]: 0,
|
|
_DayName[11:23]: 1,
|
|
_DayLowerName[11:23]: 1,
|
|
_DayName[23:37]: 2,
|
|
_DayLowerName[23:37]: 2,
|
|
_DayName[37:50]: 3,
|
|
_DayLowerName[37:50]: 3,
|
|
_DayName[50:61]: 4,
|
|
_DayLowerName[50:61]: 4,
|
|
_DayName[61:74]: 5,
|
|
_DayLowerName[61:74]: 5,
|
|
_DayName[74:85]: 6,
|
|
_DayLowerName[74:85]: 6,
|
|
}
|
|
|
|
var _DayNames = []string{
|
|
_DayName[0:11],
|
|
_DayName[11:23],
|
|
_DayName[23:37],
|
|
_DayName[37:50],
|
|
_DayName[50:61],
|
|
_DayName[61:74],
|
|
_DayName[74:85],
|
|
}
|
|
|
|
// 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
|
|
}
|
|
|
|
// DayStrings returns a slice of all String values of the enum
|
|
func DayStrings() []string {
|
|
strs := make([]string, len(_DayNames))
|
|
copy(strs, _DayNames)
|
|
return strs
|
|
}
|
|
|
|
// 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
|
|
const (
|
|
_ Number = iota
|
|
One
|
|
Two
|
|
Three
|
|
AnotherOne = One // Duplicate; note that AnotherOne doesn't appear below.
|
|
)
|
|
`
|
|
|
|
const offsetOut = `
|
|
const _NumberName = "OneTwoThree"
|
|
|
|
var _NumberIndex = [...]uint8{0, 3, 6, 11}
|
|
|
|
const _NumberLowerName = "onetwothree"
|
|
|
|
func (i Number) String() string {
|
|
i -= 1
|
|
if i < 0 || i >= Number(len(_NumberIndex)-1) {
|
|
return fmt.Sprintf("Number(%d)", i+1)
|
|
}
|
|
return _NumberName[_NumberIndex[i]:_NumberIndex[i+1]]
|
|
}
|
|
|
|
var _NumberValues = []Number{1, 2, 3}
|
|
|
|
var _NumberNameToValueMap = map[string]Number{
|
|
_NumberName[0:3]: 1,
|
|
_NumberLowerName[0:3]: 1,
|
|
_NumberName[3:6]: 2,
|
|
_NumberLowerName[3:6]: 2,
|
|
_NumberName[6:11]: 3,
|
|
_NumberLowerName[6:11]: 3,
|
|
}
|
|
|
|
var _NumberNames = []string{
|
|
_NumberName[0:3],
|
|
_NumberName[3:6],
|
|
_NumberName[6:11],
|
|
}
|
|
|
|
// NumberString retrieves an enum value from the enum constants string name.
|
|
// Throws an error if the param is not part of the enum.
|
|
func NumberString(s string) (Number, error) {
|
|
if val, ok := _NumberNameToValueMap[s]; ok {
|
|
return val, nil
|
|
}
|
|
return 0, fmt.Errorf("%s does not belong to Number values", s)
|
|
}
|
|
|
|
// NumberValues returns all values of the enum
|
|
func NumberValues() []Number {
|
|
return _NumberValues
|
|
}
|
|
|
|
// NumberStrings returns a slice of all String values of the enum
|
|
func NumberStrings() []string {
|
|
strs := make([]string, len(_NumberNames))
|
|
copy(strs, _NumberNames)
|
|
return strs
|
|
}
|
|
|
|
// IsANumber returns "true" if the value is listed in the enum definition. "false" otherwise
|
|
func (i Number) IsANumber() bool {
|
|
for _, v := range _NumberValues {
|
|
if i == v {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
`
|
|
|
|
// Gaps and an offset.
|
|
const gapIn = `type Gap int
|
|
const (
|
|
Two Gap = 2
|
|
Three Gap = 3
|
|
Five Gap = 5
|
|
Six Gap = 6
|
|
Seven Gap = 7
|
|
Eight Gap = 8
|
|
Nine Gap = 9
|
|
Eleven Gap = 11
|
|
)
|
|
`
|
|
|
|
const gapOut = `
|
|
const (
|
|
_GapName_0 = "TwoThree"
|
|
_GapLowerName_0 = "twothree"
|
|
_GapName_1 = "FiveSixSevenEightNine"
|
|
_GapLowerName_1 = "fivesixseveneightnine"
|
|
_GapName_2 = "Eleven"
|
|
_GapLowerName_2 = "eleven"
|
|
)
|
|
|
|
var (
|
|
_GapIndex_0 = [...]uint8{0, 3, 8}
|
|
_GapIndex_1 = [...]uint8{0, 4, 7, 12, 17, 21}
|
|
_GapIndex_2 = [...]uint8{0, 6}
|
|
)
|
|
|
|
func (i Gap) String() string {
|
|
switch {
|
|
case 2 <= i && i <= 3:
|
|
i -= 2
|
|
return _GapName_0[_GapIndex_0[i]:_GapIndex_0[i+1]]
|
|
case 5 <= i && i <= 9:
|
|
i -= 5
|
|
return _GapName_1[_GapIndex_1[i]:_GapIndex_1[i+1]]
|
|
case i == 11:
|
|
return _GapName_2
|
|
default:
|
|
return fmt.Sprintf("Gap(%d)", i)
|
|
}
|
|
}
|
|
|
|
var _GapValues = []Gap{2, 3, 5, 6, 7, 8, 9, 11}
|
|
|
|
var _GapNameToValueMap = map[string]Gap{
|
|
_GapName_0[0:3]: 2,
|
|
_GapLowerName_0[0:3]: 2,
|
|
_GapName_0[3:8]: 3,
|
|
_GapLowerName_0[3:8]: 3,
|
|
_GapName_1[0:4]: 5,
|
|
_GapLowerName_1[0:4]: 5,
|
|
_GapName_1[4:7]: 6,
|
|
_GapLowerName_1[4:7]: 6,
|
|
_GapName_1[7:12]: 7,
|
|
_GapLowerName_1[7:12]: 7,
|
|
_GapName_1[12:17]: 8,
|
|
_GapLowerName_1[12:17]: 8,
|
|
_GapName_1[17:21]: 9,
|
|
_GapLowerName_1[17:21]: 9,
|
|
_GapName_2[0:6]: 11,
|
|
_GapLowerName_2[0:6]: 11,
|
|
}
|
|
|
|
var _GapNames = []string{
|
|
_GapName_0[0:3],
|
|
_GapName_0[3:8],
|
|
_GapName_1[0:4],
|
|
_GapName_1[4:7],
|
|
_GapName_1[7:12],
|
|
_GapName_1[12:17],
|
|
_GapName_1[17:21],
|
|
_GapName_2[0:6],
|
|
}
|
|
|
|
// GapString retrieves an enum value from the enum constants string name.
|
|
// Throws an error if the param is not part of the enum.
|
|
func GapString(s string) (Gap, error) {
|
|
if val, ok := _GapNameToValueMap[s]; ok {
|
|
return val, nil
|
|
}
|
|
return 0, fmt.Errorf("%s does not belong to Gap values", s)
|
|
}
|
|
|
|
// GapValues returns all values of the enum
|
|
func GapValues() []Gap {
|
|
return _GapValues
|
|
}
|
|
|
|
// GapStrings returns a slice of all String values of the enum
|
|
func GapStrings() []string {
|
|
strs := make([]string, len(_GapNames))
|
|
copy(strs, _GapNames)
|
|
return strs
|
|
}
|
|
|
|
// IsAGap returns "true" if the value is listed in the enum definition. "false" otherwise
|
|
func (i Gap) IsAGap() bool {
|
|
for _, v := range _GapValues {
|
|
if i == v {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
`
|
|
|
|
// Signed integers spanning zero.
|
|
const numIn = `type Num int
|
|
const (
|
|
m_2 Num = -2 + iota
|
|
m_1
|
|
m0
|
|
m1
|
|
m2
|
|
)
|
|
`
|
|
|
|
const numOut = `
|
|
const _NumName = "m_2m_1m0m1m2"
|
|
|
|
var _NumIndex = [...]uint8{0, 3, 6, 8, 10, 12}
|
|
|
|
const _NumLowerName = "m_2m_1m0m1m2"
|
|
|
|
func (i Num) String() string {
|
|
i -= -2
|
|
if i < 0 || i >= Num(len(_NumIndex)-1) {
|
|
return fmt.Sprintf("Num(%d)", i+-2)
|
|
}
|
|
return _NumName[_NumIndex[i]:_NumIndex[i+1]]
|
|
}
|
|
|
|
var _NumValues = []Num{-2, -1, 0, 1, 2}
|
|
|
|
var _NumNameToValueMap = map[string]Num{
|
|
_NumName[0:3]: -2,
|
|
_NumLowerName[0:3]: -2,
|
|
_NumName[3:6]: -1,
|
|
_NumLowerName[3:6]: -1,
|
|
_NumName[6:8]: 0,
|
|
_NumLowerName[6:8]: 0,
|
|
_NumName[8:10]: 1,
|
|
_NumLowerName[8:10]: 1,
|
|
_NumName[10:12]: 2,
|
|
_NumLowerName[10:12]: 2,
|
|
}
|
|
|
|
var _NumNames = []string{
|
|
_NumName[0:3],
|
|
_NumName[3:6],
|
|
_NumName[6:8],
|
|
_NumName[8:10],
|
|
_NumName[10:12],
|
|
}
|
|
|
|
// NumString retrieves an enum value from the enum constants string name.
|
|
// Throws an error if the param is not part of the enum.
|
|
func NumString(s string) (Num, error) {
|
|
if val, ok := _NumNameToValueMap[s]; ok {
|
|
return val, nil
|
|
}
|
|
return 0, fmt.Errorf("%s does not belong to Num values", s)
|
|
}
|
|
|
|
// NumValues returns all values of the enum
|
|
func NumValues() []Num {
|
|
return _NumValues
|
|
}
|
|
|
|
// NumStrings returns a slice of all String values of the enum
|
|
func NumStrings() []string {
|
|
strs := make([]string, len(_NumNames))
|
|
copy(strs, _NumNames)
|
|
return strs
|
|
}
|
|
|
|
// IsANum returns "true" if the value is listed in the enum definition. "false" otherwise
|
|
func (i Num) IsANum() bool {
|
|
for _, v := range _NumValues {
|
|
if i == v {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
`
|
|
|
|
// Unsigned integers spanning zero.
|
|
const unumIn = `type Unum uint
|
|
const (
|
|
m_2 Unum = iota + 253
|
|
m_1
|
|
)
|
|
|
|
const (
|
|
m0 Unum = iota
|
|
m1
|
|
m2
|
|
)
|
|
`
|
|
|
|
const unumOut = `
|
|
const (
|
|
_UnumName_0 = "m0m1m2"
|
|
_UnumLowerName_0 = "m0m1m2"
|
|
_UnumName_1 = "m_2m_1"
|
|
_UnumLowerName_1 = "m_2m_1"
|
|
)
|
|
|
|
var (
|
|
_UnumIndex_0 = [...]uint8{0, 2, 4, 6}
|
|
_UnumIndex_1 = [...]uint8{0, 3, 6}
|
|
)
|
|
|
|
func (i Unum) String() string {
|
|
switch {
|
|
case 0 <= i && i <= 2:
|
|
return _UnumName_0[_UnumIndex_0[i]:_UnumIndex_0[i+1]]
|
|
case 253 <= i && i <= 254:
|
|
i -= 253
|
|
return _UnumName_1[_UnumIndex_1[i]:_UnumIndex_1[i+1]]
|
|
default:
|
|
return fmt.Sprintf("Unum(%d)", i)
|
|
}
|
|
}
|
|
|
|
var _UnumValues = []Unum{0, 1, 2, 253, 254}
|
|
|
|
var _UnumNameToValueMap = map[string]Unum{
|
|
_UnumName_0[0:2]: 0,
|
|
_UnumLowerName_0[0:2]: 0,
|
|
_UnumName_0[2:4]: 1,
|
|
_UnumLowerName_0[2:4]: 1,
|
|
_UnumName_0[4:6]: 2,
|
|
_UnumLowerName_0[4:6]: 2,
|
|
_UnumName_1[0:3]: 253,
|
|
_UnumLowerName_1[0:3]: 253,
|
|
_UnumName_1[3:6]: 254,
|
|
_UnumLowerName_1[3:6]: 254,
|
|
}
|
|
|
|
var _UnumNames = []string{
|
|
_UnumName_0[0:2],
|
|
_UnumName_0[2:4],
|
|
_UnumName_0[4:6],
|
|
_UnumName_1[0:3],
|
|
_UnumName_1[3:6],
|
|
}
|
|
|
|
// UnumString retrieves an enum value from the enum constants string name.
|
|
// Throws an error if the param is not part of the enum.
|
|
func UnumString(s string) (Unum, error) {
|
|
if val, ok := _UnumNameToValueMap[s]; ok {
|
|
return val, nil
|
|
}
|
|
return 0, fmt.Errorf("%s does not belong to Unum values", s)
|
|
}
|
|
|
|
// UnumValues returns all values of the enum
|
|
func UnumValues() []Unum {
|
|
return _UnumValues
|
|
}
|
|
|
|
// UnumStrings returns a slice of all String values of the enum
|
|
func UnumStrings() []string {
|
|
strs := make([]string, len(_UnumNames))
|
|
copy(strs, _UnumNames)
|
|
return strs
|
|
}
|
|
|
|
// IsAUnum returns "true" if the value is listed in the enum definition. "false" otherwise
|
|
func (i Unum) IsAUnum() bool {
|
|
for _, v := range _UnumValues {
|
|
if i == v {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
`
|
|
|
|
// Enough gaps to trigger a map implementation of the method.
|
|
// Also includes a duplicate to test that it doesn't cause problems
|
|
const primeIn = `type Prime int
|
|
const (
|
|
p2 Prime = 2
|
|
p3 Prime = 3
|
|
p5 Prime = 5
|
|
p7 Prime = 7
|
|
p77 Prime = 7 // Duplicate; note that p77 doesn't appear below.
|
|
p11 Prime = 11
|
|
p13 Prime = 13
|
|
p17 Prime = 17
|
|
p19 Prime = 19
|
|
p23 Prime = 23
|
|
p29 Prime = 29
|
|
p37 Prime = 31
|
|
p41 Prime = 41
|
|
p43 Prime = 43
|
|
)
|
|
`
|
|
|
|
const primeOut = `
|
|
const _PrimeName = "p2p3p5p7p11p13p17p19p23p29p37p41p43"
|
|
const _PrimeLowerName = "p2p3p5p7p11p13p17p19p23p29p37p41p43"
|
|
|
|
var _PrimeMap = map[Prime]string{
|
|
2: _PrimeName[0:2],
|
|
3: _PrimeName[2:4],
|
|
5: _PrimeName[4:6],
|
|
7: _PrimeName[6:8],
|
|
11: _PrimeName[8:11],
|
|
13: _PrimeName[11:14],
|
|
17: _PrimeName[14:17],
|
|
19: _PrimeName[17:20],
|
|
23: _PrimeName[20:23],
|
|
29: _PrimeName[23:26],
|
|
31: _PrimeName[26:29],
|
|
41: _PrimeName[29:32],
|
|
43: _PrimeName[32:35],
|
|
}
|
|
|
|
func (i Prime) String() string {
|
|
if str, ok := _PrimeMap[i]; ok {
|
|
return str
|
|
}
|
|
return fmt.Sprintf("Prime(%d)", i)
|
|
}
|
|
|
|
var _PrimeValues = []Prime{2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 41, 43}
|
|
|
|
var _PrimeNameToValueMap = map[string]Prime{
|
|
_PrimeName[0:2]: 2,
|
|
_PrimeLowerName[0:2]: 2,
|
|
_PrimeName[2:4]: 3,
|
|
_PrimeLowerName[2:4]: 3,
|
|
_PrimeName[4:6]: 5,
|
|
_PrimeLowerName[4:6]: 5,
|
|
_PrimeName[6:8]: 7,
|
|
_PrimeLowerName[6:8]: 7,
|
|
_PrimeName[8:11]: 11,
|
|
_PrimeLowerName[8:11]: 11,
|
|
_PrimeName[11:14]: 13,
|
|
_PrimeLowerName[11:14]: 13,
|
|
_PrimeName[14:17]: 17,
|
|
_PrimeLowerName[14:17]: 17,
|
|
_PrimeName[17:20]: 19,
|
|
_PrimeLowerName[17:20]: 19,
|
|
_PrimeName[20:23]: 23,
|
|
_PrimeLowerName[20:23]: 23,
|
|
_PrimeName[23:26]: 29,
|
|
_PrimeLowerName[23:26]: 29,
|
|
_PrimeName[26:29]: 31,
|
|
_PrimeLowerName[26:29]: 31,
|
|
_PrimeName[29:32]: 41,
|
|
_PrimeLowerName[29:32]: 41,
|
|
_PrimeName[32:35]: 43,
|
|
_PrimeLowerName[32:35]: 43,
|
|
}
|
|
|
|
var _PrimeNames = []string{
|
|
_PrimeName[0:2],
|
|
_PrimeName[2:4],
|
|
_PrimeName[4:6],
|
|
_PrimeName[6:8],
|
|
_PrimeName[8:11],
|
|
_PrimeName[11:14],
|
|
_PrimeName[14:17],
|
|
_PrimeName[17:20],
|
|
_PrimeName[20:23],
|
|
_PrimeName[23:26],
|
|
_PrimeName[26:29],
|
|
_PrimeName[29:32],
|
|
_PrimeName[32:35],
|
|
}
|
|
|
|
// PrimeString retrieves an enum value from the enum constants string name.
|
|
// Throws an error if the param is not part of the enum.
|
|
func PrimeString(s string) (Prime, error) {
|
|
if val, ok := _PrimeNameToValueMap[s]; ok {
|
|
return val, nil
|
|
}
|
|
return 0, fmt.Errorf("%s does not belong to Prime values", s)
|
|
}
|
|
|
|
// PrimeValues returns all values of the enum
|
|
func PrimeValues() []Prime {
|
|
return _PrimeValues
|
|
}
|
|
|
|
// PrimeStrings returns a slice of all String values of the enum
|
|
func PrimeStrings() []string {
|
|
strs := make([]string, len(_PrimeNames))
|
|
copy(strs, _PrimeNames)
|
|
return strs
|
|
}
|
|
|
|
// IsAPrime returns "true" if the value is listed in the enum definition. "false" otherwise
|
|
func (i Prime) IsAPrime() bool {
|
|
_, ok := _PrimeMap[i]
|
|
return ok
|
|
}
|
|
`
|
|
const primeJsonIn = `type Prime int
|
|
const (
|
|
p2 Prime = 2
|
|
p3 Prime = 3
|
|
p5 Prime = 5
|
|
p7 Prime = 7
|
|
p77 Prime = 7 // Duplicate; note that p77 doesn't appear below.
|
|
p11 Prime = 11
|
|
p13 Prime = 13
|
|
p17 Prime = 17
|
|
p19 Prime = 19
|
|
p23 Prime = 23
|
|
p29 Prime = 29
|
|
p37 Prime = 31
|
|
p41 Prime = 41
|
|
p43 Prime = 43
|
|
)
|
|
`
|
|
|
|
const primeJsonOut = `
|
|
const _PrimeName = "p2p3p5p7p11p13p17p19p23p29p37p41p43"
|
|
const _PrimeLowerName = "p2p3p5p7p11p13p17p19p23p29p37p41p43"
|
|
|
|
var _PrimeMap = map[Prime]string{
|
|
2: _PrimeName[0:2],
|
|
3: _PrimeName[2:4],
|
|
5: _PrimeName[4:6],
|
|
7: _PrimeName[6:8],
|
|
11: _PrimeName[8:11],
|
|
13: _PrimeName[11:14],
|
|
17: _PrimeName[14:17],
|
|
19: _PrimeName[17:20],
|
|
23: _PrimeName[20:23],
|
|
29: _PrimeName[23:26],
|
|
31: _PrimeName[26:29],
|
|
41: _PrimeName[29:32],
|
|
43: _PrimeName[32:35],
|
|
}
|
|
|
|
func (i Prime) String() string {
|
|
if str, ok := _PrimeMap[i]; ok {
|
|
return str
|
|
}
|
|
return fmt.Sprintf("Prime(%d)", i)
|
|
}
|
|
|
|
var _PrimeValues = []Prime{2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 41, 43}
|
|
|
|
var _PrimeNameToValueMap = map[string]Prime{
|
|
_PrimeName[0:2]: 2,
|
|
_PrimeLowerName[0:2]: 2,
|
|
_PrimeName[2:4]: 3,
|
|
_PrimeLowerName[2:4]: 3,
|
|
_PrimeName[4:6]: 5,
|
|
_PrimeLowerName[4:6]: 5,
|
|
_PrimeName[6:8]: 7,
|
|
_PrimeLowerName[6:8]: 7,
|
|
_PrimeName[8:11]: 11,
|
|
_PrimeLowerName[8:11]: 11,
|
|
_PrimeName[11:14]: 13,
|
|
_PrimeLowerName[11:14]: 13,
|
|
_PrimeName[14:17]: 17,
|
|
_PrimeLowerName[14:17]: 17,
|
|
_PrimeName[17:20]: 19,
|
|
_PrimeLowerName[17:20]: 19,
|
|
_PrimeName[20:23]: 23,
|
|
_PrimeLowerName[20:23]: 23,
|
|
_PrimeName[23:26]: 29,
|
|
_PrimeLowerName[23:26]: 29,
|
|
_PrimeName[26:29]: 31,
|
|
_PrimeLowerName[26:29]: 31,
|
|
_PrimeName[29:32]: 41,
|
|
_PrimeLowerName[29:32]: 41,
|
|
_PrimeName[32:35]: 43,
|
|
_PrimeLowerName[32:35]: 43,
|
|
}
|
|
|
|
var _PrimeNames = []string{
|
|
_PrimeName[0:2],
|
|
_PrimeName[2:4],
|
|
_PrimeName[4:6],
|
|
_PrimeName[6:8],
|
|
_PrimeName[8:11],
|
|
_PrimeName[11:14],
|
|
_PrimeName[14:17],
|
|
_PrimeName[17:20],
|
|
_PrimeName[20:23],
|
|
_PrimeName[23:26],
|
|
_PrimeName[26:29],
|
|
_PrimeName[29:32],
|
|
_PrimeName[32:35],
|
|
}
|
|
|
|
// PrimeString retrieves an enum value from the enum constants string name.
|
|
// Throws an error if the param is not part of the enum.
|
|
func PrimeString(s string) (Prime, error) {
|
|
if val, ok := _PrimeNameToValueMap[s]; ok {
|
|
return val, nil
|
|
}
|
|
return 0, fmt.Errorf("%s does not belong to Prime values", s)
|
|
}
|
|
|
|
// PrimeValues returns all values of the enum
|
|
func PrimeValues() []Prime {
|
|
return _PrimeValues
|
|
}
|
|
|
|
// PrimeStrings returns a slice of all String values of the enum
|
|
func PrimeStrings() []string {
|
|
strs := make([]string, len(_PrimeNames))
|
|
copy(strs, _PrimeNames)
|
|
return strs
|
|
}
|
|
|
|
// IsAPrime returns "true" if the value is listed in the enum definition. "false" otherwise
|
|
func (i Prime) IsAPrime() bool {
|
|
_, ok := _PrimeMap[i]
|
|
return ok
|
|
}
|
|
|
|
// MarshalJSON implements the json.Marshaler interface for Prime
|
|
func (i Prime) MarshalJSON() ([]byte, error) {
|
|
return json.Marshal(i.String())
|
|
}
|
|
|
|
// UnmarshalJSON implements the json.Unmarshaler interface for Prime
|
|
func (i *Prime) UnmarshalJSON(data []byte) error {
|
|
var s string
|
|
if err := json.Unmarshal(data, &s); err != nil {
|
|
return fmt.Errorf("Prime should be a string, got %s", data)
|
|
}
|
|
|
|
var err error
|
|
*i, err = PrimeString(s)
|
|
return err
|
|
}
|
|
`
|
|
|
|
const primeTextIn = `type Prime int
|
|
const (
|
|
p2 Prime = 2
|
|
p3 Prime = 3
|
|
p5 Prime = 5
|
|
p7 Prime = 7
|
|
p77 Prime = 7 // Duplicate; note that p77 doesn't appear below.
|
|
p11 Prime = 11
|
|
p13 Prime = 13
|
|
p17 Prime = 17
|
|
p19 Prime = 19
|
|
p23 Prime = 23
|
|
p29 Prime = 29
|
|
p37 Prime = 31
|
|
p41 Prime = 41
|
|
p43 Prime = 43
|
|
)
|
|
`
|
|
|
|
const primeTextOut = `
|
|
const _PrimeName = "p2p3p5p7p11p13p17p19p23p29p37p41p43"
|
|
const _PrimeLowerName = "p2p3p5p7p11p13p17p19p23p29p37p41p43"
|
|
|
|
var _PrimeMap = map[Prime]string{
|
|
2: _PrimeName[0:2],
|
|
3: _PrimeName[2:4],
|
|
5: _PrimeName[4:6],
|
|
7: _PrimeName[6:8],
|
|
11: _PrimeName[8:11],
|
|
13: _PrimeName[11:14],
|
|
17: _PrimeName[14:17],
|
|
19: _PrimeName[17:20],
|
|
23: _PrimeName[20:23],
|
|
29: _PrimeName[23:26],
|
|
31: _PrimeName[26:29],
|
|
41: _PrimeName[29:32],
|
|
43: _PrimeName[32:35],
|
|
}
|
|
|
|
func (i Prime) String() string {
|
|
if str, ok := _PrimeMap[i]; ok {
|
|
return str
|
|
}
|
|
return fmt.Sprintf("Prime(%d)", i)
|
|
}
|
|
|
|
var _PrimeValues = []Prime{2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 41, 43}
|
|
|
|
var _PrimeNameToValueMap = map[string]Prime{
|
|
_PrimeName[0:2]: 2,
|
|
_PrimeLowerName[0:2]: 2,
|
|
_PrimeName[2:4]: 3,
|
|
_PrimeLowerName[2:4]: 3,
|
|
_PrimeName[4:6]: 5,
|
|
_PrimeLowerName[4:6]: 5,
|
|
_PrimeName[6:8]: 7,
|
|
_PrimeLowerName[6:8]: 7,
|
|
_PrimeName[8:11]: 11,
|
|
_PrimeLowerName[8:11]: 11,
|
|
_PrimeName[11:14]: 13,
|
|
_PrimeLowerName[11:14]: 13,
|
|
_PrimeName[14:17]: 17,
|
|
_PrimeLowerName[14:17]: 17,
|
|
_PrimeName[17:20]: 19,
|
|
_PrimeLowerName[17:20]: 19,
|
|
_PrimeName[20:23]: 23,
|
|
_PrimeLowerName[20:23]: 23,
|
|
_PrimeName[23:26]: 29,
|
|
_PrimeLowerName[23:26]: 29,
|
|
_PrimeName[26:29]: 31,
|
|
_PrimeLowerName[26:29]: 31,
|
|
_PrimeName[29:32]: 41,
|
|
_PrimeLowerName[29:32]: 41,
|
|
_PrimeName[32:35]: 43,
|
|
_PrimeLowerName[32:35]: 43,
|
|
}
|
|
|
|
var _PrimeNames = []string{
|
|
_PrimeName[0:2],
|
|
_PrimeName[2:4],
|
|
_PrimeName[4:6],
|
|
_PrimeName[6:8],
|
|
_PrimeName[8:11],
|
|
_PrimeName[11:14],
|
|
_PrimeName[14:17],
|
|
_PrimeName[17:20],
|
|
_PrimeName[20:23],
|
|
_PrimeName[23:26],
|
|
_PrimeName[26:29],
|
|
_PrimeName[29:32],
|
|
_PrimeName[32:35],
|
|
}
|
|
|
|
// PrimeString retrieves an enum value from the enum constants string name.
|
|
// Throws an error if the param is not part of the enum.
|
|
func PrimeString(s string) (Prime, error) {
|
|
if val, ok := _PrimeNameToValueMap[s]; ok {
|
|
return val, nil
|
|
}
|
|
return 0, fmt.Errorf("%s does not belong to Prime values", s)
|
|
}
|
|
|
|
// PrimeValues returns all values of the enum
|
|
func PrimeValues() []Prime {
|
|
return _PrimeValues
|
|
}
|
|
|
|
// PrimeStrings returns a slice of all String values of the enum
|
|
func PrimeStrings() []string {
|
|
strs := make([]string, len(_PrimeNames))
|
|
copy(strs, _PrimeNames)
|
|
return strs
|
|
}
|
|
|
|
// IsAPrime returns "true" if the value is listed in the enum definition. "false" otherwise
|
|
func (i Prime) IsAPrime() bool {
|
|
_, ok := _PrimeMap[i]
|
|
return ok
|
|
}
|
|
|
|
// MarshalText implements the encoding.TextMarshaler interface for Prime
|
|
func (i Prime) MarshalText() ([]byte, error) {
|
|
return []byte(i.String()), nil
|
|
}
|
|
|
|
// UnmarshalText implements the encoding.TextUnmarshaler interface for Prime
|
|
func (i *Prime) UnmarshalText(text []byte) error {
|
|
var err error
|
|
*i, err = PrimeString(string(text))
|
|
return err
|
|
}
|
|
`
|
|
|
|
const primeYamlIn = `type Prime int
|
|
const (
|
|
p2 Prime = 2
|
|
p3 Prime = 3
|
|
p5 Prime = 5
|
|
p7 Prime = 7
|
|
p77 Prime = 7 // Duplicate; note that p77 doesn't appear below.
|
|
p11 Prime = 11
|
|
p13 Prime = 13
|
|
p17 Prime = 17
|
|
p19 Prime = 19
|
|
p23 Prime = 23
|
|
p29 Prime = 29
|
|
p37 Prime = 31
|
|
p41 Prime = 41
|
|
p43 Prime = 43
|
|
)
|
|
`
|
|
|
|
const primeYamlOut = `
|
|
const _PrimeName = "p2p3p5p7p11p13p17p19p23p29p37p41p43"
|
|
const _PrimeLowerName = "p2p3p5p7p11p13p17p19p23p29p37p41p43"
|
|
|
|
var _PrimeMap = map[Prime]string{
|
|
2: _PrimeName[0:2],
|
|
3: _PrimeName[2:4],
|
|
5: _PrimeName[4:6],
|
|
7: _PrimeName[6:8],
|
|
11: _PrimeName[8:11],
|
|
13: _PrimeName[11:14],
|
|
17: _PrimeName[14:17],
|
|
19: _PrimeName[17:20],
|
|
23: _PrimeName[20:23],
|
|
29: _PrimeName[23:26],
|
|
31: _PrimeName[26:29],
|
|
41: _PrimeName[29:32],
|
|
43: _PrimeName[32:35],
|
|
}
|
|
|
|
func (i Prime) String() string {
|
|
if str, ok := _PrimeMap[i]; ok {
|
|
return str
|
|
}
|
|
return fmt.Sprintf("Prime(%d)", i)
|
|
}
|
|
|
|
var _PrimeValues = []Prime{2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 41, 43}
|
|
|
|
var _PrimeNameToValueMap = map[string]Prime{
|
|
_PrimeName[0:2]: 2,
|
|
_PrimeLowerName[0:2]: 2,
|
|
_PrimeName[2:4]: 3,
|
|
_PrimeLowerName[2:4]: 3,
|
|
_PrimeName[4:6]: 5,
|
|
_PrimeLowerName[4:6]: 5,
|
|
_PrimeName[6:8]: 7,
|
|
_PrimeLowerName[6:8]: 7,
|
|
_PrimeName[8:11]: 11,
|
|
_PrimeLowerName[8:11]: 11,
|
|
_PrimeName[11:14]: 13,
|
|
_PrimeLowerName[11:14]: 13,
|
|
_PrimeName[14:17]: 17,
|
|
_PrimeLowerName[14:17]: 17,
|
|
_PrimeName[17:20]: 19,
|
|
_PrimeLowerName[17:20]: 19,
|
|
_PrimeName[20:23]: 23,
|
|
_PrimeLowerName[20:23]: 23,
|
|
_PrimeName[23:26]: 29,
|
|
_PrimeLowerName[23:26]: 29,
|
|
_PrimeName[26:29]: 31,
|
|
_PrimeLowerName[26:29]: 31,
|
|
_PrimeName[29:32]: 41,
|
|
_PrimeLowerName[29:32]: 41,
|
|
_PrimeName[32:35]: 43,
|
|
_PrimeLowerName[32:35]: 43,
|
|
}
|
|
|
|
var _PrimeNames = []string{
|
|
_PrimeName[0:2],
|
|
_PrimeName[2:4],
|
|
_PrimeName[4:6],
|
|
_PrimeName[6:8],
|
|
_PrimeName[8:11],
|
|
_PrimeName[11:14],
|
|
_PrimeName[14:17],
|
|
_PrimeName[17:20],
|
|
_PrimeName[20:23],
|
|
_PrimeName[23:26],
|
|
_PrimeName[26:29],
|
|
_PrimeName[29:32],
|
|
_PrimeName[32:35],
|
|
}
|
|
|
|
// PrimeString retrieves an enum value from the enum constants string name.
|
|
// Throws an error if the param is not part of the enum.
|
|
func PrimeString(s string) (Prime, error) {
|
|
if val, ok := _PrimeNameToValueMap[s]; ok {
|
|
return val, nil
|
|
}
|
|
return 0, fmt.Errorf("%s does not belong to Prime values", s)
|
|
}
|
|
|
|
// PrimeValues returns all values of the enum
|
|
func PrimeValues() []Prime {
|
|
return _PrimeValues
|
|
}
|
|
|
|
// PrimeStrings returns a slice of all String values of the enum
|
|
func PrimeStrings() []string {
|
|
strs := make([]string, len(_PrimeNames))
|
|
copy(strs, _PrimeNames)
|
|
return strs
|
|
}
|
|
|
|
// IsAPrime returns "true" if the value is listed in the enum definition. "false" otherwise
|
|
func (i Prime) IsAPrime() bool {
|
|
_, ok := _PrimeMap[i]
|
|
return ok
|
|
}
|
|
|
|
// MarshalYAML implements a YAML Marshaler for Prime
|
|
func (i Prime) MarshalYAML() (interface{}, error) {
|
|
return i.String(), nil
|
|
}
|
|
|
|
// UnmarshalYAML implements a YAML Unmarshaler for Prime
|
|
func (i *Prime) UnmarshalYAML(unmarshal func(interface{}) error) error {
|
|
var s string
|
|
if err := unmarshal(&s); err != nil {
|
|
return err
|
|
}
|
|
|
|
var err error
|
|
*i, err = PrimeString(s)
|
|
return err
|
|
}
|
|
`
|
|
|
|
const primeSqlIn = `type Prime int
|
|
const (
|
|
p2 Prime = 2
|
|
p3 Prime = 3
|
|
p5 Prime = 5
|
|
p7 Prime = 7
|
|
p77 Prime = 7 // Duplicate; note that p77 doesn't appear below.
|
|
p11 Prime = 11
|
|
p13 Prime = 13
|
|
p17 Prime = 17
|
|
p19 Prime = 19
|
|
p23 Prime = 23
|
|
p29 Prime = 29
|
|
p37 Prime = 31
|
|
p41 Prime = 41
|
|
p43 Prime = 43
|
|
)
|
|
`
|
|
|
|
const primeSqlOut = `
|
|
const _PrimeName = "p2p3p5p7p11p13p17p19p23p29p37p41p43"
|
|
const _PrimeLowerName = "p2p3p5p7p11p13p17p19p23p29p37p41p43"
|
|
|
|
var _PrimeMap = map[Prime]string{
|
|
2: _PrimeName[0:2],
|
|
3: _PrimeName[2:4],
|
|
5: _PrimeName[4:6],
|
|
7: _PrimeName[6:8],
|
|
11: _PrimeName[8:11],
|
|
13: _PrimeName[11:14],
|
|
17: _PrimeName[14:17],
|
|
19: _PrimeName[17:20],
|
|
23: _PrimeName[20:23],
|
|
29: _PrimeName[23:26],
|
|
31: _PrimeName[26:29],
|
|
41: _PrimeName[29:32],
|
|
43: _PrimeName[32:35],
|
|
}
|
|
|
|
func (i Prime) String() string {
|
|
if str, ok := _PrimeMap[i]; ok {
|
|
return str
|
|
}
|
|
return fmt.Sprintf("Prime(%d)", i)
|
|
}
|
|
|
|
var _PrimeValues = []Prime{2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 41, 43}
|
|
|
|
var _PrimeNameToValueMap = map[string]Prime{
|
|
_PrimeName[0:2]: 2,
|
|
_PrimeLowerName[0:2]: 2,
|
|
_PrimeName[2:4]: 3,
|
|
_PrimeLowerName[2:4]: 3,
|
|
_PrimeName[4:6]: 5,
|
|
_PrimeLowerName[4:6]: 5,
|
|
_PrimeName[6:8]: 7,
|
|
_PrimeLowerName[6:8]: 7,
|
|
_PrimeName[8:11]: 11,
|
|
_PrimeLowerName[8:11]: 11,
|
|
_PrimeName[11:14]: 13,
|
|
_PrimeLowerName[11:14]: 13,
|
|
_PrimeName[14:17]: 17,
|
|
_PrimeLowerName[14:17]: 17,
|
|
_PrimeName[17:20]: 19,
|
|
_PrimeLowerName[17:20]: 19,
|
|
_PrimeName[20:23]: 23,
|
|
_PrimeLowerName[20:23]: 23,
|
|
_PrimeName[23:26]: 29,
|
|
_PrimeLowerName[23:26]: 29,
|
|
_PrimeName[26:29]: 31,
|
|
_PrimeLowerName[26:29]: 31,
|
|
_PrimeName[29:32]: 41,
|
|
_PrimeLowerName[29:32]: 41,
|
|
_PrimeName[32:35]: 43,
|
|
_PrimeLowerName[32:35]: 43,
|
|
}
|
|
|
|
var _PrimeNames = []string{
|
|
_PrimeName[0:2],
|
|
_PrimeName[2:4],
|
|
_PrimeName[4:6],
|
|
_PrimeName[6:8],
|
|
_PrimeName[8:11],
|
|
_PrimeName[11:14],
|
|
_PrimeName[14:17],
|
|
_PrimeName[17:20],
|
|
_PrimeName[20:23],
|
|
_PrimeName[23:26],
|
|
_PrimeName[26:29],
|
|
_PrimeName[29:32],
|
|
_PrimeName[32:35],
|
|
}
|
|
|
|
// PrimeString retrieves an enum value from the enum constants string name.
|
|
// Throws an error if the param is not part of the enum.
|
|
func PrimeString(s string) (Prime, error) {
|
|
if val, ok := _PrimeNameToValueMap[s]; ok {
|
|
return val, nil
|
|
}
|
|
return 0, fmt.Errorf("%s does not belong to Prime values", s)
|
|
}
|
|
|
|
// PrimeValues returns all values of the enum
|
|
func PrimeValues() []Prime {
|
|
return _PrimeValues
|
|
}
|
|
|
|
// PrimeStrings returns a slice of all String values of the enum
|
|
func PrimeStrings() []string {
|
|
strs := make([]string, len(_PrimeNames))
|
|
copy(strs, _PrimeNames)
|
|
return strs
|
|
}
|
|
|
|
// IsAPrime returns "true" if the value is listed in the enum definition. "false" otherwise
|
|
func (i Prime) IsAPrime() bool {
|
|
_, ok := _PrimeMap[i]
|
|
return ok
|
|
}
|
|
|
|
func (i Prime) Value() (driver.Value, error) {
|
|
return i.String(), nil
|
|
}
|
|
|
|
func (i *Prime) Scan(value interface{}) error {
|
|
if value == nil {
|
|
return nil
|
|
}
|
|
|
|
str, ok := value.(string)
|
|
if !ok {
|
|
bytes, ok := value.([]byte)
|
|
if !ok {
|
|
return fmt.Errorf("value is not a byte slice")
|
|
}
|
|
|
|
str = string(bytes[:])
|
|
}
|
|
|
|
val, err := PrimeString(str)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
*i = val
|
|
return nil
|
|
}
|
|
`
|
|
|
|
const primeJsonAndSqlIn = `type Prime int
|
|
const (
|
|
p2 Prime = 2
|
|
p3 Prime = 3
|
|
p5 Prime = 5
|
|
p7 Prime = 7
|
|
p77 Prime = 7 // Duplicate; note that p77 doesn't appear below.
|
|
p11 Prime = 11
|
|
p13 Prime = 13
|
|
p17 Prime = 17
|
|
p19 Prime = 19
|
|
p23 Prime = 23
|
|
p29 Prime = 29
|
|
p37 Prime = 31
|
|
p41 Prime = 41
|
|
p43 Prime = 43
|
|
)
|
|
`
|
|
|
|
const primeJsonAndSqlOut = `
|
|
const _PrimeName = "p2p3p5p7p11p13p17p19p23p29p37p41p43"
|
|
const _PrimeLowerName = "p2p3p5p7p11p13p17p19p23p29p37p41p43"
|
|
|
|
var _PrimeMap = map[Prime]string{
|
|
2: _PrimeName[0:2],
|
|
3: _PrimeName[2:4],
|
|
5: _PrimeName[4:6],
|
|
7: _PrimeName[6:8],
|
|
11: _PrimeName[8:11],
|
|
13: _PrimeName[11:14],
|
|
17: _PrimeName[14:17],
|
|
19: _PrimeName[17:20],
|
|
23: _PrimeName[20:23],
|
|
29: _PrimeName[23:26],
|
|
31: _PrimeName[26:29],
|
|
41: _PrimeName[29:32],
|
|
43: _PrimeName[32:35],
|
|
}
|
|
|
|
func (i Prime) String() string {
|
|
if str, ok := _PrimeMap[i]; ok {
|
|
return str
|
|
}
|
|
return fmt.Sprintf("Prime(%d)", i)
|
|
}
|
|
|
|
var _PrimeValues = []Prime{2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 41, 43}
|
|
|
|
var _PrimeNameToValueMap = map[string]Prime{
|
|
_PrimeName[0:2]: 2,
|
|
_PrimeLowerName[0:2]: 2,
|
|
_PrimeName[2:4]: 3,
|
|
_PrimeLowerName[2:4]: 3,
|
|
_PrimeName[4:6]: 5,
|
|
_PrimeLowerName[4:6]: 5,
|
|
_PrimeName[6:8]: 7,
|
|
_PrimeLowerName[6:8]: 7,
|
|
_PrimeName[8:11]: 11,
|
|
_PrimeLowerName[8:11]: 11,
|
|
_PrimeName[11:14]: 13,
|
|
_PrimeLowerName[11:14]: 13,
|
|
_PrimeName[14:17]: 17,
|
|
_PrimeLowerName[14:17]: 17,
|
|
_PrimeName[17:20]: 19,
|
|
_PrimeLowerName[17:20]: 19,
|
|
_PrimeName[20:23]: 23,
|
|
_PrimeLowerName[20:23]: 23,
|
|
_PrimeName[23:26]: 29,
|
|
_PrimeLowerName[23:26]: 29,
|
|
_PrimeName[26:29]: 31,
|
|
_PrimeLowerName[26:29]: 31,
|
|
_PrimeName[29:32]: 41,
|
|
_PrimeLowerName[29:32]: 41,
|
|
_PrimeName[32:35]: 43,
|
|
_PrimeLowerName[32:35]: 43,
|
|
}
|
|
|
|
var _PrimeNames = []string{
|
|
_PrimeName[0:2],
|
|
_PrimeName[2:4],
|
|
_PrimeName[4:6],
|
|
_PrimeName[6:8],
|
|
_PrimeName[8:11],
|
|
_PrimeName[11:14],
|
|
_PrimeName[14:17],
|
|
_PrimeName[17:20],
|
|
_PrimeName[20:23],
|
|
_PrimeName[23:26],
|
|
_PrimeName[26:29],
|
|
_PrimeName[29:32],
|
|
_PrimeName[32:35],
|
|
}
|
|
|
|
// PrimeString retrieves an enum value from the enum constants string name.
|
|
// Throws an error if the param is not part of the enum.
|
|
func PrimeString(s string) (Prime, error) {
|
|
if val, ok := _PrimeNameToValueMap[s]; ok {
|
|
return val, nil
|
|
}
|
|
return 0, fmt.Errorf("%s does not belong to Prime values", s)
|
|
}
|
|
|
|
// PrimeValues returns all values of the enum
|
|
func PrimeValues() []Prime {
|
|
return _PrimeValues
|
|
}
|
|
|
|
// PrimeStrings returns a slice of all String values of the enum
|
|
func PrimeStrings() []string {
|
|
strs := make([]string, len(_PrimeNames))
|
|
copy(strs, _PrimeNames)
|
|
return strs
|
|
}
|
|
|
|
// IsAPrime returns "true" if the value is listed in the enum definition. "false" otherwise
|
|
func (i Prime) IsAPrime() bool {
|
|
_, ok := _PrimeMap[i]
|
|
return ok
|
|
}
|
|
|
|
// MarshalJSON implements the json.Marshaler interface for Prime
|
|
func (i Prime) MarshalJSON() ([]byte, error) {
|
|
return json.Marshal(i.String())
|
|
}
|
|
|
|
// UnmarshalJSON implements the json.Unmarshaler interface for Prime
|
|
func (i *Prime) UnmarshalJSON(data []byte) error {
|
|
var s string
|
|
if err := json.Unmarshal(data, &s); err != nil {
|
|
return fmt.Errorf("Prime should be a string, got %s", data)
|
|
}
|
|
|
|
var err error
|
|
*i, err = PrimeString(s)
|
|
return err
|
|
}
|
|
|
|
func (i Prime) Value() (driver.Value, error) {
|
|
return i.String(), nil
|
|
}
|
|
|
|
func (i *Prime) Scan(value interface{}) error {
|
|
if value == nil {
|
|
return nil
|
|
}
|
|
|
|
str, ok := value.(string)
|
|
if !ok {
|
|
bytes, ok := value.([]byte)
|
|
if !ok {
|
|
return fmt.Errorf("value is not a byte slice")
|
|
}
|
|
|
|
str = string(bytes[:])
|
|
}
|
|
|
|
val, err := PrimeString(str)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
*i = val
|
|
return nil
|
|
}
|
|
`
|
|
|
|
const trimPrefixIn = `type Day int
|
|
const (
|
|
DayMonday Day = iota
|
|
DayTuesday
|
|
DayWednesday
|
|
DayThursday
|
|
DayFriday
|
|
DaySaturday
|
|
DaySunday
|
|
)
|
|
`
|
|
|
|
const trimPrefixMultipleIn = `type Day int
|
|
const (
|
|
DayMonday Day = iota
|
|
NightTuesday
|
|
DayWednesday
|
|
NightThursday
|
|
DayFriday
|
|
NightSaturday
|
|
DaySunday
|
|
)
|
|
`
|
|
|
|
func TestGolden(t *testing.T) {
|
|
for _, test := range golden {
|
|
runGoldenTest(t, test, false, false, false, false, "", "")
|
|
}
|
|
for _, test := range goldenJSON {
|
|
runGoldenTest(t, test, true, false, false, false, "", "")
|
|
}
|
|
for _, test := range goldenText {
|
|
runGoldenTest(t, test, false, false, false, true, "", "")
|
|
}
|
|
for _, test := range goldenYAML {
|
|
runGoldenTest(t, test, false, true, false, false, "", "")
|
|
}
|
|
for _, test := range goldenSQL {
|
|
runGoldenTest(t, test, false, false, true, false, "", "")
|
|
}
|
|
for _, test := range goldenJSONAndSQL {
|
|
runGoldenTest(t, test, true, false, true, false, "", "")
|
|
}
|
|
for _, test := range goldenTrimPrefix {
|
|
runGoldenTest(t, test, false, false, false, false, "Day", "")
|
|
}
|
|
for _, test := range goldenTrimPrefixMultiple {
|
|
runGoldenTest(t, test, false, false, false, false, "Day,Night", "")
|
|
}
|
|
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) {
|
|
var g Generator
|
|
file := test.name + ".go"
|
|
input := "package test\n" + test.input
|
|
|
|
dir, err := ioutil.TempDir("", "stringer")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
defer func() {
|
|
err = os.RemoveAll(dir)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}()
|
|
|
|
absFile := filepath.Join(dir, file)
|
|
err = ioutil.WriteFile(absFile, []byte(input), 0644)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
g.parsePackage([]string{absFile}, nil)
|
|
// Extract the name and type of the constant from the first line.
|
|
tokens := strings.SplitN(test.input, " ", 3)
|
|
if len(tokens) != 3 {
|
|
t.Fatalf("%s: need type declaration on first line", test.name)
|
|
}
|
|
g.generate(tokens[1], generateJSON, generateYAML, generateSQL, generateText, "noop", trimPrefix, prefix, false)
|
|
got := string(g.format())
|
|
if got != test.output {
|
|
// Use this to help build a golden text when changes are needed
|
|
//goldenFile := fmt.Sprintf("./goldendata/%v-%v-%v-%v-%v-%v-%v-%v-%v-%v.golden", test.name, tokens[1], generateJSON, generateYAML, generateSQL, generateText, "noop", trimPrefix, prefix, false)
|
|
//err = ioutil.WriteFile(goldenFile, []byte(got), 0644)
|
|
//if err != nil {
|
|
// t.Error(err)
|
|
//}
|
|
t.Errorf("%s: got\n====\n%s====\nexpected\n====%s", test.name, got, test.output)
|
|
}
|
|
}
|