Fix bug in await using transformer

Perryvw · Perryvw · commit 9d9f1a9a86b9 · 2024-08-31T18:26:53.000+02:00
diff --git a/src/transformation/pre-transformers/using-transformer.ts b/src/transformation/pre-transformers/using-transformer.ts
@@ -18,7 +18,7 @@ export function usingTransformer(context: TransformationContext): ts.Transformer
                         ts.setParent(node2, parent[parent.length - 1]);
                         parent.push(node2);
                         ts.visitEachChild(node2, setParent, ctx);
-                        parent.push();
+                        parent.pop();
                         return node2;
                     }
                     ts.visitEachChild(updatedBlock, setParent, ctx);
@@ -74,7 +74,7 @@ function transformBlockWithUsing(
             );
 
             const callback = ts.factory.createFunctionExpression(
-                undefined,
+                [ts.factory.createModifier(ts.SyntaxKind.AsyncKeyword)],
                 undefined,
                 undefined,
                 undefined,
diff --git a/test/unit/using.spec.ts b/test/unit/using.spec.ts
@@ -175,3 +175,21 @@ test("await using can handle non-async disposables", () => {
         .setTsHeader(usingTestLib)
         .expectToEqual({ logs: ["Creating a", "function content", "Disposing a"] });
 });
+
+// https://github.com/TypeScriptToLua/TypeScriptToLua/issues/1571
+test.only("await using no extra diagnostics (#1571)", () => {
+    util.testModule`
+        async function getResource(): Promise<AsyncDisposable> {
+            return {
+                [Symbol.asyncDispose]: async () => {}
+            };
+        }
+
+        async function someOtherAsync() {}
+
+        async function main() {
+            await using resource = await getResource();
+            await someOtherAsync();
+        }
+    `.expectToHaveNoDiagnostics();
+});
