Skip to content

Custom Pool

WARNING

这是高级 API。如果你只需要运行测试,你可能不需要这个。它主要被库的作者使用。

Vitest 在默认情况下以多种方式运行测试:

  • threads 使用 node:worker_threads 运行测试(通过新的 worker 上下文提供隔离)
  • forks 使用 node:child_process 运行测试(通过新的 child_process.fork 进程提供隔离)
  • vmThreads 使用 node:worker_threads 运行测试(但是通过 vm 模块而不是新的 worker 上下文提供隔离)
  • browser 使用浏览器提供程序运行测试
  • typescript 在测试中运行类型检查

你可以通过指定文件路径来提供自己的池:

ts
import { defineConfig } from 'vitest/config'

export default defineConfig({
  test: {
    // 默认情况下,将使用自定义池运行每个文件
    pool: './my-custom-pool.ts',
    // 可以使用 `poolOptions` 对象提供选项
    poolOptions: {
      myCustomPool: {
        customProperty: true,
      },
    },
    // 还可以为文件子集指定池
    poolMatchGlobs: [['**/*.custom.test.ts', './my-custom-pool.ts']],
  },
})

API

pool 选项中指定的文件应该导出一个函数(可以是异步的),该函数接受 Vitest 接口作为其第一个选项。这个函数需要返回一个与 ProcessPool 接口匹配的对象:

ts
import { ProcessPool, WorkspaceProject } from 'vitest/node'

export interface ProcessPool {
  name: string
  runTests: (
    files: [project: WorkspaceProject, testFile: string][],
    invalidates?: string[]
  ) => Promise<void>
  collectTests: (
    files: [project: WorkspaceProject, testFile: string][],
    invalidates?: string[]
  ) => Promise<void>
  close?: () => Promise<void>
}

这个函数只会被调用一次(除非服务器配置被更新),通常最好在这个函数内初始化测试所需的一切,并在调用 runTests 时重复使用它。

Vitest 在安排运行新测试时调用 runTest。如果 files 为空,将不会调用它。第一个参数是一个元组数组:第一个元素是对工作区项目的引用,第二个元素是测试文件的绝对路径。在调用 runTests 之前,文件将使用 sequencer 进行排序。可能(但不太可能)会有相同的文件出现两次,但它们将始终属于不同的项目 - 这是通过 vitest.workspace.ts 配置实现的。

Vitest 会等到 runTests 执行完毕后才结束运行(即只有在 runTests 解决后才会触发 onFinished)。

如果你正在使用自定义池,需要自行提供测试文件及其结果 - 可以参考 vitest.state(最重要的是 collectFilesupdateTasks)。Vitest 使用 @vitest/runner 包中的 startTests 函数来执行这些操作。

如果通过 CLI 命令调用 vitest.collectvitest list,则 Vitest 将调用 collectTests。它的工作方式与 runTests 相同,但你不必运行测试回调,只需通过调用 vitest.state.collectFiles(files) 来报告它们的任务。

要在不同的进程之间进行通信,可以使用 vitest/node 中的 createMethodsRPC 创建方法对象,并使用你喜欢的任何形式的通信。例如,要将 WebSockets 与 birpc 一起使用,你可以编写以下内容:

ts
import { createBirpc } from 'birpc'
import { parse, stringify } from 'flatted'
import { WorkspaceProject, createMethodsRPC } from 'vitest/node'

function createRpc(project: WorkspaceProject, wss: WebSocketServer) {
  return createBirpc(createMethodsRPC(project), {
    post: msg => wss.send(msg),
    on: fn => wss.on('message', fn),
    serialize: stringify,
    deserialize: parse,
  })
}

为了确保收集每个测试,你可以调用 ctx.state.collectFiles 并将其交给 Vitest 报告器:

ts
async function runTests(project: WorkspaceProject, tests: string[]) {
  // ... 运行测试,放入 `files` 和 `tasks` 中
  const methods = createMethodsRPC(project)
  await methods.onCollected(files)
  // 大多数报告都依赖于在 `onTaskUpdate` 中更新结果
  await methods.onTaskUpdate(tasks)
}

可以在 pool/custom-pool.ts 中看到一个简单的示例。

Released under the MIT License.