fix: ensure stubService returns reference to the service

isleofcode · Jul 17, 2018 · 5924b6f · 5924b6f
1 parent f6f3721
commit 5924b6f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/addon-test-support/stub-service.js b/addon-test-support/stub-service.js
@@ -35,7 +35,7 @@ export default function stubService() {
     let [name] = arguments;
     let { owner } = getContext();
 
-    replace(owner, name);
+    return replace(owner, name);
   } else {
     throw new Error("Unexpected number of arguments");
   }

diff --git a/tests/unit/test-support/stub-service-test.js b/tests/unit/test-support/stub-service-test.js
@@ -18,14 +18,19 @@ module("Test Helpers | stub-service", function(hooks) {
 
   module("invoking without `hooks`", function(hooks) {
     hooks.beforeEach(function() {
-      stubService("to-stub");
+      this.service = stubService("to-stub");
     });
 
     test("it can replace a service", function(assert) {
       let service = this.owner.lookup("service:to-stub");
       service.method();
 
       assert.verify(service.method());
+      assert.equal(
+        service,
+        this.service,
+        "Returns a references to the stubbed service"
+      );
     });
   });
 });