Rewrite this.render to await this.render.

Author: Robert Jackson

__testfixtures__/ember-qunit-codemod/module-for-component.output.js

@@ -6,7 +6,7 @@ module('Integration | Component | FooBar', function(hooks) {
   setupRenderingTest(hooks);
 
   test('it happens', async function() {
-    this.render(hbs`derp`);
+    await this.render(hbs`derp`);
   });
 });
 

ember-qunit-codemod.js

@@ -153,6 +153,34 @@ function updateModuleForToNestedModule(j, root) {
     return [moduleInvocation, callback.body.body, setupType];
   }
 
+  function processRenderingTest(expression) {
+    let isTest = j.match(expression, { expression: { callee: { name: 'test' } } });
+    if (!isTest) {
+      return;
+    }
+
+    // mark the test function as an async function
+    expression.expression.arguments[1].async = true;
+
+    let renderUsage = j(expression).find(j.ExpressionStatement, {
+      expression: {
+        callee: {
+          object: {
+            type: 'ThisExpression',
+          },
+          property: {
+            name: 'render',
+          },
+        },
+      },
+    });
+
+    renderUsage.forEach(p => {
+      let expression = p.get('expression');
+      expression.replace(j.awaitExpression(expression.node));
+    });
+  }
+
   let programPath = root.get('program');
   let bodyPath = programPath.get('body');
 
@@ -169,10 +197,7 @@ function updateModuleForToNestedModule(j, root) {
       currentModuleCallbackBody.push(expression);
 
       if (currentTestType === 'setupRenderingTest') {
-        let isTest = j.match(expression, { expression: { callee: { name: 'test' } } });
-        if (isTest) {
-          expression.expression.arguments[1].async = true;
-        }
+        processRenderingTest(expression);
       }
     } else {
       bodyReplacement.push(expression);