Tests

ark120202 · ark120202 · commit 754675392f68 · 2019-08-01T11:23:31.000+05:00
diff --git a/test/unit/enum.spec.ts b/test/unit/enum.spec.ts
@@ -9,21 +9,18 @@ const serializeEnum = (identifier: string) => `(() => {
     return mappedTestEnum;
 })()`;
 
-// TODO: Move to namespace tests?
-test("in a namespace", () => {
-    util.testModule`
-        namespace Test {
-            export enum TestEnum {
-                A,
-                B,
+describe("initializers", () => {
+    test("string", () => {
+        util.testFunction`
+            enum TestEnum {
+                A = "A",
+                B = "B",
             }
-        }
 
-        export const result = ${serializeEnum("Test.TestEnum")}
-    `.expectToMatchJsResult();
-});
+            return ${serializeEnum("TestEnum")}
+        `.expectToMatchJsResult();
+    });
 
-describe("initializers", () => {
     test("expression", () => {
         util.testFunction`
             const value = 6;
@@ -36,6 +33,18 @@ describe("initializers", () => {
         `.expectToMatchJsResult();
     });
 
+    test("expression with side effect", () => {
+        util.testFunction`
+            let value = 0;
+            enum TestEnum {
+                A = value++,
+                B = A,
+            }
+
+            return ${serializeEnum("TestEnum")}
+        `.expectToMatchJsResult();
+    });
+
     test("inference", () => {
         util.testFunction`
             enum TestEnum {
@@ -60,7 +69,7 @@ describe("initializers", () => {
         `.expectToMatchJsResult();
     });
 
-    test("other member reference", () => {
+    test("member reference", () => {
         util.testFunction`
             enum TestEnum {
                 A,
@@ -71,26 +80,38 @@ describe("initializers", () => {
             return ${serializeEnum("TestEnum")}
         `.expectToMatchJsResult();
     });
+
+    test("string literal member reference", () => {
+        util.testFunction`
+            enum TestEnum {
+                ["A"],
+                "B" = A,
+                C = B,
+            }
+
+            return ${serializeEnum("TestEnum")}
+        `.expectToMatchJsResult();
+    });
 });
 
 describe("const enum", () => {
     const expectToBeConst: util.TapCallback = builder =>
         expect(builder.getMainLuaCodeChunk()).not.toContain("TestEnum");
 
-    test.each(["", "declare"])("%s without initializer", () => {
-        util.testFunction`
-            const enum TestEnum {
+    test.each(["", "declare "])("%swithout initializer", modifier => {
+        util.testModule`
+            ${modifier} const enum TestEnum {
                 A,
                 B,
             }
 
-            return TestEnum.A;
+            export const A = TestEnum.A;
         `
             .tap(expectToBeConst)
             .expectToMatchJsResult();
     });
 
-    test("with initializer", () => {
+    test("with string initializer", () => {
         util.testFunction`
             const enum TestEnum {
                 A = "ONE",
@@ -129,15 +150,3 @@ test("enum toString", () => {
         return test.toString();`;
     expect(util.transpileAndExecute(code)).toBe(0);
 });
-
-test("enum concat", () => {
-    const code = `
-        enum TestEnum {
-            A,
-            B,
-            C,
-        }
-        let test = TestEnum.A;
-        return test + "_foobar";`;
-    expect(util.transpileAndExecute(code)).toBe("0_foobar");
-});
diff --git a/test/unit/namespaces.spec.ts b/test/unit/namespaces.spec.ts
@@ -119,3 +119,15 @@ test("`import =` on a namespace", () => {
         export const result = importedFunc();
     `.expectToMatchJsResult();
 });
+
+test("enum in a namespace", () => {
+    util.testModule`
+        namespace Test {
+            export enum TestEnum {
+                A,
+            }
+        }
+
+        export const result = Test.TestEnum.A;
+    `.expectToMatchJsResult();
+});
