fix: for-of transform should skip for-await-of (#11023)

babel · Jan 16, 2020 · 06dace1 · 06dace1
1 parent 3daab41
commit 06dace1
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/packages/babel-plugin-transform-for-of/src/index.js b/packages/babel-plugin-transform-for-of/src/index.js
@@ -19,7 +19,10 @@ export default declare((api, options) => {
       visitor: {
         ForOfStatement(path) {
           const { scope } = path;
-          const { left, right, body } = path.node;
+          const { left, right, body, await: isAwait } = path.node;
+          if (isAwait) {
+            return;
+          }
           const i = scope.generateUidIdentifier("i");
           let array = scope.maybeGenerateMemoised(right, true);
 

diff --git a/packages/babel-plugin-transform-for-of/test/fixtures/for-of-as-array/for-await-of/input.js b/packages/babel-plugin-transform-for-of/test/fixtures/for-of-as-array/for-await-of/input.js
@@ -0,0 +1 @@
+async () => { for await (let foo of []); };
diff --git a/packages/babel-plugin-transform-for-of/test/fixtures/for-of-as-array/for-await-of/output.js b/packages/babel-plugin-transform-for-of/test/fixtures/for-of-as-array/for-await-of/output.js
@@ -0,0 +1,3 @@
+async () => {
+  for await (let foo of []);
+};