[Tests] add .root() tests for #1152

enzymejs · Jul 6, 2018 · f7fee9c · f7fee9c
1 parent 3070850
commit f7fee9c
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/packages/enzyme-test-suite/test/ReactWrapper-spec.jsx b/packages/enzyme-test-suite/test/ReactWrapper-spec.jsx
@@ -4174,4 +4174,18 @@ describeWithDOM('mount', () => {
       expect(divProps.onClick).to.be.a('function');
     });
   });
+
+  describe('.root()', () => {
+    it('returns the root component instance', () => {
+      class Fixture extends React.Component {
+        render() {
+          return <div><span /><span /></div>;
+        }
+      }
+      const wrapper = mount(<Fixture />);
+      const root = wrapper.root();
+      expect(root.is(Fixture)).to.equal(true);
+      expect(root.childAt(0).children().debug()).to.equal('<span />\n\n\n<span />');
+    });
+  });
 });
diff --git a/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx b/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx
@@ -5088,4 +5088,18 @@ describe('shallow', () => {
       expect(divProps.onClick).to.be.a('function');
     });
   });
+
+  describe('.root()', () => {
+    it('returns the root DOM node', () => {
+      class Fixture extends React.Component {
+        render() {
+          return <div><span /><span /></div>;
+        }
+      }
+      const wrapper = shallow(<Fixture />);
+      const root = wrapper.root();
+      expect(root.is('div')).to.equal(true);
+      expect(root.children().debug()).to.equal('<span />\n\n\n<span />');
+    });
+  });
 });