Merge pull request #1433 from iushankin/allowableValues-fix

Fix for allowable values in the Mustache template

modules/swagger-codegen/src/test/java/io/swagger/codegen/java/jaxrs/AllowableValuesTest.java

@@ -0,0 +1,64 @@
+package io.swagger.codegen.java.jaxrs;
+
+import io.swagger.codegen.CodegenParameter;
+
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableMap;
+import com.samskivert.mustache.Mustache;
+import com.samskivert.mustache.Template;
+import org.apache.commons.io.IOUtils;
+import org.testng.Assert;
+import org.testng.annotations.DataProvider;
+import org.testng.annotations.Test;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.nio.charset.StandardCharsets;
+
+public class AllowableValuesTest {
+
+    private static final String TEMPLATE_FILE = "JavaJaxRS/allowableValues.mustache";
+    private static final String PROVIDER_NAME = "operations";
+
+    private static String loadClassResource(Class<?> cls, String name) throws IOException {
+        InputStream in = null;
+        try {
+            in = cls.getClassLoader().getResourceAsStream(name);
+            return IOUtils.toString(in, StandardCharsets.UTF_8);
+        } finally {
+            IOUtils.closeQuietly(in);
+        }
+    }
+
+    @DataProvider(name = PROVIDER_NAME)
+    private Object[][] resource() {
+        final CodegenParameter param1 = new CodegenParameter();
+        final CodegenParameter param2 = new CodegenParameter() {{
+            allowableValues = ImmutableMap.<String, Object>of("values", ImmutableList.of("item1", "item2", "item3"));
+        }};
+        final CodegenParameter param3 = new CodegenParameter() {{
+            allowableValues = ImmutableMap.<String, Object>of("min", 1, "max", 10);
+        }};
+        final CodegenParameter param4 = new CodegenParameter() {{
+            allowableValues = ImmutableMap.<String, Object>of("min", 1);
+        }};
+        final CodegenParameter param5 = new CodegenParameter() {{
+            allowableValues = ImmutableMap.<String, Object>of("max", 10);
+        }};
+
+        return new Object[][]{
+                {param1, ""},
+                {param2, "allowableValues=\"item1, item2, item3\""},
+                {param3, "allowableValues=\"range=[1, 10]\""},
+                {param4, "allowableValues=\"range=[1, infinity]\""},
+                {param5, "allowableValues=\"range=[-infinity, 10]\""},
+        };
+    }
+
+    @Test(dataProvider = PROVIDER_NAME)
+    public void annotationsTest(CodegenParameter parameter, String expected) throws IOException {
+        final Template template = Mustache.compiler().compile(loadClassResource(this.getClass(), TEMPLATE_FILE));
+
+        Assert.assertEquals(template.execute(parameter), expected);
+    }
+}