Support for NonNullExpression (#226)

Perryvw · web-flow · commit 89ae1d24b2a4 · 2018-10-11T20:09:46.000+02:00
diff --git a/src/Transpiler.ts b/src/Transpiler.ts
@@ -803,6 +803,8 @@ export abstract class LuaTranspiler {
                 return "";
             case ts.SyntaxKind.SpreadElement:
                 return this.transpileSpreadElement(node as ts.SpreadElement);
+            case ts.SyntaxKind.NonNullExpression:
+                return this.transpileExpression((node as ts.NonNullExpression).expression);
             case ts.SyntaxKind.Block:
                 this.pushIndent();
                 const ret = "do \n" + this.transpileBlock(node as ts.Block) + "end\n";
diff --git a/test/unit/expressions.spec.ts b/test/unit/expressions.spec.ts
@@ -290,6 +290,14 @@ export class ExpressionTests {
         const result = util.transpileAndExecute(`let a = 4; {let a = 42; } return a;`);
         Expect(result).toBe(4);
     }
+
+    @Test("Non-null expression")
+    public nonNullExpression(): void {
+        const result = util.transpileAndExecute(`function abc(): number | undefined { return 3; }
+            const a: number = abc()!;
+            return a;`);
+        Expect(result).toBe(3);
+    }
     // ====================================
     // Test expected errors
     // ====================================
