[Refactor] mount/shallow: getElement: use this.single

enzymejs · Oct 2, 2018 · 6b63db3 · 6b63db3
1 parent 316a2c3
commit 6b63db3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 12 deletions.
diff --git a/packages/enzyme/src/ReactWrapper.js b/packages/enzyme/src/ReactWrapper.js
@@ -154,10 +154,7 @@ class ReactWrapper {
    * @return {ReactElement}
    */
   getElement() {
-    if (this.length !== 1) {
-      throw new Error('ReactWrapper::getElement() can only be called when wrapping one node');
-    }
-    return getAdapter(this[OPTIONS]).nodeToElement(this[NODE]);
+    return this.single('getElement', () => getAdapter(this[OPTIONS]).nodeToElement(this[NODE]));
   }
 
   /**
@@ -219,10 +216,7 @@ class ReactWrapper {
    * @returns {ReactComponent|DOMComponent}
    */
   instance() {
-    if (this.length !== 1) {
-      throw new Error('ReactWrapper::instance() can only be called on single nodes');
-    }
-    return this[NODE].instance;
+    return this.single('instance', () => this[NODE].instance);
   }
 
   /**

diff --git a/packages/enzyme/src/ShallowWrapper.js b/packages/enzyme/src/ShallowWrapper.js
@@ -273,10 +273,7 @@ class ShallowWrapper {
    * @return {ReactElement}
    */
   getElement() {
-    if (this.length !== 1) {
-      throw new Error('ShallowWrapper::getElement() can only be called when wrapping one node');
-    }
-    return getAdapter(this[OPTIONS]).nodeToElement(this.getNodeInternal());
+    return this.single('getElement', n => getAdapter(this[OPTIONS]).nodeToElement(n));
   }
 
   /**