Fix enum strings starting with numbers, unique enums in model.mustache

Author: aethanol

modules/swagger-codegen/src/main/java/io/swagger/codegen/languages/AbstractGoCodegen.java

@@ -486,13 +486,10 @@ public String toEnumVarName(String name, String datatype) {
         if ("int".equals(datatype) || "int32".equals(datatype) || "int64".equals(datatype)
                 || "uint".equals(datatype) || "uint32".equals(datatype) || "uint64".equals(datatype)
                 || "float".equals(datatype) || "float32".equals(datatype) || "float64".equals(datatype)) {
-            String varName = name;
+            String varName = "NUMBER_" + name;
             varName = varName.replaceAll("-", "MINUS_");
             varName = varName.replaceAll("\\+", "PLUS_");
             varName = varName.replaceAll("\\.", "_DOT_");
-            if (varName.matches("\\d.*")) {
-                return "_" + varName;
-            }
             return varName;
         }
 
@@ -506,7 +503,13 @@ public String toEnumVarName(String name, String datatype) {
         enumName = enumName.replaceFirst("^_", "");
         enumName = enumName.replaceFirst("_$", "");
 
-        if (isReservedWord(enumName) || enumName.matches("\\d.*")) { // reserved word or starts with number
+        // starts with a number
+        if (enumName.matches("\\d.*")) {
+            enumName = "_" + enumName;
+        }
+
+        // reserved word
+        if (isReservedWord(enumName)) {
             return escapeReservedWord(enumName);
         } else {
             return enumName;

modules/swagger-codegen/src/main/resources/go/model.mustache

@@ -11,7 +11,7 @@ type {{{classname}}} {{^format}}{{dataType}}{{/format}}{{#format}}{{{format}}}{{
 const (
 	{{#allowableValues}}
 	{{#enumVars}}
-	{{name}} {{{classname}}} = {{{value}}}
+	{{name}}_{{{classname}}} {{{classname}}} = {{{value}}}
 	{{/enumVars}}
 	{{/allowableValues}}
 ){{/isEnum}}{{^isEnum}}{{#description}}

modules/swagger-codegen/src/test/java/io/swagger/codegen/go/GoClientCodegenTest.java

@@ -1,5 +1,6 @@
 package io.swagger.codegen.go;
 
+import io.swagger.codegen.DefaultCodegen;
 import org.testng.Assert;
 import org.testng.annotations.Test;
 
@@ -37,4 +38,32 @@ public void testAdditionalPropertiesPutForConfigValues() throws Exception {
         Assert.assertEquals(codegen.isHideGenerationTimestamp(), false);
     }
 
+    @Test(description = "test enum variable names for reserved words")
+    public void testEnumReservedWord() throws Exception {
+        final DefaultCodegen codegen = new GoClientCodegen();
+        Assert.assertEquals(codegen.toEnumVarName("IF", null), "IF_");
+        Assert.assertEquals(codegen.toEnumVarName("const", null), "CONST_");
+        Assert.assertEquals(codegen.toEnumVarName("INTERFACE", null), "INTERFACE_");
+        // should not escape non-reserved
+        Assert.assertEquals(codegen.toEnumVarName("hello", null), "HELLO");
+    }
+
+    @Test(description = "test enum variable names for numbers")
+    public void testEnumNumber() throws Exception {
+        final DefaultCodegen codegen = new GoClientCodegen();
+        Assert.assertEquals(codegen.toEnumVarName("1", "int32"), "NUMBER_1");
+        Assert.assertEquals(codegen.toEnumVarName("-1", "int32"), "NUMBER_MINUS_1");
+        Assert.assertEquals(codegen.toEnumVarName("+1", "int32"), "NUMBER_PLUS_1");
+        Assert.assertEquals(codegen.toEnumVarName("1.1", "float64"), "NUMBER_1_DOT_1");
+        Assert.assertEquals(codegen.toEnumVarName("-1.2", "float64"), "NUMBER_MINUS_1_DOT_2");
+        Assert.assertEquals(codegen.toEnumVarName("+1.2", "float64"), "NUMBER_PLUS_1_DOT_2");
+    }
+
+    @Test(description = "test enum variable names for strings that start with a number")
+    public void testEnumStringNumber() throws Exception {
+        final DefaultCodegen codegen = new GoClientCodegen();
+        Assert.assertEquals(codegen.toEnumVarName("1", null), "_1");
+        Assert.assertEquals(codegen.toEnumVarName("1_SAMPLE", null), "_1_SAMPLE");
+    }
+
 }

modules/swagger-codegen/src/test/java/io/swagger/codegen/go/GoModelTest.java

@@ -4,21 +4,23 @@
 import io.swagger.codegen.CodegenProperty;
 import io.swagger.codegen.DefaultCodegen;
 import io.swagger.codegen.languages.GoClientCodegen;
+import io.swagger.codegen.languages.JavaClientCodegen;
+import io.swagger.codegen.languages.PhpClientCodegen;
 import io.swagger.models.ArrayModel;
 import io.swagger.models.Model;
 import io.swagger.models.ModelImpl;
-import io.swagger.models.properties.ArrayProperty;
-import io.swagger.models.properties.DateTimeProperty;
-import io.swagger.models.properties.LongProperty;
-import io.swagger.models.properties.MapProperty;
-import io.swagger.models.properties.RefProperty;
-import io.swagger.models.properties.StringProperty;
+import io.swagger.models.Swagger;
+import io.swagger.models.properties.*;
 
 import com.google.common.collect.Sets;
+import io.swagger.parser.SwaggerParser;
 import org.testng.Assert;
 import org.testng.annotations.DataProvider;
 import org.testng.annotations.Test;
 
+import java.util.Arrays;
+import java.util.HashMap;
+
 @SuppressWarnings("static-method")
 public class GoModelTest {
 
@@ -243,6 +245,28 @@ public void mapModelTest() {
         Assert.assertEquals(Sets.intersection(cm.imports, Sets.newHashSet("Children")).size(), 1);
     }
 
+    @Test(description = "convert a go model with an enum")
+    public void enumMdoelValueTest() {
+        final StringProperty enumProperty = new StringProperty();
+        enumProperty.setEnum(Arrays.asList("VALUE1", "VALUE2", "VALUE3"));
+        final ModelImpl model = new ModelImpl().property("name", enumProperty);
+
+        final DefaultCodegen codegen = new GoClientCodegen();
+        final CodegenModel cm = codegen.fromModel("sample", model);
+
+        Assert.assertEquals(cm.vars.size(), 1);
+
+        final CodegenProperty enumVar = cm.vars.get(0);
+        Assert.assertEquals(enumVar.baseName, "name");
+        Assert.assertEquals(enumVar.datatype, "string");
+        Assert.assertEquals(enumVar.datatypeWithEnum, "NAME");
+        Assert.assertEquals(enumVar.name, "Name");
+        Assert.assertEquals(enumVar.defaultValue, "null");
+        Assert.assertEquals(enumVar.baseType, "string");
+        Assert.assertTrue(enumVar.isEnum);
+    }
+
+
     @DataProvider(name = "modelNames")
     public static Object[][] primeNumbers() {
         return new Object[][] {