From c47550452558a2829653b46b4b3cc7c7bf7e702c Mon Sep 17 00:00:00 2001
From: Guillaume Lagrange <guillaume@glagrange.eu>
Date: Fri, 18 Jul 2025 16:04:58 +0200
Subject: [PATCH] feat(vitest): call setup and teardown when running suite

---
 packages/vitest-plugin/src/runner.ts | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/packages/vitest-plugin/src/runner.ts b/packages/vitest-plugin/src/runner.ts
index 780b3482..ed9fcdff 100644
--- a/packages/vitest-plugin/src/runner.ts
+++ b/packages/vitest-plugin/src/runner.ts
@@ -83,6 +83,8 @@ async function runBench(benchmark: Benchmark, currentSuiteName: string) {
 }
 
 async function runBenchmarkSuite(suite: Suite, parentSuiteName?: string) {
+  await callSuiteHook(suite, suite, "setup");
+
   const currentSuiteName = parentSuiteName
     ? parentSuiteName + "::" + suite.name
     : suite.name;
@@ -108,6 +110,8 @@ async function runBenchmarkSuite(suite: Suite, parentSuiteName?: string) {
   if (parentSuiteName !== undefined) {
     await callSuiteHook(suite, suite, "afterAll");
   }
+
+  await callSuiteHook(suite, suite, "teardown");
 }
 
 function patchRootSuiteWithFullFilePath(suite: Suite) {