Avoid using container.register.

emberjs · Jul 21, 2015 · bc63036 · bc63036
1 parent 08b2e79
commit bc63036
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/lib/ember-test-helpers/test-module-for-component.js b/lib/ember-test-helpers/test-module-for-component.js
@@ -47,16 +47,16 @@ export default TestModule.extend({
   setupComponentUnitTest: function() {
     var _this = this;
     var resolver = getResolver();
-    var container = this.container;
     var context = this.context;
 
     var layoutName = 'template:components/' + this.componentName;
 
     var layout = resolver.resolve(layoutName);
 
+    var thingToRegisterWith = this.registry || this.container;
     if (layout) {
-      container.register(layoutName, layout);
-      container.injection(this.subjectName, 'layout', layoutName);
+      thingToRegisterWith.register(layoutName, layout);
+      thingToRegisterWith.injection(this.subjectName, 'layout', layoutName);
     }
 
     context.dispatcher = Ember.EventDispatcher.create();

diff --git a/lib/ember-test-helpers/test-module-for-model.js b/lib/ember-test-helpers/test-module-for-model.js
@@ -19,7 +19,9 @@ export default TestModule.extend({
     var adapterFactory = container.lookupFactory('adapter:application');
     if (!adapterFactory) {
       adapterFactory = DS.JSONAPIAdapter || DS.FixtureAdapter;
-      container.register('adapter:application', adapterFactory);
+
+      var thingToRegisterWith = this.registry || this.container;
+      thingToRegisterWith.register('adapter:application', adapterFactory);
     }
 
     callbacks.store = function(){

diff --git a/tests/test-module-test.js b/tests/test-module-test.js
@@ -69,7 +69,8 @@ moduleFor('component:x-foo', 'component:x-foo -- setup context', {
       name: 'Max'
     });
 
-    this.container.register('service:blah', Ember.Object.extend({
+    var thingToRegisterWith = this.registry || this.container;
+    thingToRegisterWith.register('service:blah', Ember.Object.extend({
       purpose: 'blabering'
     }));
   }