separate tests for component.setState and wrapper.setState

Author: koba04

packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx

@@ -4764,7 +4764,7 @@ describe('shallow', () => {
     });
 
     context('component instance', () => {
-      it('should call `componentDidUpdate` when component’s `setState` is called', () => {
+      it('should call `componentDidUpdate` when wrapper’s `setState` is called', () => {
         const spy = sinon.spy();
         class Foo extends React.Component {
           constructor(props) {
@@ -4792,9 +4792,36 @@ describe('shallow', () => {
         wrapper.setState({ foo: 'wrapper setState update' });
         expect(wrapper.state('foo')).to.equal('wrapper setState update');
         expect(spy).to.have.property('callCount', 1);
+      });
+
+      it('should call `componentDidUpdate` when component’s `setState` is called', () => {
+        const spy = sinon.spy();
+        class Foo extends React.Component {
+          constructor(props) {
+            super(props);
+            this.state = {
+              foo: 'init',
+            };
+          }
+
+          componentDidUpdate() {
+            spy();
+          }
+
+          onChange() {
+            // enzyme can't handle the update because `this` is a ReactComponent instance,
+            // not a ShallowWrapper instance.
+            this.setState({ foo: 'onChange update' });
+          }
+
+          render() {
+            return <div>{this.state.foo}</div>;
+          }
+        }
+        const wrapper = shallow(<Foo />);
         wrapper.instance().onChange();
         expect(wrapper.state('foo')).to.equal('onChange update');
-        expect(spy).to.have.property('callCount', 2);
+        expect(spy).to.have.property('callCount', 1);
       });
 
       it('should call `componentDidUpdate` when component’s `setState` is called through a bound method', () => {