fix(@angular/build): perform testing module cleanup when using Vitest

alan-agius4 · alan-agius4 · commit 1cd65a08d527 · 2025-05-02T11:10:41.000Z
Ensure proper cleanup of the Angular testing module when running tests with Vitest. Closes: #30186 (cherry picked from commit 8e0b14f)
diff --git a/packages/angular/build/src/builders/unit-test/builder.ts b/packages/angular/build/src/builders/unit-test/builder.ts
@@ -124,8 +124,14 @@ export async function* execute(
     loadContent: async () => {
       const contents: string[] = [
         // Initialize the Angular testing environment
-        `import { getTestBed } from '@angular/core/testing';`,
+        `import { getTestBed, ɵgetCleanupHook as getCleanupHook } from '@angular/core/testing';`,
         `import { BrowserTestingModule, platformBrowserTesting } from '@angular/platform-browser/testing';`,
+        `import { beforeEach, afterEach } from 'vitest';`,
+        '',
+        // Same as https://github.com/angular/angular/blob/05a03d3f975771bb59c7eefd37c01fa127ee2229/packages/core/testing/src/test_hooks.ts#L21-L29
+        `beforeEach(getCleanupHook(false));`,
+        `afterEach(getCleanupHook(true));`,
+        '',
         `getTestBed().initTestEnvironment(BrowserTestingModule, platformBrowserTesting(), {`,
         `  errorOnUnknownElements: true,`,
         `  errorOnUnknownProperties: true,`,
