Add support for passing callback to stubRequest

Tobbe · Apr 14, 2017 · e75ce6b · e75ce6b
1 parent 62101e4
commit e75ce6b
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/index.js b/index.js
@@ -38,7 +38,11 @@ let mockAdapter = (config) => {
         if (stub.timeout) {
           throwTimeout(config)
         }
-        request.respondWith(stub.response)
+        if(typeof stub.response === 'function') {
+          request.respondWith(stub.response(request))
+        } else {
+          request.respondWith(stub.response)
+        }
         stub.resolve()
         break
       }

diff --git a/test.js b/test.js
@@ -131,6 +131,23 @@ describe('moxios', function () {
       })
     })
 
+    it('should stub requests with callback', function (done) {
+      moxios.stubRequest('/whoami', (request) => {
+        return {
+          status:   200,
+          response: request.url
+        }
+      })
+
+      axios.get('/whoami').then(onFulfilled)
+
+      moxios.wait(function () {
+        let response = onFulfilled.getCall(0).args[0]
+        deepEqual(response.data, '/whoami')
+        done()
+      })
+    })
+
     it('should stub timeout', function (done) {
       moxios.stubTimeout('/users/12345')