From 5a2360b876665445c1ba585d1105e7729c955d34 Mon Sep 17 00:00:00 2001
From: Vladimir Sheremet <sheremet.va@icloud.com>
Date: Tue, 21 Jun 2022 23:19:01 +0300
Subject: [PATCH] fix: align hooks return types with their behaviour

---
 packages/vitest/src/types/tasks.ts | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

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<ExtraContext>
   runIf(condition: any): SuiteAPI<ExtraContext>
 }
 
-export type HookListener<T extends any[], Return = void> = (...args: T) => Awaitable<Return | void>
+export type HookListener<T extends any[], Return = void> = (...args: T) => Awaitable<Return>
+
+export type HookCleanupCallback = (() => Awaitable<unknown>) | void
 
 export interface SuiteHooks {
-  beforeAll: HookListener<[Suite | File], () => Awaitable<void>>[]
+  beforeAll: HookListener<[Suite | File], HookCleanupCallback>[]
   afterAll: HookListener<[Suite | File]>[]
-  beforeEach: HookListener<[TestContext, Suite], () => Awaitable<void>>[]
+  beforeEach: HookListener<[TestContext, Suite], HookCleanupCallback>[]
   afterEach: HookListener<[TestContext, Suite]>[]
 }
 
-export type HookCleanupCallback = (() => Awaitable<void>) | void
-
 export interface SuiteCollector<ExtraContext = {}> {
   readonly name: string
   readonly mode: RunMode