Fix custom context as a prop usage

reduxjs · Nov 6, 2018 · 2d06cdf · 2d06cdf
1 parent 58dcf33
commit 2d06cdf
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/components/connectAdvanced.js b/src/components/connectAdvanced.js
@@ -110,7 +110,6 @@ export default function connectAdvanced(
 
     if(pure) {
       OuterBaseComponent = PureComponent
-      //FinalWrappedComponent = React.memo(WrappedComponent)
     }
 
     class PureWrapper extends Component {
@@ -207,10 +206,12 @@ export default function connectAdvanced(
       }
 
       render() {
+        const ContextToUse = this.props.context || Context
+
         return (
-          <Context.Consumer>
+          <ContextToUse.Consumer>
             {this.renderWrappedComponent}
-          </Context.Consumer>
+          </ContextToUse.Consumer>
         )
       }
     }