[enzyme-adapter-react-{16,16.3}] [new] add isCustomComponentElement

enzymejs · Sep 20, 2018 · 3254521 · 3254521
1 parent a7f1a47
commit 3254521
Show file tree

Hide file tree

Showing 3 changed files with 43 additions and 0 deletions.
diff --git a/packages/enzyme-adapter-react-16.3/src/ReactSixteenThreeAdapter.js b/packages/enzyme-adapter-react-16.3/src/ReactSixteenThreeAdapter.js
@@ -11,6 +11,7 @@ import TestUtils from 'react-dom/test-utils';
 import {
   isElement,
   isPortal,
+  isForwardRef,
   isValidElementType,
   AsyncMode,
   Fragment,
@@ -483,6 +484,13 @@ class ReactSixteenThreeAdapter extends EnzymeAdapter {
     return typeOfNode(fragment) === Fragment;
   }
 
+  isCustomComponentElement(inst) {
+    if (!inst || !this.isValidElement(inst)) {
+      return false;
+    }
+    return typeof inst.type === 'function' || isForwardRef(inst);
+  }
+
   createElement(...args) {
     return React.createElement(...args);
   }

diff --git a/packages/enzyme-adapter-react-16/src/ReactSixteenAdapter.js b/packages/enzyme-adapter-react-16/src/ReactSixteenAdapter.js
@@ -11,6 +11,7 @@ import TestUtils from 'react-dom/test-utils';
 import {
   isElement,
   isPortal,
+  isForwardRef,
   isValidElementType,
   AsyncMode,
   Fragment,
@@ -525,6 +526,13 @@ class ReactSixteenAdapter extends EnzymeAdapter {
     return typeOfNode(fragment) === Fragment;
   }
 
+  isCustomComponentElement(inst) {
+    if (!inst || !this.isValidElement(inst)) {
+      return false;
+    }
+    return typeof inst.type === 'function' || isForwardRef(inst);
+  }
+
   createElement(...args) {
     return React.createElement(...args);
   }

diff --git a/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx b/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx
@@ -6923,6 +6923,33 @@ describe('shallow', () => {
       expect(underwater.is(RendersDOM)).to.equal(true);
     });
 
+    describeIf(is('>=16.3.0'), 'forwardRef Elements', () => {
+      const ForwardRefWrapsRendersDOM = forwardRef && forwardRef(() => <WrapsRendersDOM />);
+      const NestedForwarRefsWrapsRendersDom = forwardRef
+        && forwardRef(() => <ForwardRefWrapsRendersDOM />);
+
+      if (forwardRef) {
+        NestedForwarRefsWrapsRendersDom.contextTypes = { foo: PropTypes.string };
+        ForwardRefWrapsRendersDOM.contextTypes = { foo: PropTypes.string };
+      }
+
+      it('dives + shallow-renders a forwardRef component', () => {
+        const wrapper = shallow(<ForwardRefWrapsRendersDOM />);
+        expect(wrapper.is(WrapsRendersDOM)).to.equal(true);
+
+        const underwater = wrapper.dive();
+        expect(underwater.is(RendersDOM)).to.equal(true);
+      });
+
+      it('dives + shallow-renders a with nested forwardRefs component', () => {
+        const wrapper = shallow(<NestedForwarRefsWrapsRendersDom />);
+        expect(wrapper.is(ForwardRefWrapsRendersDOM)).to.equal(true);
+
+        const underwater = wrapper.dive();
+        expect(underwater.is(WrapsRendersDOM)).to.equal(true);
+      });
+    });
+
     it('merges and pass options through', () => {
       const wrapper = shallow(<ContextWrapsRendersDOM />, { context: { foo: 'hello' } });
       expect(wrapper.context()).to.deep.equal({ foo: 'hello' });