add store context-preservation tests

Author: vhmth

src/components/connectAdvanced.js

@@ -106,6 +106,10 @@ export default function connectAdvanced(
           `or explicitly pass "${storeKey}" as a prop to "${displayName}".`
         )
 
+        // make sure `getState` is properly bound in order to avoid breaking
+        // custom store implementations that rely on the store's context
+        this.getState = this.store.getState.bind(this.store);
+
         this.initSelector()
         this.initSubscription()
       }
@@ -160,7 +164,7 @@ export default function connectAdvanced(
 
       initSelector() {
         const { dispatch } = this.store
-        let getState = this.store.getState.bind(this.store)
+        const { getState } = this;
         const sourceSelector = selectorFactory(dispatch, selectorFactoryOptions)
 
         // wrap the selector in an object that tracks its results between runs

test/components/connect.spec.js

@@ -25,6 +25,30 @@ describe('React', () => {
       }
     }
 
+    class ContextBoundStore {
+      constructor(reducer) {
+        this.reducer = reducer
+        this.listeners = []
+        this.state = undefined
+        this.dispatch({})
+      }
+
+      getState() {
+        return this.state
+      }
+
+      subscribe(listener) {
+        this.listeners.push(listener)
+        return (() => this.listeners.filter(l => l !== listener))
+      }
+
+      dispatch(action) {
+        this.state = this.reducer(this.getState(), action)
+        this.listeners.forEach(l => l())
+        return action
+      }
+    }
+
     ProviderMock.childContextTypes = {
       store: PropTypes.object.isRequired
     }
@@ -134,6 +158,30 @@ describe('React', () => {
       expect(stub.props.string).toBe('ab')
     })
 
+    it('should retain the store\'s context', () => {
+      const store = new ContextBoundStore(stringBuilder)
+
+      let Container = connect(
+        state => ({ string: state })
+      )(function Container(props) {
+        return <Passthrough {...props}/>
+      })
+
+      const spy = expect.spyOn(console, 'error')
+      const tree = TestUtils.renderIntoDocument(
+        <ProviderMock store={store}>
+          <Container />
+        </ProviderMock>
+      )
+      spy.destroy()
+      expect(spy.calls.length).toBe(0)
+
+      const stub = TestUtils.findRenderedComponentWithType(tree, Passthrough)
+      expect(stub.props.string).toBe('')
+      store.dispatch({ type: 'APPEND', body: 'a' })
+      expect(stub.props.string).toBe('a')
+    })
+
     it('should handle dispatches before componentDidMount', () => {
       const store = createStore(stringBuilder)