Merge pull request #4 from cibernox/transform-container-lookup

Transform container lookup into owner lookup

Author: cibernox

__testfixtures__/ember-qunit-codemod/lookup.input.js

@@ -0,0 +1,17 @@
+import { moduleFor, test } from 'ember-qunit';
+
+moduleFor('service:foo', 'Unit | Service | Foo', {
+  beforeEach() {
+    let service = this.container.lookup('service:thingy');
+  }
+});
+
+test('it happens', function() {
+  this.container.lookup('service:thingy').doSomething();
+});
+
+moduleFor('service:bar', 'Unit | Service | Bar');
+
+test('it happens again?', function() {
+  this.container.lookup('service:thingy').doSomething();
+});

__testfixtures__/ember-qunit-codemod/lookup.output.js

@@ -0,0 +1,22 @@
+import { module, test } from 'qunit';
+import { setupTest } from 'ember-qunit';
+
+module('Unit | Service | Foo', function(hooks) {
+  setupTest(hooks);
+
+  hooks.beforeEach(function() {
+    let service = this.owner.lookup('service:thingy');
+  });
+
+  test('it happens', function() {
+    this.owner.lookup('service:thingy').doSomething();
+  });
+});
+
+module('Unit | Service | Bar', function(hooks) {
+  setupTest(hooks);
+
+  test('it happens again?', function() {
+    this.owner.lookup('service:thingy').doSomething();
+  });
+});

ember-qunit-codemod.js

@@ -241,6 +241,21 @@ function updateModuleForToNestedModule(j, root) {
   bodyPath.replace(bodyReplacement);
 }
 
+function updateLookupCalls(j, root) {
+  return root
+    .find(j.MemberExpression, {
+      object: {
+        object: { type: 'ThisExpression' },
+        property: { name: 'container' },
+      },
+      property: { name: 'lookup' },
+    })
+    .forEach(path => {
+      let thisDotOwner = j.memberExpression(j.thisExpression(), j.identifier('owner'));
+      path.replace(j.memberExpression(thisDotOwner, path.value.property));
+    });
+}
+
 module.exports = function(file, api, options) {
   const j = api.jscodeshift;
 
@@ -256,6 +271,7 @@ module.exports = function(file, api, options) {
   moveQUnitImportsFromEmberQUnit(j, root);
   updateToNewEmberQUnitImports(j, root);
   updateModuleForToNestedModule(j, root);
+  updateLookupCalls(j, root);
 
   return root.toSource(printOptions);
 };