feat(vitest): call setup and teardown when running suite

GuillaumeLagrange · GuillaumeLagrange · commit c47550452558 · 2025-07-18T16:04:58.000+02:00
diff --git 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) {

Original file line number	Diff line number	Diff line change
`@@ -83,6 +83,8 @@ async function runBench(benchmark: Benchmark, currentSuiteName: string) {`
`83`	`83`	`}`
`84`	`84`
`85`	`85`	`async function runBenchmarkSuite(suite: Suite, parentSuiteName?: string) {`
	`86`	`+ await callSuiteHook(suite, suite, "setup");`
	`87`	`+`
`86`	`88`	`const currentSuiteName = parentSuiteName`
`87`	`89`	`? parentSuiteName + "::" + suite.name`
`88`	`90`	`: suite.name;`
`@@ -108,6 +110,8 @@ async function runBenchmarkSuite(suite: Suite, parentSuiteName?: string) {`
`108`	`110`	`if (parentSuiteName !== undefined) {`
`109`	`111`	`await callSuiteHook(suite, suite, "afterAll");`
`110`	`112`	`}`
	`113`	`+`
	`114`	`+ await callSuiteHook(suite, suite, "teardown");`
`111`	`115`	`}`
`112`	`116`
`113`	`117`	`function patchRootSuiteWithFullFilePath(suite: Suite) {`