fix(findComponent): return root instance if it matches (#834)

src/vueWrapper.ts

@@ -160,11 +160,6 @@ export class VueWrapper<T extends ComponentPublicInstance>
       }
     }
 
-    const result = find(this.vm.$.subTree, selector)
-    if (result.length) {
-      return createWrapper(null, result[0])
-    }
-
     // https://github.com/vuejs/vue-test-utils-next/issues/211
     // VTU v1 supported finding the component mounted itself.
     // eg: mount(Comp).findComponent(Comp)
@@ -173,6 +168,11 @@ export class VueWrapper<T extends ComponentPublicInstance>
       return createWrapper(null, this.vm.$.vnode.component?.proxy!)
     }
 
+    const result = find(this.vm.$.subTree, selector)
+    if (result.length) {
+      return createWrapper(null, result[0])
+    }
+
     return createWrapperError('VueWrapper')
   }
 

tests/findComponent.spec.ts

@@ -76,15 +76,6 @@ describe('findComponent', () => {
     )
   })
 
-  it('finds a component when root of mounted component', async () => {
-    const wrapper = mount(compD)
-    // make sure it finds the component, not its root
-    expect(wrapper.findComponent('.c-as-root-on-d').vm).toHaveProperty(
-      '$options.name',
-      'ComponentC'
-    )
-  })
-
   it('finds component by name', () => {
     const wrapper = mount(compA)
     expect(wrapper.findComponent({ name: 'Hello' }).text()).toBe('Hello world')
@@ -109,6 +100,19 @@ describe('findComponent', () => {
     expect(wrapper.html()).toContain('bar')
   })
 
+  it('finds root component when recursion is used', async () => {
+    const Comp = defineComponent({
+      props: ['depth'],
+      name: 'Comp',
+      template: `
+        <Comp :depth="depth + 1" v-if="depth < 2" />
+        Depth {{ depth }}
+      `
+    })
+    const wrapper = mount(Comp, { props: { depth: 0 } })
+    expect(wrapper.findComponent(Comp).props('depth')).toBe(0)
+  })
+
   it('finds component without a name by using its object definition', () => {
     const Component = {
       template: '<div><component-without-name/></div>',