fix(types): Allow Callbacks Passed to before*/after* to Return Anything (#6393)

Author: LuciNyan

packages/runner/src/types/tasks.ts

@@ -498,28 +498,31 @@ export type HookListener<T extends any[], Return = void> = (
   ...args: T
 ) => Awaitable<Return>
 
-export type HookCleanupCallback = (() => Awaitable<unknown>) | void
+/**
+ * @deprecated
+ */
+export type HookCleanupCallback = unknown
 
 export interface BeforeAllListener {
-  (suite: Readonly<Suite | File>): Awaitable<HookCleanupCallback>
+  (suite: Readonly<Suite | File>): Awaitable<unknown>
 }
 
 export interface AfterAllListener {
-  (suite: Readonly<Suite | File>): Awaitable<void>
+  (suite: Readonly<Suite | File>): Awaitable<unknown>
 }
 
 export interface BeforeEachListener<ExtraContext = object> {
   (
     context: ExtendedContext<Test | Custom> & ExtraContext,
     suite: Readonly<Suite>
-  ): Awaitable<HookCleanupCallback>
+  ): Awaitable<unknown>
 }
 
 export interface AfterEachListener<ExtraContext = object> {
   (
     context: ExtendedContext<Test | Custom> & ExtraContext,
     suite: Readonly<Suite>
-  ): Awaitable<void>
+  ): Awaitable<unknown>
 }
 
 export interface SuiteHooks<ExtraContext = object> {

test/core/test/hooks.test.ts

@@ -1,4 +1,4 @@
-import { beforeAll, beforeEach, expect, it, suite } from 'vitest'
+import { afterAll, afterEach, beforeAll, beforeEach, expect, it, suite } from 'vitest'
 
 let count = -1
 
@@ -80,4 +80,26 @@ suite('hooks cleanup', () => {
   it('end', () => {
     expect(cleanUpCount).toBe(0)
   })
+
+  suite('do nothing when given a non-function value as cleanupCallback', () => {
+    beforeAll(() => {
+      return 1
+    })
+    beforeEach(() => {
+      return null
+    })
+    afterAll(() => {
+      return '1'
+    })
+    afterEach(() => {
+      return {}
+    })
+
+    it('one', () => {
+      expect(cleanUpCount).toBe(0)
+    })
+  })
+  it('end', () => {
+    expect(cleanUpCount).toBe(0)
+  })
 })