Add support for chaining

ctimmerm · Apr 14, 2016 · 8cb6864 · 8cb6864
1 parent 3905e46
commit 8cb6864
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -70,6 +70,14 @@ mock.onGet(/\/users\/\d+/).reply(function(config) {
 });
 ```
 
+Chaining is also supported
+
+```js
+mock
+  .onGet('/users').reply(200, users)
+  .onGet('/posts').reply(200, posts);
+```
+
 Mocking any request to a given url
 
 ```js

diff --git a/index.js b/index.js
@@ -54,6 +54,7 @@ MockAdapter.prototype.onAny = function(matcher) {
       verbs.forEach(function(verb) {
         _this.matchers[verb].push(handler);
       });
+      return _this;
     }
   };
 };
@@ -71,6 +72,7 @@ verbs.forEach(function(method) {
     return {
       reply: function reply(code, response, headers) {
         _this.matchers[method].push([matcher, code, response, headers]);
+        return _this;
       }
     };
   };

diff --git a/test/mock_adapter_test.js b/test/mock_adapter_test.js
@@ -163,6 +163,16 @@ describe('MockAdapter', function() {
     expect(newInstance.defaults.adapter).to.equal(adapter);
   });
 
+  it('can chain calls to add mock handlers', function() {
+    mock
+      .onGet('/foo').reply(200)
+      .onAny('/bar').reply(404)
+      .onPost('/baz').reply(500);
+
+    expect(mock.matchers['get']).to.not.be.empty;
+    expect(mock.matchers['post']).to.not.be.empty;
+  });
+
   context('on the default instance', function() {
     afterEach(function() {
       axios.defaults.adapter = undefined;