Merge pull request #5 from rwjblue/migrate-this._element

Migrate `this._element` to `this.element`.

Author: rwjblue

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

@@ -9,9 +9,11 @@ test('it happens', function() {
   this.render(hbs`derp`);
 });
 
-test('it happens with comments', function() {
+test('it happens with comments', function(assert) {
   // comments above this.render are preserved
   this.render(hbs`derp`);
+
+  assert.equal(this._element.textContent, 'derp');
 });
 
 test('multiple renders', function() {

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

@@ -9,9 +9,11 @@ module('Integration | Component | FooBar', function(hooks) {
     await render(hbs`derp`);
   });
 
-  test('it happens with comments', async function() {
+  test('it happens with comments', async function(assert) {
     // comments above this.render are preserved
     await render(hbs`derp`);
+
+    assert.equal(this.element.textContent, 'derp');
   });
 
   test('multiple renders', async function() {

ember-qunit-codemod.js

@@ -177,18 +177,19 @@ function updateModuleForToNestedModule(j, root) {
     return [moduleInvocation, callback.body.body, setupType];
   }
 
-  function processRenderingTest(expression) {
-    let isTest = j.match(expression, { expression: { callee: { name: 'test' } } });
+  function processRenderingTest(testExpression) {
+    let isTest = j.match(testExpression, { expression: { callee: { name: 'test' } } });
     if (!isTest) {
       return;
     }
 
     // mark the test function as an async function
-    expression.expression.arguments[1].async = true;
+    testExpression.expression.arguments[1].async = true;
+    let testExpressionCollection = j(testExpression);
 
     // Transform to await render() or await clearRender()
     ['render', 'clearRender'].forEach(type => {
-      findTestHelperUsageOf(j, j(expression), type).forEach(p => {
+      findTestHelperUsageOf(j, testExpressionCollection, type).forEach(p => {
         let expression = p.get('expression');
 
         let awaitExpression = j.awaitExpression(
@@ -197,6 +198,21 @@ function updateModuleForToNestedModule(j, root) {
         expression.replace(awaitExpression);
       });
     });
+
+    // Migrate `this._element` -> `this.element`
+    testExpressionCollection
+      .find(j.MemberExpression, {
+        object: {
+          type: 'ThisExpression',
+        },
+        property: {
+          name: '_element',
+        },
+      })
+      .forEach(p => {
+        let property = p.get('property');
+        property.node.name = 'element';
+      });
   }
 
   let programPath = root.get('program');