Enum test

Perryvw · Perryvw · commit 2b85ad4ad048 · 2018-02-20T20:52:19.000+01:00
diff --git a/test/integration/decorators.spec.ts b/test/integration/decorators.spec.ts
@@ -0,0 +1,27 @@
+import { Expect, Test, TestCase } from "alsatian";
+
+import * as util from "../src/util";
+
+export class DecoratorTests {
+
+    @Test("RegularEnum")
+    public regularEnum() {
+        const lua = util.transpileFile("test/integration/testfiles/enum.ts");
+        Expect(lua).toBe(
+            "TestEnum={}\n"+
+            "TestEnum.val1=0\n"+
+            "TestEnum.val2=2\n"+
+            "TestEnum.val3=3"
+        );
+    }
+
+    @Test("MembersOnlyEnumDecorator")
+    public membersOnly() {
+        const lua = util.transpileFile("test/integration/testfiles/membersOnlyEnum.ts");
+        Expect(lua).toBe(
+            "val1=0\n"+
+            "val2=2\n"+
+            "val3=3"
+        );
+    }
+}
diff --git a/test/integration/testfiles/enum.ts b/test/integration/testfiles/enum.ts
@@ -0,0 +1,5 @@
+enum TestEnum {
+    val1 = 0,
+    val2 = 2,
+    val3
+}
diff --git a/test/integration/testfiles/membersOnlyEnum.ts b/test/integration/testfiles/membersOnlyEnum.ts
@@ -0,0 +1,6 @@
+/** !CompileMembersOnly */
+enum TestEnum {
+    val1 = 0,
+    val2 = 2,
+    val3
+}
diff --git a/test/src/util.ts b/test/src/util.ts
@@ -18,6 +18,18 @@ export function transpileString(str: string, dummyType: any): string {
     return result.trim();
 }
 
+export function transpileFile(path: string): string {
+    const program = ts.createProgram([path], {});
+    const checker = program.getTypeChecker();
+
+    // Output errors
+    const diagnostics = ts.getPreEmitDiagnostics(program).filter(diag => diag.code != 6054);
+    diagnostics.forEach(diagnostic => console.log(`${ts.flattenDiagnosticMessageText(diagnostic.messageText, '\n')}`));
+
+    const lua = LuaTranspiler.transpileSourceFile(program.getSourceFile(path), checker, true);
+    return lua.trim();
+}
+
 export function executeLua(lua: string, withLib = true): any {
     if (withLib) {
         lua = minimalTestLib + lua
