Also transform this.registry.register

Author: cibernox

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

@@ -3,15 +3,18 @@ import { moduleFor, test } from 'ember-qunit';
 moduleFor('service:foo', 'Unit | Service | Foo', {
   beforeEach() {
     this.register('service:thingy', thingy);
+    this.registry.register('service:thingy', thingy);
   }
 });
 
 test('it happens', function() {
   this.register('service:thingy', thingy);
+  this.registry.register('service:thingy', thingy);
 });
 
 moduleFor('service:bar', 'Unit | Service | Bar');
 
 test('it happens again?', function() {
   this.register('service:thingy', thingy);
+  this.registry.register('service:thingy', thingy);
 });

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

@@ -6,10 +6,12 @@ module('Unit | Service | Foo', function(hooks) {
 
   hooks.beforeEach(function() {
     this.owner.register('service:thingy', thingy);
+    this.owner.register('service:thingy', thingy);
   });
 
   test('it happens', function() {
     this.owner.register('service:thingy', thingy);
+    this.owner.register('service:thingy', thingy);
   });
 });
 
@@ -18,5 +20,6 @@ module('Unit | Service | Bar', function(hooks) {
 
   test('it happens again?', function() {
     this.owner.register('service:thingy', thingy);
+    this.owner.register('service:thingy', thingy);
   });
 });

ember-qunit-codemod.js

@@ -242,7 +242,7 @@ function updateModuleForToNestedModule(j, root) {
 }
 
 function updateLookupCalls(j, root) {
-  return root
+  root
     .find(j.MemberExpression, {
       object: {
         object: { type: 'ThisExpression' },
@@ -257,7 +257,20 @@ function updateLookupCalls(j, root) {
 }
 
 function updateRegisterCalls(j, root) {
-  return root
+  root
+    .find(j.MemberExpression, {
+      object: {
+        object: { type: 'ThisExpression' },
+        property: { name: 'registry' },
+      },
+      property: { name: 'register' },
+    })
+    .forEach(path => {
+      let thisDotOwner = j.memberExpression(j.thisExpression(), j.identifier('owner'));
+      path.replace(j.memberExpression(thisDotOwner, path.value.property));
+    });
+
+  root
     .find(j.MemberExpression, {
       object: { type: 'ThisExpression' },
       property: { name: 'register' },