TypeScriptToLua
diff --git a/‎test/unit/__snapshots__/loops.spec.ts.snap‎
Lines changed: 11 additions & 1 deletion b/‎test/unit/__snapshots__/loops.spec.ts.snap‎
Lines changed: 11 additions & 1 deletion
diff --git a/‎test/util.ts‎
Lines changed: 2 additions & 1 deletion b/‎test/util.ts‎
Lines changed: 2 additions & 1 deletion
@@ -1,3 +1,13 @@
 // Jest Snapshot v1, https://goo.gl/fbAQLP
 
-exports[`forin[Array] 1`] = `"main.ts(3,9): error TSTL: Iterating over arrays with 'for ... in' is not allowed."`;
+exports[`forin[Array]: code 1`] = `
+"local ____exports = {}
+function ____exports.__main(self)
+    local array = {}
+    for key in pairs(array) do
+    end
+end
+return ____exports"
+`;
+
+exports[`forin[Array]: diagnostics 1`] = `"main.ts(3,9): error TSTL: Iterating over arrays with 'for ... in' is not allowed."`;
@@ -371,7 +371,8 @@ export abstract class TestBuilder {
             { getCurrentDirectory: () => "", getCanonicalFileName: fileName => fileName, getNewLine: () => "\n" }
         );
 
-        expect(diagnosticMessages.trim()).toMatchSnapshot();
+        expect(this.getMainLuaCodeChunk()).toMatchSnapshot('code');
+        expect(diagnosticMessages.trim()).toMatchSnapshot('diagnostics');
 
         return this;
     }
Original file line number	Diff line number	Diff line change
`@@ -371,7 +371,8 @@ export abstract class TestBuilder {`
`371`	`371`	`{ getCurrentDirectory: () => "", getCanonicalFileName: fileName => fileName, getNewLine: () => "\n" }`
`372`	`372`	`);`
`373`	`373`
`374`		`- expect(diagnosticMessages.trim()).toMatchSnapshot();`
	`374`	`+ expect(this.getMainLuaCodeChunk()).toMatchSnapshot('code');`
	`375`	`+ expect(diagnosticMessages.trim()).toMatchSnapshot('diagnostics');`
`375`	`376`
`376`	`377`	`return this;`
`377`	`378`	`}`