increase test coverage

2015-11-10 14:26:40 -08:00
parent a55f210c7f
commit 884252b90f
6 changed files with 23 additions and 17 deletions
--- a/packages/babel-plugin-transform-regenerator/src/visit.js
+++ b/packages/babel-plugin-transform-regenerator/src/visit.js
@@ -37,13 +37,7 @@ exports.visitor = {
        return;
      }

-      if (node.expression) {
-        // Transform expression lambdas into normal functions.
-        node.expression = false;
-        node.body = t.blockStatement([
-          t.returnStatement(node.body)
-        ]);
-      }
+      path.ensureBlock();

      if (node.async) {
        path.get("body").traverse(awaitVisitor);