fix(@schematics/angular): add missing imports for lifecycle hooks in jasmine-vitest migration

Author: cexbrayat

packages/schematics/angular/refactor/jasmine-vitest/test-file-transformer.ts

@@ -183,7 +183,15 @@ export function transformJasmineToVitest(
       if (ts.isCallExpression(transformedNode)) {
         if (options.addImports && ts.isIdentifier(transformedNode.expression)) {
           const name = transformedNode.expression.text;
-          if (name === 'describe' || name === 'it' || name === 'expect') {
+          if (
+            name === 'describe' ||
+            name === 'it' ||
+            name === 'expect' ||
+            name === 'beforeEach' ||
+            name === 'afterEach' ||
+            name === 'beforeAll' ||
+            name === 'afterAll'
+          ) {
             addVitestValueImport(pendingVitestValueImports, name);
           }
         }

packages/schematics/angular/refactor/jasmine-vitest/test-file-transformer_add-imports_spec.ts

@@ -81,5 +81,41 @@ describe('Jasmine to Vitest Transformer', () => {
       `;
       await expectTransformation(input, expected, true);
     });
+
+    it('should add imports for beforeEach and afterEach when addImports is true', async () => {
+      const input = `
+        describe('My Suite', () => {
+          beforeEach(() => {});
+          afterEach(() => {});
+        });
+      `;
+      const expected = `
+        import { afterEach, beforeEach, describe } from 'vitest';
+
+        describe('My Suite', () => {
+          beforeEach(() => {});
+          afterEach(() => {});
+        });
+      `;
+      await expectTransformation(input, expected, true);
+    });
+
+    it('should add imports for beforeAll and afterAll when addImports is true', async () => {
+      const input = `
+        describe('My Suite', () => {
+          beforeAll(() => {});
+          afterAll(() => {});
+        });
+      `;
+      const expected = `
+        import { afterAll, beforeAll, describe } from 'vitest';
+
+        describe('My Suite', () => {
+          beforeAll(() => {});
+          afterAll(() => {});
+        });
+      `;
+      await expectTransformation(input, expected, true);
+    });
   });
 });