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.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263import type { VitestPackage } from './api/pkg'import type { ExtensionWorkerProcess } from './api/types'import type { WsConnectionMetadata } from './api/ws'import type { ExtensionDiagnostic } from './diagnostic'import type { TestTree } from './testTree'import { createServer } from 'node:http'import { pathToFileURL } from 'node:url'import getPort from 'get-port'import * as vscode from 'vscode'import { WebSocketServer } from 'ws'import { VitestFolderAPI } from './api'import { onWsConnection } from './api/ws'import { getConfig } from './config'import { workerPath } from './constants'import { log } from './log'import { TestRunner } from './runner'import { findNode } from './utils'
export async function debugTests( controller: vscode.TestController, tree: TestTree, pkg: VitestPackage, diagnostic: ExtensionDiagnostic | undefined,
request: vscode.TestRunRequest, token: vscode.CancellationToken, debugManager: DebugManager,) { 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, ...( config.debugOutFiles?.length ? { outFiles: config.debugOutFiles } : {} ), 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', }, }
if (debugManager.sessions.size) { await Promise.all( [...debugManager.sessions].map(session => vscode.debug.stopDebugging(session)), ).catch((error) => { log.error('[DEBUG] Failed to stop debugging sessions', error) }) }
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[] = []
wss.on( 'connection', ws => onWsConnection( ws, pkg, true, config.shellType, async (metadata) => { try { const api = new VitestFolderAPI(pkg, { ...metadata, process: new ExtensionDebugProcess( metadata, ), }) const runner = new TestRunner( controller, tree, api, diagnostic, ) disposables.push(api, runner)
token.onCancellationRequested(async () => { await metadata.rpc.close() })
await runner.runTests(request, token)
deferredPromise.resolve() } catch (err: any) { if (err.message.startsWith('[birpc] rpc is closed')) { deferredPromise.resolve() return }
deferredPromise.reject(err) } }, (err) => { if (err.message.startsWith('[birpc] rpc is closed')) { deferredPromise.resolve() return }
deferredPromise.reject(err) }, ), )
const onDidWorkerTerminate = vscode.debug.onDidTerminateDebugSession((session) => { const parent = session.parentSession
// dispose all test runners if (parent && parent.configuration.__name === 'Vitest') { disposables.reverse().forEach(d => d.dispose()) disposables.length = 0 } })
const onDidTerminate = vscode.debug.onDidTerminateDebugSession((session) => { if (session.configuration.__name !== 'Vitest') return server.close() onDidTerminate.dispose() onDidWorkerTerminate.dispose() })
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') { const executable = await findNode(pkg.cwd) return { runtimeExecutable: executable, runtimeArgs: execArgv, } } return { runtimeExecutable: 'node', runtimeArgs: execArgv, }}
class ExtensionDebugProcess implements ExtensionWorkerProcess { public id: number = Math.random() public closed = false
private _onDidExit = new vscode.EventEmitter<void>()
constructor(private metadata: WsConnectionMetadata) { // if websocket connection stopped working, close the debug session // otherwise it might hang indefinitely metadata.ws.on('close', () => { this.closed = true this._onDidExit.fire() this._onDidExit.dispose() }) }
async close() { if (this.metadata.rpc.$closed) { return } await this.metadata.rpc.close() }
onError() { // do nothing return () => {} }
onExit(listener: (code: number | null) => void) { const { dispose } = this._onDidExit.event(() => { listener(null) }) return dispose }}
export class DebugManager { public sessions: Set<vscode.DebugSession> = new Set()
constructor() { vscode.debug.onDidStartDebugSession((session) => { if (session.configuration.__name === 'Vitest') { this.sessions.add(session) } })
vscode.debug.onDidTerminateDebugSession((session) => { this.sessions.delete(session) }) }}