Add public root method

enzymejs · Sep 18, 2017 · 5db6d61 · 5db6d61
1 parent 35df18a
commit 5db6d61
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/packages/enzyme/src/ReactWrapper.js b/packages/enzyme/src/ReactWrapper.js
@@ -102,6 +102,15 @@ class ReactWrapper {
     privateSet(this, OPTIONS, root ? root[OPTIONS] : options);
   }
 
+  /**
+   * Returns the root wrapper
+   *
+   * @return {ReactWrapper}
+   */
+  root() {
+    return this[ROOT];
+  }
+
   /**
    * Returns the wrapped component.
    *
@@ -1075,7 +1084,6 @@ function privateWarning(prop, extraMessage) {
 privateWarning('node', 'Consider using the getElement() method instead.');
 privateWarning('nodes', 'Consider using the getElements() method instead.');
 privateWarning('renderer', '');
-privateWarning('root', '');
 privateWarning('options', '');
 privateWarning('complexSelector', '');
 

diff --git a/packages/enzyme/src/ShallowWrapper.js b/packages/enzyme/src/ShallowWrapper.js
@@ -145,6 +145,17 @@ class ShallowWrapper {
     privateSet(this, OPTIONS, root ? root[OPTIONS] : options);
   }
 
+
+  /**
+   * Returns the root wrapper
+   *
+   * @return {ShallowWrapper}
+   */
+  root() {
+    return this[ROOT];
+  }
+
+
   getNodeInternal() {
     if (this.length !== 1) {
       throw new Error(
@@ -1183,7 +1194,6 @@ function privateWarning(prop, extraMessage) {
 privateWarning('node', 'Consider using the getElement() method instead.');
 privateWarning('nodes', 'Consider using the getElements() method instead.');
 privateWarning('renderer', '');
-privateWarning('root', '');
 privateWarning('options', '');
 privateWarning('complexSelector', '');