Exclude enum test

vojtechhabarta · May 4, 2017 · eed856e · eed856e
1 parent 9ea922c
commit eed856e
Showing 1 changed file with 37 additions and 0 deletions.
diff --git a/typescript-generator-core/src/test/java/cz/habarta/typescript/generator/EnumTest.java b/typescript-generator-core/src/test/java/cz/habarta/typescript/generator/EnumTest.java
@@ -1,8 +1,10 @@
 
 package cz.habarta.typescript.generator;
 
+import com.fasterxml.jackson.annotation.JsonFormat;
 import com.fasterxml.jackson.annotation.JsonProperty;
 import com.fasterxml.jackson.annotation.JsonValue;
+import java.util.*;
 import static org.junit.Assert.*;
 import org.junit.Test;
 
@@ -104,6 +106,14 @@ public void testEmptyEnum() {
         assertEquals(expected, output);
     }
 
+    @Test
+    public void testExcludeObjectEnum() {
+        final Settings settings = TestUtils.settings();
+        settings.setExcludeFilter(Arrays.asList(StatusType.class.getName()), Arrays.<String>asList());
+        final String output = new TypeScriptGenerator(settings).generateTypeScript(Input.from(ClassWithObjectEnum.class, StatusType.class));
+        assertTrue(!output.contains("StatusType"));
+    }
+
     private static class AClass {
         public Direction direction;
     }
@@ -143,4 +153,31 @@ public Object getJsonValue() {
     enum EmptyEnum {
     }
 
+    static class ClassWithObjectEnum {
+        public StatusType status;
+        public List<Map<String, StatusType>> statuses;
+    }
+
+    @JsonFormat(shape = JsonFormat.Shape.OBJECT)
+    public enum StatusType {
+        GOOD(0, "Good"),
+        FULL(1, "Full");
+
+        private final int code;
+        private final String label;
+
+        private StatusType(int code, String label) {
+            this.label = label;
+            this.code = code;
+        }
+
+        public int getCode() {
+            return code;
+        }
+
+        public String getLabel() {
+            return label;
+        }
+    }
+
 }