Merge pull request #27 from rwjblue/get-owner-this

Ensure `getOwner(this)` usage is rewritten.

Author: rwjblue

__testfixtures__/ember-qunit-codemod/get-owner-this.input.js

@@ -0,0 +1,28 @@
+import { moduleFor, test } from 'ember-qunit';
+
+moduleFor('service:flash', 'Unit | Service | Flash', {
+  unit: true
+});
+
+test('can fix getOwner(this) usage in a test', function (assert) {
+  let owner = getOwner(this);
+});
+
+moduleFor('service:flash', 'Unit | Service | Flash', {
+  unit: true,
+  beforeEach() {
+    let owner = getOwner(this);
+  }
+});
+
+test('can use getOwner(this) in beforeEach', function (assert) {
+  // stuff
+});
+
+moduleFor('service:flash', 'Unit | Service | Flash', {
+  unit: true
+});
+
+test('can use Ember.getOwner(this) also', function (assert) {
+  let owner = Ember.getOwner(this);
+});

__testfixtures__/ember-qunit-codemod/get-owner-this.output.js

@@ -0,0 +1,30 @@
+import { module, test } from 'qunit';
+import { setupTest } from 'ember-qunit';
+
+module('Unit | Service | Flash', function(hooks) {
+  setupTest(hooks);
+
+  test('can fix getOwner(this) usage in a test', function (assert) {
+    let owner = this.owner;
+  });
+});
+
+module('Unit | Service | Flash', function(hooks) {
+  setupTest(hooks);
+
+  hooks.beforeEach(function() {
+    let owner = this.owner;
+  });
+
+  test('can use getOwner(this) in beforeEach', function (assert) {
+    // stuff
+  });
+});
+
+module('Unit | Service | Flash', function(hooks) {
+  setupTest(hooks);
+
+  test('can use Ember.getOwner(this) also', function (assert) {
+    let owner = this.owner;
+  });
+});

ember-qunit-codemod.js

@@ -452,6 +452,36 @@ module.exports = function(file, api, options) {
       });
   }
 
+  function updateGetOwnerThisUsage() {
+    let thisDotOwner = j.memberExpression(j.thisExpression(), j.identifier('owner'));
+
+    root
+      .find(j.CallExpression, {
+        callee: {
+          name: 'getOwner',
+        },
+      })
+      .forEach(path => {
+        path.replace(thisDotOwner);
+      });
+
+    root
+      .find(j.CallExpression, {
+        callee: {
+          type: 'MemberExpression',
+          object: {
+            name: 'Ember',
+          },
+          property: {
+            name: 'getOwner',
+          },
+        },
+      })
+      .forEach(path => {
+        path.replace(thisDotOwner);
+      });
+  }
+
   const printOptions = options.printOptions || { quote: 'single' };
 
   // Find `ember-qunit` imports
@@ -466,6 +496,7 @@ module.exports = function(file, api, options) {
   updateLookupCalls();
   updateRegisterCalls();
   updateInjectCalls();
+  updateGetOwnerThisUsage();
 
   return root.toSource(printOptions);
 };