fix(@schematics/angular): support testRunner option in library schematic

This commit introduces the `testRunner` option to the library schematic, allowing users to explicitly select between `vitest` and `karma` as their test runner.
Additionally, the `ng-new` schematic has been updated to configure the default `testRunner` for libraries in `angular.json` based on the workspace creation options.

Closes #31887

Author: alan-agius4

packages/schematics/angular/application/index.ts

@@ -392,17 +392,15 @@ function addAppToWorkspaceFile(options: ApplicationOptions, appDir: string): Rul
       test:
         options.skipTests || options.minimal
           ? undefined
-          : options.testRunner === 'vitest'
-            ? {
-                builder: Builders.BuildUnitTest,
-                options: {},
-              }
-            : {
-                builder: Builders.BuildUnitTest,
-                options: {
-                  runner: 'karma',
-                },
-              },
+          : {
+              builder: Builders.BuildUnitTest,
+              options:
+                options.testRunner === 'vitest'
+                  ? {}
+                  : {
+                      runner: 'karma',
+                    },
+            },
     },
   };
 

packages/schematics/angular/library/index.ts

@@ -91,7 +91,6 @@ function addLibToWorkspaceFile(
   projectRoot: string,
   projectName: string,
   hasZoneDependency: boolean,
-  hasVitest: boolean,
 ): Rule {
   return updateWorkspace((workspace) => {
     workspace.projects.add({
@@ -113,20 +112,21 @@ function addLibToWorkspaceFile(
             },
           },
         },
-        test: hasVitest
-          ? {
-              builder: Builders.BuildUnitTest,
-              options: {
-                tsConfig: `${projectRoot}/tsconfig.spec.json`,
+        test:
+          options.testRunner === 'vitest'
+            ? {
+                builder: Builders.BuildUnitTest,
+                options: {
+                  tsConfig: `${projectRoot}/tsconfig.spec.json`,
+                },
+              }
+            : {
+                builder: Builders.BuildKarma,
+                options: {
+                  tsConfig: `${projectRoot}/tsconfig.spec.json`,
+                  polyfills: hasZoneDependency ? ['zone.js', 'zone.js/testing'] : undefined,
+                },
               },
-            }
-          : {
-              builder: Builders.BuildKarma,
-              options: {
-                tsConfig: `${projectRoot}/tsconfig.spec.json`,
-                polyfills: hasZoneDependency ? ['zone.js', 'zone.js/testing'] : undefined,
-              },
-            },
       },
     });
   });
@@ -158,7 +158,6 @@ export default function (options: LibraryOptions): Rule {
 
     const distRoot = `dist/${folderName}`;
     const sourceDir = `${libDir}/src/lib`;
-    const hasVitest = getDependency(host, 'vitest') !== null;
 
     const templateSource = apply(url('./files'), [
       applyTemplates({
@@ -172,7 +171,7 @@ export default function (options: LibraryOptions): Rule {
         angularLatestVersion: latestVersions.Angular.replace(/~|\^/, ''),
         tsLibLatestVersion: latestVersions['tslib'].replace(/~|\^/, ''),
         folderName,
-        testTypesPackage: hasVitest ? 'vitest/globals' : 'jasmine',
+        testTypesPackage: options.testRunner === 'vitest' ? 'vitest/globals' : 'jasmine',
       }),
       move(libDir),
     ]);
@@ -181,7 +180,7 @@ export default function (options: LibraryOptions): Rule {
 
     return chain([
       mergeWith(templateSource),
-      addLibToWorkspaceFile(options, libDir, packageName, hasZoneDependency, hasVitest),
+      addLibToWorkspaceFile(options, libDir, packageName, hasZoneDependency),
       options.skipPackageJson ? noop() : addDependenciesToPackageJson(!!options.skipInstall),
       options.skipTsConfig ? noop() : updateTsConfig(packageName, './' + distRoot),
       options.skipTsConfig

packages/schematics/angular/library/index_spec.ts

@@ -407,11 +407,11 @@ describe('Library Schematic', () => {
     expect(workspace.projects.foo.architect.build.builder).toBe('@angular/build:ng-packagr');
   });
 
-  it(`should add 'karma' test builder`, async () => {
+  it(`should add 'unit-test' test builder`, async () => {
     const tree = await schematicRunner.runSchematic('library', defaultOptions, workspaceTree);
 
     const workspace = JSON.parse(tree.readContent('/angular.json'));
-    expect(workspace.projects.foo.architect.test.builder).toBe('@angular/build:karma');
+    expect(workspace.projects.foo.architect.test.builder).toBe('@angular/build:unit-test');
   });
 
   it(`should add 'unit-test' test builder`, async () => {

packages/schematics/angular/library/schema.json

@@ -53,6 +53,12 @@
       "type": "boolean",
       "default": true,
       "x-user-analytics": "ep.ng_standalone"
+    },
+    "testRunner": {
+      "description": "The unit testing runner to use.",
+      "type": "string",
+      "enum": ["vitest", "karma"],
+      "default": "vitest"
     }
   },
   "required": ["name"]

packages/schematics/angular/ng-new/index.ts

@@ -67,16 +67,21 @@ export default function (options: NgNewOptions): Rule {
     mergeWith(
       apply(empty(), [
         schematic('workspace', workspaceOptions),
-        options.createApplication ? schematic('application', applicationOptions) : noop,
-        schematic('ai-config', {
-          tool: options.aiConfig?.length ? options.aiConfig : undefined,
-        }),
         (tree: Tree) => {
           if (options.testRunner === 'karma') {
             const file = new JSONFile(tree, 'angular.json');
-            file.modify(['schematics', '@schematics/angular:application', 'testRunner'], 'karma');
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const schematics = file.get(['schematics']) ?? ({} as any);
+            (schematics['@schematics/angular:application'] ??= {}).testRunner = 'karma';
+            (schematics['@schematics/angular:library'] ??= {}).testRunner = 'karma';
+
+            file.modify(['schematics'], schematics);
           }
         },
+        options.createApplication ? schematic('application', applicationOptions) : noop,
+        schematic('ai-config', {
+          tool: options.aiConfig?.length ? options.aiConfig : undefined,
+        }),
         move(options.directory),
       ]),
     ),

packages/schematics/angular/ng-new/index_spec.ts

@@ -183,6 +183,7 @@ describe('Ng New Schematic', () => {
 
     const { schematics } = JSON.parse(tree.readContent('/bar/angular.json'));
     expect(schematics['@schematics/angular:application'].testRunner).toBe('karma');
+    expect(schematics['@schematics/angular:library'].testRunner).toBe('karma');
   });
 
   it(`should not add type to class name when file name style guide is '2016'`, async () => {