Exclude forward ref on deep mount

Author: mmarcuccio

src/mount.js

@@ -1,5 +1,6 @@
 import omitBy from 'lodash/omitBy';
 import isNil from 'lodash/isNil';
+import {ForwardRef} from 'react-is';
 
 import {typeName} from 'enzyme/build/Debug';
 import {childrenOfNode, propsOfNode} from 'enzyme/build/RSTTraversal';
@@ -48,7 +49,10 @@ function internalNodeToJson(node, options) {
     return node.map(child => internalNodeToJson(child, options));
   }
 
-  if (options.mode === 'deep' && typeof node.type === 'function') {
+  if (
+    options.mode === 'deep' &&
+    (typeof node.type === 'function' || node.type.$$typeof === ForwardRef)
+  ) {
     return internalNodeToJson(node.rendered, options);
   }
 

tests/__snapshots__/mount-deep.test.js.snap

@@ -227,6 +227,23 @@ exports[`converts pure mount with mixed children 1`] = `
 </div>
 `;
 
+exports[`excludes forwardRef node but renders wrapped component 1`] = `
+<div
+  className="basic-class undefined"
+  onClick={[Function]}
+>
+  <div
+    className="group"
+    id="group-id"
+  >
+    <span />
+    <span
+      className="empty"
+    />
+  </div>
+</div>
+`;
+
 exports[`handles a component which returns null 1`] = `""`;
 
 exports[`includes undefined props 1`] = `

tests/mount-deep.test.js

@@ -19,6 +19,7 @@ import {
   ClassWithNull,
   ClassArrayRender,
 } from './fixtures/class';
+import {ForwardRefWithDefaultProps} from './fixtures/forwardRef';
 
 Enzyme.configure({adapter: new Adapter()});
 const deepOptions = {mode: 'deep'};
@@ -128,3 +129,9 @@ it('renders multiple elements as a result of find', () => {
 
   expect(mountToJson(mounted.find('li'), deepOptions)).toMatchSnapshot();
 });
+
+it('excludes forwardRef node but renders wrapped component', () => {
+  const mounted = mount(<ForwardRefWithDefaultProps />);
+
+  expect(mountToJson(mounted, deepOptions)).toMatchSnapshot();
+});