Fix for-await printing (#5322)

Only the first for-await was correctly printed all subsequent for-await statements where printed as for-of as the variable op was changed inside the buildForXStatement
babel · Feb 15, 2017 · 28853bf · 28853bf
1 parent ff2c24e
commit 28853bf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/packages/babel-generator/src/generators/statements.js b/packages/babel-generator/src/generators/statements.js
@@ -87,13 +87,13 @@ const buildForXStatement = function (op) {
     if (op === "await") {
       this.word("await");
       this.space();
-      op = "of";
+      // do not attempt to change op here, as it will break subsequent for-await statements
     }
     this.token("(");
 
     this.print(node.left, node);
     this.space();
-    this.word(op);
+    this.word(op === "await" ? "of" : op);
     this.space();
     this.print(node.right, node);
     this.token(")");

diff --git a/...ges/babel-plugin-transform-object-rest-spread/test/fixtures/object-rest/for-x/expected.js b/...ges/babel-plugin-transform-object-rest-spread/test/fixtures/object-rest/for-x/expected.js
@@ -18,11 +18,11 @@ async function a() {
 for ({ a } in {}) {}
 for ({ a } of []) {}
 async function a() {
-  for ({ a } of []) {}
+  for await ({ a } of []) {}
 }
 
 for (a in {}) {}
 for (a of []) {}
 async function a() {
-  for (a of []) {}
+  for await (a of []) {}
 }