/
coverage.ts
76 lines (53 loc) · 2.34 KB
/
coverage.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
import type { CoverageOptions, CoverageProvider, CoverageProviderModule } from '../types'
interface Loader {
executeId: (id: string) => Promise<{ default: CoverageProviderModule }>
}
export const CoverageProviderMap: Record<string, string> = {
v8: '@vitest/coverage-v8',
istanbul: '@vitest/coverage-istanbul',
}
async function resolveCoverageProviderModule(options: CoverageOptions | undefined, loader: Loader) {
if (!options?.enabled || !options.provider)
return null
const provider = options.provider
if (provider === 'v8' || provider === 'istanbul') {
const { default: coverageModule } = await loader.executeId(CoverageProviderMap[provider])
if (!coverageModule)
throw new Error(`Failed to load ${CoverageProviderMap[provider]}. Default export is missing.`)
return coverageModule
}
let customProviderModule
try {
customProviderModule = await loader.executeId(options.customProviderModule)
}
catch (error) {
throw new Error(`Failed to load custom CoverageProviderModule from ${options.customProviderModule}`, { cause: error })
}
if (customProviderModule.default == null)
throw new Error(`Custom CoverageProviderModule loaded from ${options.customProviderModule} was not the default export`)
return customProviderModule.default
}
export async function getCoverageProvider(options: CoverageOptions | undefined, loader: Loader): Promise<CoverageProvider | null> {
const coverageModule = await resolveCoverageProviderModule(options, loader)
if (coverageModule)
return coverageModule.getProvider()
return null
}
export async function startCoverageInsideWorker(options: CoverageOptions | undefined, loader: Loader) {
const coverageModule = await resolveCoverageProviderModule(options, loader)
if (coverageModule)
return coverageModule.startCoverage?.()
return null
}
export async function takeCoverageInsideWorker(options: CoverageOptions | undefined, loader: Loader) {
const coverageModule = await resolveCoverageProviderModule(options, loader)
if (coverageModule)
return coverageModule.takeCoverage?.()
return null
}
export async function stopCoverageInsideWorker(options: CoverageOptions | undefined, loader: Loader) {
const coverageModule = await resolveCoverageProviderModule(options, loader)
if (coverageModule)
return coverageModule.stopCoverage?.()
return null
}