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

@@ -222,6 +222,23 @@ function updateModuleForToNestedModule(j, root) {
   bodyPath.replace(bodyReplacement);
 }
 
+function updateLookupCalls(j, root) {
+  return root
+    .find(j.MemberExpression)
+    .filter(path => {
+      return (
+        path.value.property.name === 'lookup' &&
+        j.MemberExpression.check(path.value.object) &&
+        path.value.object.property.name === 'container' &&
+        j.ThisExpression.check(path.value.object.object)
+      );
+    })
+    .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;
 
@@ -237,6 +254,7 @@ module.exports = function(file, api, options) {
   moveQUnitImportsFromEmberQUnit(j, root);
   updateToNewEmberQUnitImports(j, root);
   updateModuleForToNestedModule(j, root);
+  updateLookupCalls(j, root);
 
   return root.toSource(printOptions);
 };