Fix yield expression transform (#9076)

babel · Nov 27, 2018 · 2bb24f9 · 2bb24f9
1 parent 61f2aed
commit 2bb24f9
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 1 deletion.
diff --git a/packages/babel-generator/src/node/parentheses.js b/packages/babel-generator/src/node/parentheses.js
@@ -166,6 +166,7 @@ export function YieldExpression(node: Object, parent: Object): boolean {
     t.isCallExpression(parent) ||
     t.isMemberExpression(parent) ||
     t.isNewExpression(parent) ||
+    (t.isAwaitExpression(parent) && t.isYieldExpression(node)) ||
     (t.isConditionalExpression(parent) && node === parent.test) ||
     isClassExtendsClause(node, parent)
   );

diff --git a/packages/babel-generator/test/fixtures/parentheses/yield-expression/input.js b/packages/babel-generator/test/fixtures/parentheses/yield-expression/input.js
@@ -10,3 +10,7 @@ function* asdf() {
 function* a(b) {
   (yield xhr({ url: "views/test.html" })).data;
 }
+
+(async function* () {
+  await (yield 1);
+});
diff --git a/packages/babel-generator/test/fixtures/parentheses/yield-expression/output.js b/packages/babel-generator/test/fixtures/parentheses/yield-expression/output.js
@@ -11,4 +11,8 @@ function* a(b) {
   (yield xhr({
     url: "views/test.html"
   })).data;
-}
+}
+
+(async function* () {
+  await (yield 1);
+});