Something went wrong. Try again.
[READ-ONLY] Mirror of https://github.com/vitest-dev/vscode. VS Code extension for Vitest vitest.dev/vscode
testing vitest vscode
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175import { createServer } from 'node:http'import { pathToFileURL } from 'node:url'import * as vscode from 'vscode'import { WebSocketServer } from 'ws'import getPort from 'get-port'import type { ResolvedMeta } from '../api'import { VitestFolderAPI } from '../api'import type { VitestPackage } from '../api/pkg'import type { TestTree } from '../testTree'import { log } from '../log'import { getConfig } from '../config'import { TestRunner } from '../runner/runner'import { workerPath } from '../constants'import { waitForWsResolvedMeta } from '../api/ws'
export async function debugTests( controller: vscode.TestController, tree: TestTree, pkg: VitestPackage,
request: vscode.TestRunRequest, token: vscode.CancellationToken,) { const port = await getPort() const server = createServer().listen(port) const wss = new WebSocketServer({ server }) const wsAddress = `ws://localhost:${port}`
const config = getConfig(pkg.folder) const deferredPromise = Promise.withResolvers<void>()
const { runtimeArgs, runtimeExecutable } = await getRuntimeOptions(pkg) const env = config.env || {} const logLevel = config.logLevel
log.info('[DEBUG]', 'Starting debugging session', runtimeExecutable, ...(runtimeArgs || []))
const debugConfig = { __name: 'Vitest', type: config.shellType === 'terminal' ? 'node-terminal' : 'pwa-node', request: 'launch', name: 'Debug Tests', autoAttachChildProcesses: true, skipFiles: config.debugExclude, smartStep: true, ...(config.shellType === 'terminal' ? { command: `${runtimeExecutable} ${workerPath}`, } : { program: workerPath, runtimeArgs, runtimeExecutable, } ), cwd: pkg.cwd, env: { ...process.env, ...env, VITEST_VSCODE_LOG: env.VITEST_VSCODE_LOG ?? process.env.VITEST_VSCODE_LOG ?? logLevel, VITEST_VSCODE: 'true', VITEST_WS_ADDRESS: wsAddress, // same env var as `startVitest` // https://github.com/vitest-dev/vitest/blob/5c7e9ca05491aeda225ce4616f06eefcd068c0b4/packages/vitest/src/node/cli/cli-api.ts TEST: 'true', VITEST: 'true', NODE_ENV: env.NODE_ENV ?? process.env.NODE_ENV ?? 'test', }, }
vscode.debug.startDebugging( pkg.folder, debugConfig, { suppressDebugView: true }, ).then( (fulfilled) => { if (fulfilled) { log.info('[DEBUG] Debugging started') } else { deferredPromise.reject(new Error('Failed to start debugging. See output for more information.')) log.error('[DEBUG] Debugging failed') } }, (err) => { deferredPromise.reject(new Error('Failed to start debugging', { cause: err })) log.error('[DEBUG] Start debugging failed') log.error(err.toString()) }, )
const disposables: vscode.Disposable[] = []
const onDidStart = vscode.debug.onDidStartDebugSession(async (session) => { if (session.configuration.__name !== 'Vitest') return if (token.isCancellationRequested) { vscode.debug.stopDebugging(session) return } let vitest!: ResolvedMeta
try { vitest = await waitForWsResolvedMeta(wss, pkg, true, 'child_process') const api = new VitestFolderAPI(pkg, vitest) const runner = new TestRunner( controller, tree, api, ) disposables.push(api, runner)
token.onCancellationRequested(async () => { await vitest.rpc.close() await vscode.debug.stopDebugging(session) })
await runner.runTests(request, token)
deferredPromise.resolve() } catch (err: any) { if (err.message.startsWith('[birpc] rpc is closed')) { deferredPromise.resolve() return }
deferredPromise.reject(err) }
if (!token.isCancellationRequested) { await vitest?.rpc.close() await vscode.debug.stopDebugging(session) } })
const onDidTerminate = vscode.debug.onDidTerminateDebugSession((session) => { if (session.configuration.__name !== 'Vitest') return disposables.reverse().forEach(d => d.dispose()) server.close() })
disposables.push(onDidStart, onDidTerminate)
await deferredPromise.promise}
async function getRuntimeOptions(pkg: VitestPackage) { const config = getConfig(pkg.folder)
const runtimeArgs = config.nodeExecArgs || [] const pnpLoader = pkg.loader const pnp = pkg.pnp const execArgv = pnpLoader && pnp ? [ '--require', pnp, '--experimental-loader', pathToFileURL(pnpLoader).toString(), ...runtimeArgs, ] : runtimeArgs if (config.shellType === 'child_process') { return { runtimeExecutable: config.nodeExecutable || 'node', runtimeArgs: execArgv, } } return { runtimeExecutable: 'node', runtimeArgs: execArgv, }}