fix: handle await detection in method property (#53)

Menci · Aug 8, 2024 · 46992de · 46992de
1 parent 8006241
commit 46992de
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/src/find.spec.ts b/src/find.spec.ts
@@ -116,4 +116,17 @@ describe("Find top-level await usage in module", () => {
     `
     );
   });
+
+  it("should work with await in object method property", () => {
+    test(
+      null,
+      `
+      const obj = {
+        async method() {
+          await x.func();
+        }
+      }
+    `
+    );
+  });
 });
diff --git a/src/find.ts b/src/find.ts
@@ -33,6 +33,7 @@ class FindPatternsVisitor extends Visitor {
     hook("visitClass");
     hook("visitArrowFunctionExpression");
     hook("visitFunction");
+    hook("visitMethodProperty");
   }
 
   visitAwaitExpression(expr: SWC.AwaitExpression): SWC.Expression {