Also update react integration to likely fix #122

Author: developit

src/integrations/preact.js

@@ -64,8 +64,4 @@ export function connect(mapStateToProps, actions) {
 export function Provider(props) {
 	this.getChildContext = () => ({ store: props.store });
 }
-// Provider.prototype.getChildContext = function () {
-// 	return { store: this.props.store };
-// };
 Provider.prototype.render = props => props.children && props.children[0] || props.children;
-// Provider.prototype.render = props => Array.isArray(props.children) ? props.children[0] : props.children;

src/integrations/react.js

@@ -25,11 +25,11 @@ export function connect(mapStateToProps, actions) {
 	return Child => {
 		function Wrapper(props, context) {
 			Component.call(this, props, context);
-			let { store } = context;
+			const store = context.store;
 			let state = mapStateToProps(store ? store.getState() : {}, props);
-			let boundActions = actions ? mapActions(actions, store) : { store };
+			const boundActions = actions ? mapActions(actions, store) : { store };
 			let update = () => {
-				let mapped = mapStateToProps(store ? store.getState() : {}, this.props);
+				let mapped = mapStateToProps(store ? store.getState() : {}, props);
 				for (let i in mapped) if (mapped[i]!==state[i]) {
 					state = mapped;
 					return this.forceUpdate();
@@ -39,7 +39,10 @@ export function connect(mapStateToProps, actions) {
 					return this.forceUpdate();
 				}
 			};
-			this.componentWillReceiveProps = update;
+			this.componentWillReceiveProps = p => {
+				props = p;
+				update();
+			};
 			this.componentDidMount = () => {
 				store.subscribe(update);
 			};

test/react/unistore.test.js

@@ -240,7 +240,7 @@ describe('<Provider>', () => {
 		// await sleep(100);
 
 		expect(mapStateToProps).toHaveBeenCalledTimes(1);
-		expect(mapStateToProps).toHaveBeenCalledWith({}, {});
+		expect(mapStateToProps).toHaveBeenCalledWith({}, { a: 'b' });
 		// outer props were changed
 		expect(Child).toHaveBeenCalledWith(
 			{ mappings: 2, a: 'b', store },