From dfd85066c0b8a149b26fbb13ee1a9d3dc2441c73 Mon Sep 17 00:00:00 2001
From: Vladimir <sheremet.va@icloud.com>
Date: Thu, 23 Jun 2022 00:09:35 +0300
Subject: [PATCH] fix: align hooks return types with their behaviour (#1522)

---
 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