refactor: cleanup find methods

Author: dobromir-hristov

packages/test-utils/src/wrapper.js

@@ -225,41 +225,8 @@ export default class Wrapper implements BaseWrapper {
         'Use `findComponent` instead'
       )
     }
-    const node = find(this.rootNode, this.vm, selector)[0]
-
-    if (!node) {
-      return new ErrorWrapper(rawSelector)
-    }
 
-    const wrapper = createWrapper(node, this.options)
-    wrapper.selector = rawSelector
-    return wrapper
-  }
-
-  /**
-   * Finds DOM elements in tree of the current wrapper that matches
-   * the provided selector.
-   */
-  findAll(rawSelector: Selector): WrapperArray {
-    const selector = getSelector(rawSelector, 'findAll')
-    if (selector.type !== DOM_SELECTOR) {
-      warnDeprecated(
-        'finding components with `findAll`',
-        'Use `findAllComponents` instead'
-      )
-    }
-    const nodes = find(this.rootNode, this.vm, selector)
-    const wrappers = nodes.map(node => {
-      // Using CSS Selector, returns a VueWrapper instance if the root element
-      // binds a Vue instance.
-      const wrapper = createWrapper(node, this.options)
-      wrapper.selector = rawSelector
-      return wrapper
-    })
-
-    const wrapperArray = new WrapperArray(wrappers)
-    wrapperArray.selector = rawSelector
-    return wrapperArray
+    return this.__find(rawSelector, selector)
   }
 
   /**
@@ -273,6 +240,16 @@ export default class Wrapper implements BaseWrapper {
         'findComponent requires a Vue constructor or valid find object. If you are searching for DOM nodes, use `find` instead'
       )
     }
+    if (!this.vm) {
+      throwError(
+        'You cannot chain findComponent off a DOM element. It can only be used on Vue Components.'
+      )
+    }
+
+    return this.__find(rawSelector, selector)
+  }
+
+  __find(rawSelector: Selector, selector: Object): Wrapper | ErrorWrapper {
     const node = find(this.rootNode, this.vm, selector)[0]
 
     if (!node) {
@@ -284,17 +261,41 @@ export default class Wrapper implements BaseWrapper {
     return wrapper
   }
 
+  /**
+   * Finds DOM elements in tree of the current wrapper that matches
+   * the provided selector.
+   */
+  findAll(rawSelector: Selector): WrapperArray {
+    const selector = getSelector(rawSelector, 'findAll')
+    if (selector.type !== DOM_SELECTOR) {
+      warnDeprecated(
+        'finding components with `findAll`',
+        'Use `findAllComponents` instead'
+      )
+    }
+    return this.__findAll(rawSelector, selector)
+  }
+
   /**
    * Finds components in tree of the current wrapper that matches
    * the provided selector.
    */
   findAllComponents(rawSelector: Selector): WrapperArray {
     const selector = getSelector(rawSelector, 'findAll')
+    if (!this.vm) {
+      throwError(
+        'You cannot chain findAllComponents off a DOM element. It can only be used on Vue Components.'
+      )
+    }
     if (selector.type === DOM_SELECTOR) {
       throwError(
-        'findAllComponent requires a Vue constructor or valid find object. If you are searching for DOM nodes, use `find` instead'
+        'findAllComponents requires a Vue constructor or valid find object. If you are searching for DOM nodes, use `find` instead'
       )
     }
+    return this.__findAll(rawSelector, selector)
+  }
+
+  __findAll(rawSelector: Selector, selector: Object): WrapperArray {
     const nodes = find(this.rootNode, this.vm, selector)
     const wrappers = nodes.map(node => {
       // Using CSS Selector, returns a VueWrapper instance if the root element