Merge pull request #197 from NullVoxPopuli/fix-inline-andThen

Fix inline andThen transforms

Author: Turbo87

transforms/convert-module-for-to-setup-test/__testfixtures__/module-for-acceptance.input.js

@@ -60,6 +60,13 @@ test('it works with andThen', function() {
   find('#my-block');
 });
 
+test('it works with inline andThen', function() {
+  visit('my-route');
+  andThen(() => assert.ok(true));
+  andThen(() => assert.ok(false));
+  find('#my-block');
+});
+
 test('it works with es5 andThen', function() {
   visit('my-route');
   andThen(function() {

transforms/convert-module-for-to-setup-test/__testfixtures__/module-for-acceptance.output.js

@@ -59,6 +59,13 @@ module('Acceptance | AndThenRoute', function(hooks) {
     await find('#my-block');
   });
 
+  test('it works with inline andThen', async function() {
+    await visit('my-route');
+    assert.ok(true);
+    assert.ok(false);
+    await find('#my-block');
+  });
+
   test('it works with es5 andThen', async function() {
     await visit('my-route');
     assert.ok(true);

transforms/convert-module-for-to-setup-test/index.js

@@ -470,7 +470,8 @@ module.exports = function(file, api) {
         .map(path => path.parent)
         .replaceWith(p => {
           let body = p.node.expression.arguments[0].body;
-          return body.body;
+
+          return body.body || j.expressionStatement(body);
         });
 
       if (replacements.length > 0) {