Handle changes from dispatch inside componentDidMount (fixes #208)

Author: gaearon

src/components/createConnector.js

@@ -44,6 +44,7 @@ export default function createConnector(React) {
 
     componentDidMount() {
       this.unsubscribe = this.context.store.subscribe(::this.handleChange);
+      this.handleChange();
     }
 
     componentWillReceiveProps(nextProps) {
@@ -59,7 +60,9 @@ export default function createConnector(React) {
 
     handleChange(props = this.props) {
       const nextState = this.selectState(props, this.context);
-      this.setState(nextState);
+      if (!this.isSliceEqual(this.state.slice, nextState.slice)) {
+        this.setState(nextState);
+      }
     }
 
     selectState(props, context) {

test/components/Connector.spec.js

@@ -245,7 +245,7 @@ describe('React', () => {
         }
 
         render() {
-          return (<div>{this.props.string}</div>);
+          return <div>{this.props.string}</div>;
         }
       }
 
@@ -261,5 +261,35 @@ describe('React', () => {
 
       expect(() => renderToString(el)).toNotThrow();
     });
+
+    it('should handle dispatch inside componentDidMount', () => {
+      const store = createStore(stringBuilder);
+
+      class TestComp extends Component {
+        componentDidMount() {
+          store.dispatch({
+            type: 'APPEND',
+            body: 'a'
+          });
+        }
+
+        render() {
+          return <div>{this.props.string}</div>;
+        }
+      }
+
+      const tree = TestUtils.renderIntoDocument(
+        <Provider store={store}>
+          {() => (
+            <Connector select={string => ({ string })}>
+              {({ string }) => <TestComp string={string} />}
+            </Connector>
+          )}
+        </Provider>
+      );
+
+      const testComp = TestUtils.findRenderedComponentWithType(tree, TestComp);
+      expect(testComp.props.string).toBe('a');
+    });
   });
 });