Merge pull request #74 from quaertym/patch-2

Update test-module-for-model.js
emberjs · Jul 6, 2015 · 4320bf0 · 4320bf0
2 parents f3f653a + 923767f
commit 4320bf0
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lib/ember-test-helpers/test-module-for-model.js b/lib/ember-test-helpers/test-module-for-model.js
@@ -23,16 +23,17 @@ export default TestModule.extend({
 
     callbacks.store = function(){
       var container = this.container;
-
-      return container.lookup('store:main');
+      var store = container.lookup('service:store') || container.lookup('store:main');
+      return store;
     };
 
     if (callbacks.subject === defaultSubject) {
       callbacks.subject = function(options) {
         var container = this.container;
 
         return Ember.run(function() {
-          return container.lookup('store:main').createRecord(modelName, options);
+          var store = container.lookup('service:store') || container.lookup('store:main');
+          return store.createRecord(modelName, options);
         });
       };
     }