refactor: refactoring emit some edge solution

Author: cuixiaorui

src/emitMixin.ts renamed to src/emit.ts

@@ -1,12 +1,24 @@
-import { setDevtoolsHook, ComponentPublicInstance, devtools } from 'vue'
+import { setDevtoolsHook, devtools } from 'vue'
 
 const enum DevtoolsHooks {
   COMPONENT_EMIT = 'component:emit'
 }
 
-export const attachEmitListener = (vm: ComponentPublicInstance) => {
-  const events: Record<string, unknown[]> = {}
-  ;(vm as any).__emitted = events
+let events: Record<string, unknown[]>
+
+export function emitted<T = unknown>(
+  eventName?: string
+): T[] | Record<string, T[]> {
+  if (eventName) {
+    const emitted = (events as Record<string, T[]>)[eventName]
+    return emitted
+  }
+
+  return events as Record<string, T[]>
+}
+
+export const attachEmitListener = () => {
+  events = {}
   // use devtools to capture this "emit"
   setDevtoolsHook(createDevTools(events))
 }
@@ -29,9 +41,6 @@ function createDevTools(events): any {
 
 function recordEvent(events, event, args) {
   // Record the event message sent by the emit
-  // Stored by a vm
-  // emitted by a subsequent wrapper.emitted
-  // An event object and a vm.__emitted is a reference
   events[event]
     ? (events[event] = [...events[event], [...args]])
     : (events[event] = [[...args]])

src/mount.ts

@@ -32,7 +32,7 @@ import {
 } from './utils'
 import { processSlot } from './utils/compileSlots'
 import { createWrapper, VueWrapper } from './vueWrapper'
-import { attachEmitListener } from './emitMixin'
+import { attachEmitListener } from './emit'
 import { createDataMixin } from './dataMixin'
 import { MOUNT_COMPONENT_REF, MOUNT_PARENT_NAME } from './constants'
 import { createStub, stubComponents } from './stubs'
@@ -346,6 +346,9 @@ export function mount(
     return vm.$nextTick()
   }
 
+  // add tracking for emitted events
+  attachEmitListener()
+
   // create the app
   const app = createApp(Parent)
 
@@ -446,9 +449,6 @@ export function mount(
   const $vm = Reflect.ownKeys(appRef).length ? appRef : vm
   console.warn = warnSave
 
-  // add tracking for emitted events
-  attachEmitListener($vm)
-
   return createWrapper(app, $vm, setProps)
 }
 

src/vueWrapper.ts

@@ -8,6 +8,7 @@ import { createWrapperError } from './errorWrapper'
 import { TriggerOptions } from './createDomEvent'
 import { find, matches } from './utils/find'
 import { mergeDeep } from './utils'
+import { emitted } from './emit'
 
 export class VueWrapper<T extends ComponentPublicInstance> {
   private componentVM: T
@@ -72,12 +73,7 @@ export class VueWrapper<T extends ComponentPublicInstance> {
   emitted<T = unknown>(): Record<string, T[]>
   emitted<T = unknown>(eventName?: string): T[]
   emitted<T = unknown>(eventName?: string): T[] | Record<string, T[]> {
-    if (eventName) {
-      const emitted = (this.vm['__emitted'] as Record<string, T[]>)[eventName]
-      return emitted
-    }
-
-    return this.vm['__emitted'] as Record<string, T[]>
+    return emitted(eventName)
   }
 
   html() {

tests/emit.spec.ts

@@ -154,4 +154,14 @@ describe('emitted', () => {
     expect(wrapper.emitted('hello')).toHaveLength(1)
     expect(wrapper.emitted('hello')[0]).toEqual(['foo', 'bar'])
   })
+
+  it('captures an event emitted in setup', () => {
+    const Comp = {
+      setup(_, { emit }) {
+        emit('foo')
+      }
+    }
+    const wrapper = mount(Comp)
+    expect(wrapper.emitted().foo).toBeTruthy()
+  })
 })