Function collectTests

packages/runner/src/collect.ts:23–148

(
  specs: string[] | FileSpecification[],
  runner: VitestRunner,
)

const now = globalThis.performance ? globalThis.performance.now.bind(globalThis.performance) : Date.now

export async function collectTests(
  specs: string[] | FileSpecification[],
  runner: VitestRunner,
): Promise<File[]> {
  const files: File[] = []

  const config = runner.config
  const $ = runner.trace!
  let defaultTagsFilter: ((testTags: string[]) => boolean) | undefined

  for (const spec of specs) {
    const filepath = typeof spec === 'string' ? spec : spec.filepath
    await $(
      'collect_spec',
      { 'code.file.path': filepath },
      async () => {
        runner._currentSpecification = typeof spec === 'string' ? { filepath: spec } : spec

        const testLocations = typeof spec === 'string' ? undefined : spec.testLocations
        const testNamePattern = typeof spec === 'string' ? undefined : spec.testNamePattern
        const testIds = typeof spec === 'string' ? undefined : spec.testIds
        const testTagsFilter = typeof spec === 'object' && spec.testTagsFilter
          ? createTagsFilter(spec.testTagsFilter, config.tags)
          : undefined

        const fileTags: string[] = typeof spec === 'string' ? [] : (spec.fileTags || [])

        const file = createFileTask(filepath, config.root, config.name, runner.pool, runner.viteEnvironment)
        file.tags = fileTags
        file.shuffle = config.sequence.shuffle

        try {
          validateTags(runner.config, fileTags)

          runner.onCollectStart?.(file)

          clearCollectorContext(file, runner)

          const setupFiles = toArray(config.setupFiles)
          if (setupFiles.length) {
            const setupStart = now()
            await runSetupFiles(config, setupFiles, runner)
            const setupEnd = now()
            file.setupDuration = setupEnd - setupStart
          }
          else {
            file.setupDuration = 0
          }

          const collectStart = now()

          await runner.importFile(filepath, 'collect')

          const durations = runner.getImportDurations?.()
          if (durations) {
            file.importDurations = durations
          }