Add missing tests

enzymejs · Jan 6, 2018 · 09cdcc7 · 09cdcc7
1 parent 9a96e42
commit 09cdcc7
Show file tree

Hide file tree

Showing 2 changed files with 107 additions and 4 deletions.
diff --git a/packages/enzyme-test-suite/test/ReactWrapper-spec.jsx b/packages/enzyme-test-suite/test/ReactWrapper-spec.jsx
@@ -344,7 +344,9 @@ describeWithDOM('mount', () => {
 
     describeIf(!REACT013, 'stateless components', () => {
       it('should match composite components', () => {
-        const Foo = () => <div />;
+        function Foo() {
+          return <div />;
+        }
         const wrapper = mount((
           <div>
             <Foo />
@@ -353,6 +355,45 @@ describeWithDOM('mount', () => {
         const b = <Foo />;
         expect(wrapper.contains(b)).to.equal(true);
       });
+
+      it('should match composite components if rendered by function', () => {
+        function Foo() {
+          return <div />;
+        }
+        const renderStatelessComponent = () => <Foo />;
+        const wrapper = mount((
+          <div>
+            {renderStatelessComponent()}
+          </div>
+        ));
+        const b = <Foo />;
+        expect(wrapper.contains(b)).to.equal(true);
+      });
+
+      it('should match composite components based on component display name', () => {
+        function Foo() {
+          return <div />;
+        }
+        const wrapper = mount((
+          <div>
+            <Foo />
+          </div>
+        ));
+        expect(wrapper.contains('Foo')).to.equal(true);
+      });
+
+      it('should match composite components based on component display name if rendered by function', () => {
+        function Foo() {
+          return <div />;
+        }
+        const renderStatelessComponent = () => <Foo />;
+        const wrapper = mount((
+          <div>
+            {renderStatelessComponent()}
+          </div>
+        ));
+        expect(wrapper.contains('Foo')).to.equal(true);
+      });
     });
   });
 
@@ -483,6 +524,29 @@ describeWithDOM('mount', () => {
       expect(wrapper.find('Foo').type()).to.equal(Foo);
     });
 
+    describeIf(!REACT013, 'stateless components', () => {
+      it('should find a stateless component based on a component displayName', () => {
+        const Foo = () => <div />;
+        const wrapper = mount((
+          <div>
+            <Foo className="foo" />
+          </div>
+        ));
+        expect(wrapper.find('Foo').type()).to.equal(Foo);
+      });
+
+      it('should find a stateless component based on a component displayName if rendered by function', () => {
+        const Foo = () => <div />;
+        const renderStatelessComponent = () => <Foo className="foo" />;
+        const wrapper = mount((
+          <div>
+            {renderStatelessComponent()}
+          </div>
+        ));
+        expect(wrapper.find('Foo').type()).to.equal(Foo);
+      });
+    });
+
     it('should find component based on a react prop', () => {
       const wrapper = mount((
         <div>

diff --git a/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx b/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx
@@ -296,9 +296,9 @@ describe('shallow', () => {
 
     describeIf(!REACT013, 'stateless function components', () => {
       it('should match composite components', () => {
-        const Foo = () => (
-          <div />
-        );
+        function Foo() {
+          return <div />;
+        }
 
         const wrapper = shallow((
           <div>
@@ -308,6 +308,45 @@ describe('shallow', () => {
         const b = <Foo />;
         expect(wrapper.contains(b)).to.equal(true);
       });
+
+      it('should match composite components if rendered by function', () => {
+        function Foo() {
+          return <div />;
+        }
+        const renderStatelessComponent = () => <Foo />;
+        const wrapper = shallow((
+          <div>
+            {renderStatelessComponent()}
+          </div>
+        ));
+        const b = <Foo />;
+        expect(wrapper.contains(b)).to.equal(true);
+      });
+
+      it('should match composite components based on component display name', () => {
+        function Foo() {
+          return <div />;
+        }
+        const wrapper = shallow((
+          <div>
+            <Foo />
+          </div>
+        ));
+        expect(wrapper.contains('Foo')).to.equal(true);
+      });
+
+      it('should match composite components based on component display name if rendered by function', () => {
+        function Foo() {
+          return <div />;
+        }
+        const renderStatelessComponent = () => <Foo />;
+        const wrapper = shallow((
+          <div>
+            {renderStatelessComponent()}
+          </div>
+        ));
+        expect(wrapper.contains('Foo')).to.equal(true);
+      });
     });
   });