diff --git a/packages/vitest/src/types/tasks.ts b/packages/vitest/src/types/tasks.ts index 0e197879fe34..aaa91746bfc9 100644 --- a/packages/vitest/src/types/tasks.ts +++ b/packages/vitest/src/types/tasks.ts @@ -122,17 +122,17 @@ SuiteCollector runIf(condition: any): SuiteAPI } -export type HookListener = (...args: T) => Awaitable +export type HookListener = (...args: T) => Awaitable + +export type HookCleanupCallback = (() => Awaitable) | void export interface SuiteHooks { - beforeAll: HookListener<[Suite | File], () => Awaitable>[] + beforeAll: HookListener<[Suite | File], HookCleanupCallback>[] afterAll: HookListener<[Suite | File]>[] - beforeEach: HookListener<[TestContext, Suite], () => Awaitable>[] + beforeEach: HookListener<[TestContext, Suite], HookCleanupCallback>[] afterEach: HookListener<[TestContext, Suite]>[] } -export type HookCleanupCallback = (() => Awaitable) | void - export interface SuiteCollector { readonly name: string readonly mode: RunMode