From 45073964e57cc0ef0a42ecd1d0f4729c2467860d Mon Sep 17 00:00:00 2001
From: Sstern6 <scott.stern06@gmail.com>
Date: Mon, 14 Jan 2019 18:30:24 -0800
Subject: [PATCH] =?UTF-8?q?[fix]=20`shallow`:=20`getNodesInternal`:=20dele?=
 =?UTF-8?q?gate=20to=20the=20adapter=E2=80=99s=20`shouldUpdateComponent`?=
 =?UTF-8?q?=20method=20before=20updating=20the=20wrapper.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Fixes #1916.
---
 .../test/ShallowWrapper-spec.jsx              |  7 ++-
 .../test/shared/methods/find.jsx              | 44 +++++++++++++++++++
 packages/enzyme/src/ShallowWrapper.js         |  9 +++-
 3 files changed, 57 insertions(+), 3 deletions(-)

diff --git a/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx b/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx
index 253b11de1..f676be4f1 100644
--- a/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx
+++ b/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx
@@ -1783,9 +1783,12 @@ describe('shallow', () => {
           }
         }
         const wrapper = shallow(<MyComponent />, { disableLifecycleMethods: true });
-        expect(wrapper.find(Table).length).to.equal(0);
+        expect(wrapper.find(Table)).to.have.lengthOf(0);
+
         wrapper.instance().componentDidMount();
-        expect(wrapper.find(Table).length).to.equal(1);
+        // wrapper.update(); // TODO: uncomment or delete
+
+        expect(wrapper.find(Table)).to.have.lengthOf(1);
       });
 
       it('calls shouldComponentUpdate when disableLifecycleMethods flag is true', () => {
diff --git a/packages/enzyme-test-suite/test/shared/methods/find.jsx b/packages/enzyme-test-suite/test/shared/methods/find.jsx
index 2af572afe..0b58969d4 100644
--- a/packages/enzyme-test-suite/test/shared/methods/find.jsx
+++ b/packages/enzyme-test-suite/test/shared/methods/find.jsx
@@ -293,6 +293,50 @@ export default function describeFind({
       expect(wrapper.find('.b').find('.c')).to.have.lengthOf(6);
     });
 
+    it('can call find on the same wrapper more than once', () => {
+      class TestComponent extends React.Component {
+        render() {
+          return (
+            <div>
+              <h1>Title</h1>
+              <span key="1">1</span>
+              <span key="2">2</span>
+            </div>
+          );
+        }
+      }
+      const component = Wrap(<TestComponent />);
+
+      const cards = component.find('span');
+
+      const title = component.find('h1'); // for side effects
+      expect(title.is('h1')).to.equal(true);
+
+      expect(cards.at(0).parent().is('div')).to.equal(true);
+    });
+
+    describeIf(is('> 0.13'), 'stateless function components (SFCs)', () => {
+      it('can call find on the same wrapper more than once', () => {
+        function TestComponentSFC() {
+          return (
+            <div>
+              <h1>Title</h1>
+              <span key="1">1</span>
+              <span key="2">2</span>
+            </div>
+          );
+        }
+        const component = Wrap(<TestComponentSFC />);
+
+        const cards = component.find('span');
+
+        const title = component.find('h1'); // for side effects
+        expect(title.is('h1')).to.equal(true);
+
+        expect(cards.at(0).parent().debug()).to.equal('<div />');
+      });
+    });
+
     it('works with an adjacent sibling selector', () => {
       const a = 'some';
       const b = 'text';
diff --git a/packages/enzyme/src/ShallowWrapper.js b/packages/enzyme/src/ShallowWrapper.js
index 99941e553..22cdc4847 100644
--- a/packages/enzyme/src/ShallowWrapper.js
+++ b/packages/enzyme/src/ShallowWrapper.js
@@ -460,8 +460,13 @@ class ShallowWrapper {
    */
   getNodesInternal() {
     if (this[ROOT] === this && this.length === 1) {
-      this.update();
+      const adapter = getAdapter(this[OPTIONS]);
+      const prevProps = (this[UNRENDERED] && this[UNRENDERED].props) || {};
+      if (!adapter.shouldUpdateComponent || adapter.shouldUpdateComponent(prevProps, this[ROOT])) {
+        this.update();
+      }
     }
+
     return this[NODES];
   }
 
@@ -556,8 +561,10 @@ class ShallowWrapper {
    */
   unmount() {
     this[RENDERER].unmount();
+    this.update();
     if (this[ROOT][WRAPPING_COMPONENT]) {
       this[ROOT][WRAPPING_COMPONENT].unmount();
+      this[ROOT][WRAPPING_COMPONENT].update();
     }
     return this;
   }