Add few basic tests

ark120202 · ark120202 · commit 9d20e413e06d · 2019-12-18T07:08:14.000Z
diff --git a/test/transpile/plugins/plugins.spec.ts b/test/transpile/plugins/plugins.spec.ts
@@ -0,0 +1,24 @@
+import * as path from "path";
+import * as util from "../../util";
+
+test("printer", () => {
+    util.testModule``
+        .setOptions({ luaPlugins: [{ plugin: path.join(__dirname, "printer.ts") }] })
+        .tap(builder => expect(builder.getMainLuaCodeChunk()).toMatch("Plugin"));
+});
+
+test("visitor", () => {
+    util.testFunction`
+        return false;
+    `
+        .setOptions({ luaPlugins: [{ plugin: path.join(__dirname, "visitor.ts") }] })
+        .expectToEqual(true);
+});
+
+test("visitor using super", () => {
+    util.testFunction`
+        return "foo";
+    `
+        .setOptions({ luaPlugins: [{ plugin: path.join(__dirname, "visitor-super.ts") }] })
+        .expectToEqual("bar");
+});
diff --git a/test/transpile/plugins/printer.ts b/test/transpile/plugins/printer.ts
@@ -0,0 +1,12 @@
+import * as tstl from "../../../src";
+
+const plugin: tstl.Plugin = {
+    printer: (program, emitHost, fileName, ...args) => {
+        const result = new tstl.LuaPrinter(program.getCompilerOptions(), emitHost, fileName).print(...args);
+        result.code = `-- Plugin\n${result.code}`;
+        return result;
+    },
+};
+
+// tslint:disable-next-line: no-default-export
+export default plugin;
diff --git a/test/transpile/plugins/visitor-super.ts b/test/transpile/plugins/visitor-super.ts
@@ -0,0 +1,19 @@
+import * as ts from "typescript";
+import * as tstl from "../../../src";
+
+const plugin: tstl.Plugin = {
+    visitors: {
+        [ts.SyntaxKind.StringLiteral]: (node, context) => {
+            const result = context.superTransformExpression(node);
+
+            if (tstl.isStringLiteral(result)) {
+                result.value = "bar";
+            }
+
+            return result;
+        },
+    },
+};
+
+// tslint:disable-next-line: no-default-export
+export default plugin;
diff --git a/test/transpile/plugins/visitor.ts b/test/transpile/plugins/visitor.ts
@@ -0,0 +1,11 @@
+import * as ts from "typescript";
+import * as tstl from "../../../src";
+
+const plugin: tstl.Plugin = {
+    visitors: {
+        [ts.SyntaxKind.ReturnStatement]: () => tstl.createReturnStatement([tstl.createBooleanLiteral(true)]),
+    },
+};
+
+// tslint:disable-next-line: no-default-export
+export default plugin;
diff --git a/test/transpile/transformers/transformers.spec.ts b/test/transpile/transformers/transformers.spec.ts
@@ -2,10 +2,6 @@ import * as path from "path";
 import * as tstl from "../../../src";
 import * as util from "../../util";
 
-const optionsOfTransformer = (transformer: tstl.TransformerImport): tstl.CompilerOptions => ({
-    plugins: [transformer],
-});
-
 test("should ignore language service plugins", () => {
     util.testFunction`
         return;
@@ -19,7 +15,7 @@ describe("resolution", () => {
         util.testFunction`
             return;
         `
-            .setOptions(optionsOfTransformer(transformer))
+            .setOptions({ plugins: [transformer] })
             .expectToEqual(true);
     };
 
@@ -42,25 +38,20 @@ describe("resolution", () => {
 
     test("error if transformer could not be resolved", () => {
         util.testModule``
-            .setOptions(optionsOfTransformer({ transform: path.join(__dirname, "error.ts") }))
+            .setOptions({ plugins: [{ transform: path.join(__dirname, "error.ts") }] })
             .expectToHaveDiagnostics();
     });
 });
 
 describe("factory types", () => {
     for (const type of ["program", "config", "checker", "raw", "compilerOptions"] as const) {
         test(type, () => {
-            const options = optionsOfTransformer({
-                transform: path.join(__dirname, "types.ts"),
-                type,
-                import: type,
-                value: true,
-            });
-
             util.testFunction`
                 return false;
             `
-                .setOptions(options)
+                .setOptions({
+                    plugins: [{ transform: path.join(__dirname, "types.ts"), type, import: type, value: true }],
+                })
                 .expectToEqual(true);
         });
     }
