From f7fee9ce4b9463e3ba16028ce93f3f4a182d1fb7 Mon Sep 17 00:00:00 2001
From: Jordan Harband <ljharb@gmail.com>
Date: Fri, 6 Jul 2018 10:17:41 -0700
Subject: [PATCH] [Tests] add `.root()` tests for #1152

---
 .../enzyme-test-suite/test/ReactWrapper-spec.jsx   | 14 ++++++++++++++
 .../enzyme-test-suite/test/ShallowWrapper-spec.jsx | 14 ++++++++++++++
 2 files changed, 28 insertions(+)

diff --git a/packages/enzyme-test-suite/test/ReactWrapper-spec.jsx b/packages/enzyme-test-suite/test/ReactWrapper-spec.jsx
index d40f51c20..fbdbb4362 100644
--- 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
index 82fa3a661..2e023b853 100644
--- 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 />');
+    });
+  });
 });