Function getSpecificationsOptions

packages/vitest/src/utils/test-helpers.ts:6–48

(
  specifications: Array<TestSpecification>,
)

import { promises as fs } from 'node:fs'

export async function getSpecificationsOptions(
  specifications: Array<TestSpecification>,
): Promise<{
  environments: WeakMap<TestSpecification, ContextTestEnvironment>
  tags: WeakMap<TestSpecification, string[]>
}> {
  const environments = new WeakMap<TestSpecification, ContextTestEnvironment>()
  const cache = new Map<string, string>()
  const tags = new WeakMap<TestSpecification, string[]>()
  await Promise.all(
    specifications.map(async (spec) => {
      const { moduleId: filepath, project, pool } = spec
      // browser pool handles its own environment
      if (pool === 'browser') {
        return
      }

      // reuse if projects have the same test files
      let code = cache.get(filepath)
      if (!code) {
        code = await fs.readFile(filepath, 'utf-8').catch(() => '')
        cache.set(filepath, code)
      }

      const {
        env = project.config.environment || 'node',
        envOptions,
        tags: specTags = [],
      } = detectCodeBlock(code)
      tags.set(spec, specTags)

      const envKey = env === 'happy-dom' ? 'happyDOM' : env
      const environment: ContextTestEnvironment = {
        name: env as VitestEnvironment,
        options: envOptions
          ? ({ [envKey]: envOptions } as EnvironmentOptions)
          : null,
      }
      environments.set(spec, environment)
    }),
  )
  return { environments, tags }
}

export function detectCodeBlock(content: string): {
  env?: string