Merge pull request #3 from rwjblue/add-clear-render

Add support for `this.clearRender()` -> `await clearRender()`

Author: rwjblue

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

@@ -14,6 +14,17 @@ test('it happens with comments', function() {
   this.render(hbs`derp`);
 });
 
+test('multiple renders', function() {
+  this.render(hbs`lololol`);
+
+  assert.ok(this.$().text(), 'lololol');
+
+  this.clearRender();
+  this.render(hbs`other stuff`);
+
+  assert.ok(this.$().text(), 'other stuff');
+});
+
 moduleForComponent('foo-bar', 'Unit | Component | FooBar', {
   needs: [],
 });

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

@@ -1,5 +1,5 @@
 import { module, test } from 'qunit';
-import { setupRenderingTest, setupTest, render } from 'ember-qunit';
+import { setupRenderingTest, setupTest, render, clearRender } from 'ember-qunit';
 import hbs from 'htmlbars-inline-precompile';
 
 module('Integration | Component | FooBar', function(hooks) {
@@ -13,6 +13,17 @@ module('Integration | Component | FooBar', function(hooks) {
     // comments above this.render are preserved
     await render(hbs`derp`);
   });
+
+  test('multiple renders', async function() {
+    await render(hbs`lololol`);
+
+    assert.ok(this.$().text(), 'lololol');
+
+    await clearRender();
+    await render(hbs`other stuff`);
+
+    assert.ok(this.$().text(), 'other stuff');
+  });
 });
 
 module('Unit | Component | FooBar', function(hooks) {

ember-qunit-codemod.js

@@ -70,25 +70,27 @@ function updateToNewEmberQUnitImports(j, root) {
     // Remove all existing import specifiers
     .remove();
 
-  let renderUsages = findRenderUsages(j, root);
-  if (renderUsages.size() > 0) {
-    specifiers.add('render');
-  }
+  ['render', 'clearRender'].forEach(type => {
+    let usages = findTestHelperUsageOf(j, root, type);
+    if (usages.size() > 0) {
+      specifiers.add(type);
+    }
+  });
 
   emberQUnitImports
     .get('specifiers')
     .replace(Array.from(specifiers).map(s => j.importSpecifier(j.identifier(s))));
 }
 
-function findRenderUsages(j, root) {
+function findTestHelperUsageOf(j, root, property) {
   return root.find(j.ExpressionStatement, {
     expression: {
       callee: {
         object: {
           type: 'ThisExpression',
         },
         property: {
-          name: 'render',
+          name: property,
         },
       },
     },
@@ -184,15 +186,16 @@ function updateModuleForToNestedModule(j, root) {
     // mark the test function as an async function
     expression.expression.arguments[1].async = true;
 
-    let renderUsage = findRenderUsages(j, j(expression));
+    // Transform to await render() or await clearRender()
+    ['render', 'clearRender'].forEach(type => {
+      findTestHelperUsageOf(j, j(expression), type).forEach(p => {
+        let expression = p.get('expression');
 
-    renderUsage.forEach(p => {
-      let expression = p.get('expression');
-
-      let awaitExpression = j.awaitExpression(
-        j.callExpression(j.identifier('render'), expression.node.arguments)
-      );
-      expression.replace(awaitExpression);
+        let awaitExpression = j.awaitExpression(
+          j.callExpression(j.identifier(type), expression.node.arguments)
+        );
+        expression.replace(awaitExpression);
+      });
     });
   }