Add withExtraArgument()

reduxjs · May 10, 2016 · 41aefcc · 41aefcc · QuotableWater7 · Sep 15, 2016
1 parent e0773ff
commit 41aefcc
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 3 deletions.
diff --git a/src/index.js b/src/index.js
@@ -1,9 +1,14 @@
-export default function thunkMiddleware({ dispatch, getState }) {
-  return next => action => {
+function createThunkMiddleware(extraArgument) {
+  return ({ dispatch, getState }) => next => action => {
     if (typeof action === 'function') {
-      return action(dispatch, getState);
+      return action(dispatch, getState, extraArgument);
     }
 
     return next(action);
   };
 }
+
+const thunk = createThunkMiddleware();
+thunk.withExtraArgument = createThunkMiddleware;
+
+export default thunk;
diff --git a/test/index.js b/test/index.js
@@ -76,4 +76,19 @@ describe('thunk middleware', () => {
       }
     });
   });
+
+  describe('withExtraArgument', () => {
+    it('must pass the third argument', done => {
+      const extraArg = { lol: true };
+      thunkMiddleware.withExtraArgument(extraArg)({
+        dispatch: doDispatch,
+        getState: doGetState,
+      })()((dispatch, getState, arg) => {
+        chai.assert.strictEqual(dispatch, doDispatch);
+        chai.assert.strictEqual(getState, doGetState);
+        chai.assert.strictEqual(arg, extraArg);
+        done();
+      });
+    });
+  });
 });