Style tweaks

Author: gaearon

src/components/createConnect.js

@@ -12,6 +12,8 @@ const defaultMergeProps = (stateProps, dispatchProps, parentProps) => ({
   ...dispatchProps
 });
 
+const resetValue = (_, key) => ({ [key]: undefined });
+
 function getDisplayName(Component) {
   return Component.displayName || Component.name || 'Component';
 }
@@ -132,11 +134,8 @@ export default function createConnect(React) {
           const nextState = this.computeNextState(props);
           if (!shallowEqual(nextState, this.state)) {
             this.setState({
-              ...Object.keys(this.state)
-                .reduce((r, k) => ({
-                  [k]: undefined
-                }), {}),
-              ...nextState,
+              ...Object.keys(this.state).reduce(resetValue, {}),
+              ...nextState
             });
           }
         }

test/components/connect.spec.js

@@ -182,10 +182,10 @@ describe('React', () => {
 
     it('should remove undefined props', () => {
       const store = createStore(() => ({}));
-      let props = {x: true};
+      let props = { x: true };
       let container;
 
-      @connect(()=>({}), ()=>({}))
+      @connect(() => ({}), () => ({}))
       class ConnectContainer extends Component {
         render() {
           return (