[Tests] add test cases for finding forwardRefs by display name

Closes #1896.
enzymejs · Nov 12, 2018 · d71ba69 · d71ba69
1 parent f6f097f
commit d71ba69
Show file tree

Hide file tree

Showing 2 changed files with 78 additions and 0 deletions.
diff --git a/packages/enzyme-test-suite/test/ReactWrapper-spec.jsx b/packages/enzyme-test-suite/test/ReactWrapper-spec.jsx
@@ -1487,6 +1487,45 @@ describeWithDOM('mount', () => {
 
       expect(containerDiv.querySelectorAll('h1')).to.have.lengthOf(1);
     });
+
+    describeIf(is('>= 16.3'), 'forwardRef', () => {
+      it('finds forwardRefs', () => {
+        const Component = forwardRef(() => <div />);
+        class Foo extends React.Component {
+          render() {
+            return (
+              <div>
+                <Component />
+                <Component />
+              </div>
+            );
+          }
+        }
+
+        const wrapper = mount(<Foo />);
+        expect(wrapper.find(Component)).to.have.lengthOf(2);
+        expect(wrapper.find('ForwardRef')).to.have.lengthOf(2);
+      });
+
+      it('finds forwardRef by custom display name', () => {
+        const Component = forwardRef(() => <div />);
+        Component.displayName = 'CustomForwardRef';
+        class Foo extends React.Component {
+          render() {
+            return (
+              <div>
+                <Component />
+                <Component />
+              </div>
+            );
+          }
+        }
+
+        const wrapper = mount(<Foo />);
+        expect(wrapper.find(Component)).to.have.lengthOf(2);
+        expect(wrapper.find(Component.displayName)).to.have.lengthOf(2);
+      });
+    });
   });
 
   describe('.findWhere(predicate)', () => {

diff --git a/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx b/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx
@@ -1234,6 +1234,45 @@ describe('shallow', () => {
 
       expect(wrapper.find('span')).to.have.lengthOf(1);
     });
+
+    describeIf(is('>= 16.3'), 'forwardRef', () => {
+      it('finds forwardRefs', () => {
+        const Component = forwardRef(() => <div />);
+        class Foo extends React.Component {
+          render() {
+            return (
+              <div>
+                <Component />
+                <Component />
+              </div>
+            );
+          }
+        }
+
+        const wrapper = shallow(<Foo />);
+        expect(wrapper.find(Component)).to.have.lengthOf(2);
+        expect(wrapper.find('ForwardRef')).to.have.lengthOf(2);
+      });
+
+      it('finds forwardRef by custom display name', () => {
+        const Component = forwardRef(() => <div />);
+        Component.displayName = 'CustomForwardRef';
+        class Foo extends React.Component {
+          render() {
+            return (
+              <div>
+                <Component />
+                <Component />
+              </div>
+            );
+          }
+        }
+
+        const wrapper = shallow(<Foo />);
+        expect(wrapper.find(Component)).to.have.lengthOf(2);
+        expect(wrapper.find(Component.displayName)).to.have.lengthOf(2);
+      });
+    });
   });
 
   describe('.findWhere(predicate)', () => {