From cba3ff0937c51f46b6ff9789e568b32fc03f3e36 Mon Sep 17 00:00:00 2001
From: Vladimir Sheremet <sleuths.slews0s@icloud.com>
Date: Wed, 5 Oct 2022 14:46:19 +0200
Subject: [PATCH] fix: check hook teardown return type, closes #2092

---
 packages/vitest/src/runtime/run.ts | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/packages/vitest/src/runtime/run.ts b/packages/vitest/src/runtime/run.ts
index 19b468eaa05e..cdcad0bd4b9a 100644
--- a/packages/vitest/src/runtime/run.ts
+++ b/packages/vitest/src/runtime/run.ts
@@ -2,7 +2,7 @@ import { performance } from 'perf_hooks'
 import limit from 'p-limit'
 import type { BenchTask, Benchmark, BenchmarkResult, File, HookCleanupCallback, HookListener, ResolvedConfig, Suite, SuiteHooks, Task, TaskResult, TaskState, Test } from '../types'
 import { vi } from '../integrations/vi'
-import { clearTimeout, createDefer, getFullName, getWorkerState, hasFailed, hasTests, isBrowser, isNode, isRunningInBenchmark, partitionSuiteChildren, setTimeout, shuffle } from '../utils'
+import { assertTypes, clearTimeout, createDefer, getFullName, getWorkerState, hasFailed, hasTests, isBrowser, isNode, isRunningInBenchmark, partitionSuiteChildren, setTimeout, shuffle } from '../utils'
 import { getState, setState } from '../integrations/chai/jest-expect'
 import { GLOBAL_EXPECT } from '../integrations/chai/constants'
 import { takeCoverageInsideWorker } from '../integrations/coverage'
@@ -84,6 +84,15 @@ async function sendTasksUpdate() {
   }
 }
 
+const callCleanupHooks = async (cleanups: HookCleanupCallback[]) => {
+  await Promise.all(cleanups.map(async (fn) => {
+    if (!fn)
+      return
+    assertTypes(fn, 'hook teardown', ['function'])
+    await fn()
+  }))
+}
+
 export async function runTest(test: Test) {
   if (test.mode !== 'run') {
     const { getSnapshotClient } = await import('../integrations/snapshot/chai')
@@ -157,7 +166,7 @@ export async function runTest(test: Test) {
 
     try {
       await callSuiteHook(test.suite, test, 'afterEach', [test.context, test.suite])
-      await Promise.all(beforeEachCleanups.map(i => i?.()))
+      await callCleanupHooks(beforeEachCleanups)
     }
     catch (e) {
       test.result.state = 'fail'
@@ -264,7 +273,7 @@ export async function runSuite(suite: Suite) {
       }
 
       await callSuiteHook(suite, suite, 'afterAll', [suite])
-      await Promise.all(beforeAllCleanups.map(i => i?.()))
+      await callCleanupHooks(beforeAllCleanups)
     }
     catch (e) {
       suite.result.state = 'fail'