Merge pull request #930 from haihappen/patch-1

Improves Writing Tests doc.

Author: omnidan

docs/recipes/WritingTests.md

@@ -113,12 +113,12 @@ const middlewares = [thunk];
 /**
  * Creates a mock of Redux store with middleware.
  */
-function mockStore(getState, expectedActions, onLastAction) {
+function mockStore(getState, expectedActions, done) {
   if (!Array.isArray(expectedActions)) {
     throw new Error('expectedActions should be an array of expected actions.');
   }
-  if (typeof onLastAction !== 'undefined' && typeof onLastAction !== 'function') {
-    throw new Error('onLastAction should either be undefined or function.');
+  if (typeof done !== 'undefined' && typeof done !== 'function') {
+    throw new Error('done should either be undefined or function.');
   }
 
   function mockStoreWithoutMiddleware() {
@@ -131,11 +131,16 @@ function mockStore(getState, expectedActions, onLastAction) {
 
       dispatch(action) {
         const expectedAction = expectedActions.shift();
-        expect(action).toEqual(expectedAction);
-        if (onLastAction && !expectedActions.length) {
-          onLastAction();
+        
+        try {
+          expect(action).toEqual(expectedAction);
+          if (done && !expectedActions.length) {
+            done();
+          }
+          return action;
+        } catch (e) {
+          done(e);
         }
-        return action;
       }
     }
   }