Fix adding await to function arguments

Fixes #20

Author: simonihmig

__testfixtures__/acceptance/visit.input.js

@@ -21,3 +21,8 @@ test('visiting /bar', async function(assert) {
   await visit('/bar');
   assert.equal(currentURL(), '/bar');
 });
+
+test('visiting /bar', async function(assert) {
+  await assert.rejects(visit('/bar'));
+  assert.equal(currentURL(), '/bar');
+});

__testfixtures__/acceptance/visit.output.js

@@ -18,3 +18,8 @@ test('visiting /bar', async function(assert) {
   await visit('/bar');
   assert.equal(currentURL(), '/bar');
 });
+
+test('visiting /bar', async function(assert) {
+  await assert.rejects(visit('/bar'));
+  assert.equal(currentURL(), '/bar');
+});

lib/utils.js

@@ -500,7 +500,7 @@ function dropAndThen(j, root) {
 
 function makeAwait(j, collection) {
   collection
-    .filter(({ parent: { node } }) => !j.AwaitExpression.check(node))
+    .filter(({ parent: { node } }) => j.ExpressionStatement.check(node))
     .replaceWith(({ node }) => j.awaitExpression(node))
     .forEach((path) => {