fix: find component by its definition when using stub without name

src/stubs.ts

@@ -8,7 +8,8 @@ import {
   ComponentOptions,
   defineComponent,
   VNodeProps,
-  VNodeTypes
+  VNodeTypes,
+  ConcreteComponent
 } from 'vue'
 import { hyphenate } from './utils/vueShared'
 import { MOUNT_COMPONENT_REF, MOUNT_PARENT_NAME } from './constants'
@@ -23,7 +24,7 @@ interface StubOptions {
   renderStubDefaultSlot?: boolean
 }
 
-const stubsMap: WeakMap<ComponentOptions, VNodeTypes> = new WeakMap()
+const stubsMap: WeakMap<ConcreteComponent, VNodeTypes> = new WeakMap()
 
 export const getOriginalVNodeTypeFromStub = (
   type: ComponentOptions
@@ -184,6 +185,7 @@ export function stubComponents(
 
       // case 2: custom implementation
       if (stub && stub !== true) {
+        stubsMap.set(stubs[name], type)
         // pass the props and children, for advanced stubbing
         return [stubs[name], props, children, patchFlag, dynamicProps]
       }

tests/findComponent.spec.ts

@@ -122,6 +122,24 @@ describe('findComponent', () => {
     expect(wrapper.findComponent(Hello).exists()).toBe(true)
   })
 
+  it('finds a component by its definition when using stub without name', () => {
+    const StubWithoutName = {
+      template: '<div>stub-without-name</div>'
+    }
+
+    const Component = {
+      template: '<div><other-name /></div>',
+      components: {
+        OtherName: Hello
+      }
+    }
+
+    const wrapper = mount(Component, {
+      global: { stubs: { Hello: StubWithoutName } }
+    })
+    expect(wrapper.findComponent(Hello).exists()).toBe(true)
+  })
+
   it('finds a component without a name by its locally assigned name', () => {
     const Component = {
       template: '<div><component-without-name/></div>',