feat(vite): add v8 coverage support to vitest generator

closed nrwl#17886
mpgregson · Jul 14, 2023 · 6769648 · 6769648
1 parent 4f9f04f
commit 6769648
Show file tree

Hide file tree

Showing 5 changed files with 46 additions and 13 deletions.
diff --git a/packages/vite/migrations.json b/packages/vite/migrations.json
@@ -75,6 +75,10 @@
         "@vitest/coverage-istanbul": {
           "version": "^0.32.0",
           "alwaysAddToPackageJson": false
+        },
+        "@vitest/coverage-v8": {
+          "version": "^0.32.0",
+          "alwaysAddToPackageJson": false
         }
       }
     },
@@ -104,6 +108,10 @@
         "@vitest/coverage-istanbul": {
           "version": "^0.31.0",
           "alwaysAddToPackageJson": false
+        },
+        "@vitest/coverage-v8": {
+          "version": "^0.31.0",
+          "alwaysAddToPackageJson": false
         }
       }
     },
@@ -150,6 +158,10 @@
         "@vitest/coverage-istanbul": {
           "version": "^0.25.8",
           "alwaysAddToPackageJson": false
+        },
+        "@vitest/coverage-v8": {
+          "version": "^0.25.8",
+          "alwaysAddToPackageJson": false
         }
       }
     }

diff --git a/packages/vite/src/generators/vitest/schema.d.ts b/packages/vite/src/generators/vitest/schema.d.ts
@@ -1,7 +1,7 @@
 export interface VitestGeneratorSchema {
   project: string;
   uiFramework: 'react' | 'none';
-  coverageProvider: 'c8' | 'istanbul';
+  coverageProvider: 'v8' | 'c8' | 'istanbul';
   inSourceTests?: boolean;
   skipViteConfig?: boolean;
   testTarget?: string;

diff --git a/packages/vite/src/generators/vitest/schema.json b/packages/vite/src/generators/vitest/schema.json
@@ -9,7 +9,9 @@
     "project": {
       "type": "string",
       "description": "The name of the project to test.",
-      "$default": { "$source": "projectName" }
+      "$default": {
+        "$source": "projectName"
+      }
     },
     "uiFramework": {
       "type": "string",
@@ -29,8 +31,8 @@
     },
     "coverageProvider": {
       "type": "string",
-      "enum": ["c8", "istanbul"],
-      "default": "c8",
+      "enum": ["v8", "c8", "istanbul"],
+      "default": "v8",
       "description": "Coverage provider to use."
     },
     "testTarget": {

diff --git a/packages/vite/src/generators/vitest/vitest-generator.ts b/packages/vite/src/generators/vitest/vitest-generator.ts
@@ -22,6 +22,7 @@ import initGenerator from '../init/init';
 import {
   vitestCoverageC8Version,
   vitestCoverageIstanbulVersion,
+  vitestCoverageV8Version,
 } from '../../utils/versions';
 
 import { addTsLibDependencies } from '@nx/js';
@@ -64,16 +65,14 @@ export async function vitestGenerator(
   createFiles(tree, schema, root);
   updateTsConfig(tree, schema, root);
 
+  const coverageProviderDependency = getCoverageProviderDependency(
+    schema.coverageProvider
+  );
+
   const installCoverageProviderTask = addDependenciesToPackageJson(
     tree,
     {},
-    schema.coverageProvider === 'istanbul'
-      ? {
-          '@vitest/coverage-istanbul': vitestCoverageIstanbulVersion,
-        }
-      : {
-          '@vitest/coverage-c8': vitestCoverageC8Version,
-        }
+    coverageProviderDependency
   );
   tasks.push(installCoverageProviderTask);
 
@@ -150,5 +149,24 @@ function createFiles(
   });
 }
 
+function getCoverageProviderDependency(
+  coverageProvider: VitestGeneratorSchema['coverageProvider']
+) {
+  switch (coverageProvider) {
+    case 'c8':
+      return {
+        '@vitest/coverage-c8': vitestCoverageC8Version,
+      };
+    case 'istanbul':
+      return {
+        '@vitest/coverage-istanbul': vitestCoverageIstanbulVersion,
+      };
+    default:
+      return {
+        '@vitest/coverage-v8': vitestCoverageV8Version,
+      };
+  }
+}
+
 export default vitestGenerator;
 export const vitestSchematic = convertNxGenerator(vitestGenerator);
diff --git a/packages/vite/src/utils/versions.ts b/packages/vite/src/utils/versions.ts
@@ -11,5 +11,6 @@ export const happyDomVersion = '~9.20.3';
 export const edgeRuntimeVmVersion = '~3.0.2';
 
 // Coverage providers
-export const vitestCoverageC8Version = '~0.32.0';
-export const vitestCoverageIstanbulVersion = '~0.32.0';
+export const vitestCoverageC8Version = '^0.32.0';
+export const vitestCoverageV8Version = '^0.32.0';
+export const vitestCoverageIstanbulVersion = '^0.32.0';