diff --git a/src/packages/recompose/__tests__/mapPropsOnChange-test.js b/src/packages/recompose/__tests__/mapPropsOnChange-test.js index c47013ea..09276e47 100644 --- a/src/packages/recompose/__tests__/mapPropsOnChange-test.js +++ b/src/packages/recompose/__tests__/mapPropsOnChange-test.js @@ -33,6 +33,8 @@ describe('mapPropsOnChange()', () => { renderIntoDocument() expect(omit(spy.getProps(), ['updateStrings'])).to.eql({ + a: 'a', + b: 'b', c: 'c', foobar: 'ab', d: 'new' @@ -44,6 +46,8 @@ describe('mapPropsOnChange()', () => { expect(mapSpy.callCount).to.equal(1) expect(omit(spy.getProps(), ['updateStrings', 'updateFoobar'])).to.eql({ + a: 'a', + b: 'b', c: 'baz', foobar: 'ab', d: 'new' @@ -51,6 +55,8 @@ describe('mapPropsOnChange()', () => { spy.getProps().updateStrings(strings => ({ ...strings, a: 'foo', 'b': 'bar', d: 'old' })) expect(omit(spy.getProps(), ['updateStrings', 'updateFoobar'])).to.eql({ + a: 'foo', + b: 'bar', c: 'baz', foobar: 'foobar', d: 'new' diff --git a/src/packages/recompose/mapPropsOnChange.js b/src/packages/recompose/mapPropsOnChange.js index b6896c8d..9df2975b 100644 --- a/src/packages/recompose/mapPropsOnChange.js +++ b/src/packages/recompose/mapPropsOnChange.js @@ -1,6 +1,5 @@ import { Component } from 'react' import pick from 'lodash/pick' -import omit from 'lodash/omit' import shallowEqual from './shallowEqual' import createHelper from './createHelper' import createElement from './createElement' @@ -22,7 +21,7 @@ const mapPropsOnChange = (depdendentPropKeys, propsMapper) => BaseComponent => { render() { return createElement(BaseComponent, { - ...omit(this.props, depdendentPropKeys), + ...this.props, ...this.computedProps }) }