[enzyme-adapter-react-16] [fix] avoid a crash with lazy components

Fixes #2539
enzymejs · Sep 17, 2021 · 49bc86b · 49bc86b
1 parent 58d61d3
commit 49bc86b
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 1 deletion.
diff --git a/packages/enzyme-adapter-react-16/src/findCurrentFiberUsingSlowPath.js b/packages/enzyme-adapter-react-16/src/findCurrentFiberUsingSlowPath.js
@@ -93,7 +93,7 @@ function findCurrentFiberUsingSlowPath(fiber) {
       }
     }
   }
-  if (a.stateNode.current === a) {
+  if (a.stateNode?.current === a) {
     // We've determined that A is the current branch.
     return fiber;
   }

diff --git a/packages/enzyme-test-suite/test/ReactWrapper-spec.jsx b/packages/enzyme-test-suite/test/ReactWrapper-spec.jsx
@@ -28,6 +28,8 @@ import {
   memo,
   Profiler,
   Suspense,
+  useState,
+  useEffect,
 } from './_helpers/react-compat';
 import {
   describeWithDOM,
@@ -1285,6 +1287,38 @@ describeWithDOM('mount', () => {
       );
       expect(() => mount(<SuspenseComponent />, { suspenseFallback: false })).to.throw();
     });
+
+    it('avoids a TypeError', () => {
+      const Component2 = lazy(() => Promise.resolve({ default: () => <div /> }));
+
+      function Component() {
+        const [alreadyRun, setAlreadyRun] = useState(false);
+
+        useEffect(() => {
+          setAlreadyRun(true);
+        }, []);
+
+        if (!alreadyRun) {
+          return null;
+        }
+
+        return <Component2 />;
+      }
+
+      const wrapper = mount(
+        <Suspense fallback={<div>loading</div>}>
+          <Component />
+        </Suspense>,
+      );
+      expect(wrapper.debug()).to.equal(
+        `<Suspense fallback={{...}}>
+  <Component />
+  <div>
+    loading
+  </div>
+</Suspense>`,
+      );
+    });
   });
 
   describe('.mount()', () => {