From 082cee76f25d91203681b960c326932ac1c2dbdb Mon Sep 17 00:00:00 2001
From: Leo Conforti <leo@leoconforti.us>
Date: Wed, 25 Sep 2024 14:37:56 +0000
Subject: [PATCH] add timeouts

---
 packages/vitest/src/index.ts    | 12 ++++++++----
 packages/vitest/src/internal.ts | 20 ++++++++++++--------
 2 files changed, 20 insertions(+), 12 deletions(-)

diff --git a/packages/vitest/src/index.ts b/packages/vitest/src/index.ts
index b0ff5d7563..a2479ef887 100644
--- a/packages/vitest/src/index.ts
+++ b/packages/vitest/src/index.ts
@@ -61,7 +61,8 @@ export const addEqualityTesters: () => void = internal.addEqualityTesters
 export const beforeAllEffect: <E>(
   self: (
     suite: Readonly<V.RunnerTestSuite | V.RunnerTestFile>
-  ) => Effect.Effect<V.HookCleanupCallback | PromiseLike<V.HookCleanupCallback>, E, never>
+  ) => Effect.Effect<V.HookCleanupCallback | PromiseLike<V.HookCleanupCallback>, E, never>,
+  timeout?: number
 ) => void = internal.beforeAll
 
 /**
@@ -71,14 +72,16 @@ export const beforeEachEffect: <E>(
   self: (
     ctx: V.TaskContext<V.RunnerCustomCase<object> | V.RunnerTestCase<object>> & V.TestContext & object,
     suite: V.RunnerTestSuite
-  ) => Effect.Effect<V.HookCleanupCallback | PromiseLike<V.HookCleanupCallback>, E, never>
+  ) => Effect.Effect<V.HookCleanupCallback | PromiseLike<V.HookCleanupCallback>, E, never>,
+  timeout?: number
 ) => void = internal.beforeEach
 
 /**
  * @since 1.1.0
  */
 export const afterAllEffect: <E>(
-  self: (suite: Readonly<V.RunnerTestSuite | V.RunnerTestFile>) => Effect.Effect<void | PromiseLike<void>, E, never>
+  self: (suite: Readonly<V.RunnerTestSuite | V.RunnerTestFile>) => Effect.Effect<void | PromiseLike<void>, E, never>,
+  timeout?: number
 ) => void = internal.afterAll
 
 /**
@@ -88,7 +91,8 @@ export const afterEachEffect: <E>(
   self: (
     ctx: V.TaskContext<V.RunnerCustomCase<object> | V.RunnerTestCase<object>> & V.TestContext & object,
     suite: V.RunnerTestSuite
-  ) => Effect.Effect<void | PromiseLike<void>, E, never>
+  ) => Effect.Effect<void | PromiseLike<void>, E, never>,
+  timeout?: number
 ) => void = internal.afterEach
 
 /**
diff --git a/packages/vitest/src/internal.ts b/packages/vitest/src/internal.ts
index a8a6dec638..1d8933dce6 100644
--- a/packages/vitest/src/internal.ts
+++ b/packages/vitest/src/internal.ts
@@ -86,29 +86,33 @@ export const addEqualityTesters = () => {
 export const beforeAll = <E>(
   self: (
     suite: Readonly<V.RunnerTestSuite | V.RunnerTestFile>
-  ) => Effect.Effect<V.HookCleanupCallback | PromiseLike<V.HookCleanupCallback>, E, never>
-): void => V.beforeAll((suite) => runHook(self(suite)))
+  ) => Effect.Effect<V.HookCleanupCallback | PromiseLike<V.HookCleanupCallback>, E, never>,
+  timeout?: number
+): void => V.beforeAll((suite) => runHook(self(suite)), timeout)
 
 /** @internal */
 export const beforeEach = <E>(
   self: (
     ctx: V.TaskContext<V.RunnerCustomCase<object> | V.RunnerTestCase<object>> & V.TestContext & object,
     suite: V.RunnerTestSuite
-  ) => Effect.Effect<V.HookCleanupCallback | PromiseLike<V.HookCleanupCallback>, E, never>
-): void => V.beforeEach((ctx, suite) => runHook(self(ctx, suite)))
+  ) => Effect.Effect<V.HookCleanupCallback | PromiseLike<V.HookCleanupCallback>, E, never>,
+  timeout?: number
+): void => V.beforeEach((ctx, suite) => runHook(self(ctx, suite)), timeout)
 
 /** @internal */
 export const afterAll = <E>(
-  self: (suite: Readonly<V.RunnerTestSuite | V.RunnerTestFile>) => Effect.Effect<void | PromiseLike<void>, E, never>
-): void => V.afterAll((suite) => runHook(self(suite)))
+  self: (suite: Readonly<V.RunnerTestSuite | V.RunnerTestFile>) => Effect.Effect<void | PromiseLike<void>, E, never>,
+  timeout?: number
+): void => V.afterAll((suite) => runHook(self(suite)), timeout)
 
 /** @internal */
 export const afterEach = <E>(
   self: (
     ctx: V.TaskContext<V.RunnerCustomCase<object> | V.RunnerTestCase<object>> & V.TestContext & object,
     suite: V.RunnerTestSuite
-  ) => Effect.Effect<void | PromiseLike<void>, E, never>
-): void => V.afterEach((ctx, suite) => runHook(self(ctx, suite)))
+  ) => Effect.Effect<void | PromiseLike<void>, E, never>,
+  timeout?: number
+): void => V.afterEach((ctx, suite) => runHook(self(ctx, suite)), timeout)
 
 /** @internal */
 const makeTester = <R, E2 = never>(