[add] Intercepting response

goatslacker · May 26, 2015 · 6f074a3 · 6f074a3
1 parent cea29a4
commit 6f074a3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/src/alt/store/StoreMixin.js b/src/alt/store/StoreMixin.js
@@ -47,20 +47,21 @@ const StoreMixin = {
         const state = this.getInstance().getState()
         const value = spec.local && spec.local(state, ...args)
         const shouldFetch = spec.shouldFetch ? spec.shouldFetch(state, ...args) : !value
+        const intercept = spec.interceptResponse || (x => x)
 
         // if we don't have it in cache then fetch it
         if (shouldFetch) {
           loadCounter += 1
           /* istanbul ignore else */
-          if (spec.loading) spec.loading()
+          if (spec.loading) spec.loading(intercept(null, spec.loading))
           spec.remote(state, ...args)
             .then((v) => {
               loadCounter -= 1
-              spec.success(v)
+              spec.success(intercept(v, spec.success))
             })
             .catch((v) => {
               loadCounter -= 1
-              spec.error(v)
+              spec.error(intercept(v, spec.error))
             })
         } else {
           // otherwise emit the change now

diff --git a/test/async-test.js b/test/async-test.js
@@ -60,9 +60,12 @@ const StargazerSource = {
 
   fetchRepos: {
     remote() {
-      return new Promise((resolve, reject) => {
-        setTimeout(() => resolve('TESTTEST'), 200)
-      })
+      return Promise.resolve('batman')
+    },
+    interceptResponse(x, action) {
+      assert(x === 'batman')
+      assert(action === StargazerActions.usersReceived)
+      return 'TESTTEST'
     },
     success: StargazerActions.usersReceived,
     error: StargazerActions.failed