From 65256f42bc67678c0ff81462e1f950f9cac6d547 Mon Sep 17 00:00:00 2001
From: Esa-Matti Suuronen <esa-matti@suuronen.org>
Date: Mon, 28 Sep 2015 13:42:34 +0300
Subject: [PATCH 1/2] Add failing test for #118

---
 test/components/connect.spec.js | 59 +++++++++++++++++++++++++++++++++
 1 file changed, 59 insertions(+)

diff --git a/test/components/connect.spec.js b/test/components/connect.spec.js
index 30741853f..5deb23ab4 100644
--- a/test/components/connect.spec.js
+++ b/test/components/connect.spec.js
@@ -1148,6 +1148,65 @@ describe('React', () => {
       expect(target.props.statefulValue).toEqual(1);
     });
 
+    it('calls mapState and mapDispatch for impure components', () => {
+      const store = createStore(() => ({
+        foo: 'foo',
+        bar: 'bar'
+      }));
+
+      const mapStateSpy = expect.createSpy();
+      const mapDispatchSpy = expect.createSpy().andReturn({});
+
+      class ImpureComponent extends Component {
+        render() {
+          return <Passthrough statefulValue={this.props.value} />;
+        }
+      }
+
+      const decorator = connect(
+        (state, {storeGetter}) => {
+          mapStateSpy();
+          return {value: state[storeGetter.storeKey]};
+        },
+        mapDispatchSpy,
+        null,
+        { pure: false }
+      );
+      const Decorated = decorator(ImpureComponent);
+
+      class StatefulWrapper extends Component {
+        state = {
+          storeGetter: {storeKey: 'foo'}
+        };
+
+        render() {
+          return <Decorated storeGetter={this.state.storeGetter} />;
+        };
+      }
+
+      const tree = TestUtils.renderIntoDocument(
+        <ProviderMock store={store}>
+          <StatefulWrapper />
+        </ProviderMock>
+      );
+
+      const target = TestUtils.findRenderedComponentWithType(tree, Passthrough);
+      const wrapper = TestUtils.findRenderedComponentWithType(tree, StatefulWrapper);
+
+      expect(mapStateSpy.calls.length).toBe(2);
+      expect(mapDispatchSpy.calls.length).toBe(2);
+      expect(target.props.statefulValue).toEqual('foo');
+
+      // Impure update
+      const storeGetter = wrapper.state.storeGetter;
+      storeGetter.storeKey = 'bar';
+      wrapper.setState({ storeGetter });
+
+      expect(mapStateSpy.calls.length).toBe(3);
+      expect(mapDispatchSpy.calls.length).toBe(3);
+      expect(target.props.statefulValue).toEqual('bar');
+    });
+
     it('should pass state consistently to mapState', () => {
       const store = createStore(stringBuilder);
 

From 8bc405d89f78ff0d80831057a05bd21d795ecad2 Mon Sep 17 00:00:00 2001
From: Esa-Matti Suuronen <esa-matti@suuronen.org>
Date: Mon, 28 Sep 2015 13:42:55 +0300
Subject: [PATCH 2/2] Call map fn(s) for impure components

Closes #118
---
 src/components/createConnect.js | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/components/createConnect.js b/src/components/createConnect.js
index a22ada650..083cd9ba0 100644
--- a/src/components/createConnect.js
+++ b/src/components/createConnect.js
@@ -90,6 +90,8 @@ export default function createConnect(React) {
 
         shouldComponentUpdate(nextProps, nextState) {
           if (!pure) {
+            this.updateStateProps(nextProps);
+            this.updateDispatchProps(nextProps);
             this.updateState(nextProps);
             return true;
           }