fix: allow using findComponent on a functional component (#1593)

vuejs · Jul 4, 2020 · 2d6b497 · 2d6b497
1 parent 5eb9ab1
commit 2d6b497
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/packages/test-utils/src/wrapper.js b/packages/test-utils/src/wrapper.js
@@ -237,7 +237,7 @@ export default class Wrapper implements BaseWrapper {
    */
   findComponent(rawSelector: Selector): Wrapper | ErrorWrapper {
     const selector = getSelector(rawSelector, 'findComponent')
-    if (!this.vm) {
+    if (!this.vm && !this.isFunctionalComponent) {
       throwError(
         'You cannot chain findComponent off a DOM element. It can only be used on Vue Components.'
       )

diff --git a/test/specs/wrapper/find.spec.js b/test/specs/wrapper/find.spec.js
@@ -162,6 +162,18 @@ describeWithShallowAndMount('find', mountingMethod => {
       .with.property('message', message)
   })
 
+  it('allows using findComponent on functional component', () => {
+    const FuncComponentWithChildren = {
+      functional: true,
+      components: {
+        ChildComponent: Component
+      },
+      render: h => h('div', {}, [h(Component)])
+    }
+    const wrapper = mountingMethod(FuncComponentWithChildren)
+    expect(wrapper.findComponent(Component).exists()).to.be.true
+  })
+
   itSkipIf(isRunningPhantomJS, 'returns Wrapper of class component', () => {
     const TestComponent = {
       template: `