update of wrapDispatch to be backward compatible to v2 api (#117)

erikras · Oct 20, 2016 · baeb2e4 · baeb2e4
1 parent 4e87e0e
commit baeb2e4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/wrapDispatch.js b/src/wrapDispatch.js
@@ -4,7 +4,8 @@ export default function wrapDispatch(dispatch, reducerKey) {
   const wrappedDispatch = (action) => {
     let wrappedAction;
     if (typeof action === 'function') {
-      wrappedAction = (_, getState) => action(wrappedDispatch, getState);
+      wrappedAction = (globalDispatch, getState) =>
+        action(wrappedDispatch, getState, globalDispatch, reducerKey);
     } else if (typeof action === 'object') {
       wrappedAction = wrapAction(action, reducerKey);
     }

diff --git a/test/wrapDispatch.spec.js b/test/wrapDispatch.spec.js
@@ -16,7 +16,7 @@ describe('wrapDispatch', () => {
 
   it('dispatch thunk action', () => {
     const getState = () => {};
-    const dispatch = createSpy().andCall(action => action(null, getState));
+    const dispatch = createSpy().andCall(action => action(dispatch, getState));
     const action = createSpy();
     const wrappedDispatch = wrapDispatch(dispatch, 'additional');
     wrappedDispatch(action);
@@ -27,6 +27,8 @@ describe('wrapDispatch', () => {
     expect(action.calls[0].arguments).toEqual([
       wrappedDispatch,
       getState,
+      dispatch,
+      'additional'
     ]);
   });
 });