Skip to content

Commit

Permalink
fix: parse env options from comments (fix vitest-dev#1962)
Browse files Browse the repository at this point in the history
  • Loading branch information
nickmccurdy committed Sep 6, 2022
1 parent ffbb83f commit a79bcae
Showing 1 changed file with 15 additions and 32 deletions.
47 changes: 15 additions & 32 deletions packages/vitest/src/runtime/entry.ts
@@ -1,7 +1,6 @@
import { promises as fs } from 'fs'
import type { ResolvedConfig, VitestEnvironment } from '../types'
import type { EnvironmentOptions, ResolvedConfig, VitestEnvironment } from '../types'
import { getWorkerState, resetModules } from '../utils'
import { envs } from '../integrations/env'
import { setupGlobalEnv, withEnv } from './setup'
import { startTests } from './run'

Expand All @@ -22,45 +21,29 @@ export async function run(files: string[], config: ResolvedConfig): Promise<void
const filesWithEnv = await Promise.all(files.map(async (file) => {
const code = await fs.readFile(file, 'utf-8')
const env = code.match(/@(?:vitest|jest)-environment\s+?([\w-]+)\b/)?.[1] || config.environment || 'node'
const envOptions = JSON.parse(code.match(/@(?:vitest|jest)-environment-options\s+?(.+)/)?.[1] || 'null')
return {
file,
env: env as VitestEnvironment,
envOptions: envOptions as EnvironmentOptions | undefined,
}
}))

const filesByEnv = filesWithEnv.reduce((acc, { file, env }) => {
acc[env] ||= []
acc[env].push(file)
return acc
}, {} as Record<VitestEnvironment, string[]>)

const orderedEnvs = envs.concat(
Object.keys(filesByEnv).filter(env => !envs.includes(env)),
)

for (const env of orderedEnvs) {
const environment = env as VitestEnvironment
const files = filesByEnv[environment]

if (!files || !files.length)
continue

await withEnv(environment, config.environmentOptions || {}, async () => {
for (const file of files) {
// it doesn't matter if running with --threads
// if running with --no-threads, we usually want to reset everything before running a test
// but we have --isolate option to disable this
if (config.isolate) {
workerState.mockMap.clear()
resetModules(workerState.moduleCache, true)
}
for (const { file, env, envOptions } of filesWithEnv) {
await withEnv(env, envOptions || config.environmentOptions, async () => {
// it doesn't matter if running with --threads
// if running with --no-threads, we usually want to reset everything before running a test
// but we have --isolate option to disable this
if (config.isolate) {
workerState.mockMap.clear()
resetModules(workerState.moduleCache, true)
}

workerState.filepath = file
workerState.filepath = file

await startTests([file], config)
await startTests([file], config)

workerState.filepath = undefined
}
workerState.filepath = undefined
})
}
}

0 comments on commit a79bcae

Please sign in to comment.