[Fix] mount: simulateError: update afterwards

Author: jgzuke

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

@@ -5033,22 +5033,31 @@ describeWithDOM('mount', () => {
         class ErrorBoundary extends React.Component {
           constructor(...args) {
             super(...args);
-            this.state = { throws: false };
+            this.state = {
+              throws: false,
+              didThrow: false,
+            };
           }
 
           componentDidCatch(error, info) {
             const { spy } = this.props;
             spy(error, info);
-            this.setState({ throws: false });
+            this.setState({
+              throws: false,
+              didThrow: true,
+            });
           }
 
           render() {
-            const { throws } = this.state;
+            const { throws, didThrow } = this.state;
             return (
               <div>
                 <MaybeFragment>
                   <span>
                     <Thrower throws={throws} />
+                    <main>
+                      {didThrow ? 'HasThrown' : 'HasNotThrown'}
+                    </main>
                   </span>
                 </MaybeFragment>
               </div>
@@ -5096,6 +5105,16 @@ describeWithDOM('mount', () => {
           });
         });
 
+        it('rerenders on a simulated error', () => {
+          const wrapper = mount(<ErrorBoundary spy={sinon.stub()} />);
+
+          expect(wrapper.find('main').text()).to.equal('HasNotThrown');
+
+          expect(() => wrapper.find(Thrower).simulateError(errorToThrow)).not.to.throw();
+
+          expect(wrapper.find('main').text()).to.equal('HasThrown');
+        });
+
         it('catches errors during render', () => {
           const spy = sinon.spy();
           const wrapper = mount(<ErrorBoundary spy={spy} />);

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

@@ -5067,22 +5067,31 @@ describe('shallow', () => {
         class ErrorBoundary extends React.Component {
           constructor(...args) {
             super(...args);
-            this.state = { throws: false };
+            this.state = {
+              throws: false,
+              didThrow: false,
+            };
           }
 
           componentDidCatch(error, info) {
             const { spy } = this.props;
             spy(error, info);
-            this.setState({ throws: false });
+            this.setState({
+              throws: false,
+              didThrow: true,
+            });
           }
 
           render() {
-            const { throws } = this.state;
+            const { throws, didThrow } = this.state;
             return (
               <div>
                 <MaybeFragment>
                   <span>
                     <Thrower throws={throws} />
+                    <main>
+                      {didThrow ? 'HasThrown' : 'HasNotThrown'}
+                    </main>
                   </span>
                 </MaybeFragment>
               </div>
@@ -5124,6 +5133,16 @@ describe('shallow', () => {
           });
         });
 
+        it('rerenders on a simulated error', () => {
+          const wrapper = shallow(<ErrorBoundary spy={sinon.stub()} />);
+
+          expect(wrapper.find('main').text()).to.equal('HasNotThrown');
+
+          expect(() => wrapper.find(Thrower).simulateError(errorToThrow)).not.to.throw();
+
+          expect(wrapper.find('main').text()).to.equal('HasThrown');
+        });
+
         it('does not catch errors during shallow render', () => {
           const spy = sinon.spy();
           const wrapper = shallow(<ErrorBoundary spy={spy} />);
@@ -5141,6 +5160,8 @@ describe('shallow', () => {
           expect(() => thrower.dive()).to.throw(errorToThrow);
 
           expect(spy).to.have.property('callCount', 0);
+
+          expect(wrapper.find('main').text()).to.equal('HasNotThrown');
         });
       });
     });

packages/enzyme/src/ReactWrapper.js

@@ -643,6 +643,7 @@ class ReactWrapper {
       const nodeHierarchy = [thisNode].concat(nodeParents(this, thisNode));
       renderer.simulateError(nodeHierarchy, rootNode, error);
 
+      this[ROOT].update();
       return this;
     });
   }