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.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386import { lstatSync, readlinkSync } from 'node:fs'import { resolve } from 'node:path'import * as vscode from 'vscode'import { basename, dirname, normalize } from 'pathe'import type { RunnerTask, RunnerTestFile } from 'vitest'import { TestCase, TestFile, TestFolder, TestSuite, getTestData } from './testTreeData'import { log } from './log'import type { VitestFolderAPI } from './api'import { ExtensionWatcher } from './watcher'
// testItem -> vscode.TestItem// testData -> our wrapper// task -> vitest.Task
export class TestTree extends vscode.Disposable { private flatTestItems = new Map<string, vscode.TestItem>() private fileItems: Map<string, vscode.TestItem> = new Map() private folderItems: Map<string, vscode.TestItem> = new Map()
// it's possible to have several test items for the same file // file test items have the project name in their id, so we need a separate map // to store all of them private testItemsByFile = new Map<string, vscode.TestItem[]>() private testFiles = new Set<string>()
private watcher: ExtensionWatcher
constructor( private readonly controller: vscode.TestController, private readonly loaderItem: vscode.TestItem, ) { super(() => { this.folderItems.clear() this.fileItems.clear() this.flatTestItems.clear() this.testItemsByFile.clear() this.watcher.dispose() }) this.watcher = new ExtensionWatcher(this) }
public getFileTestItems(fsPath: string) { return this.testItemsByFile.get(normalize(fsPath)) || [] }
public getAllFileItems() { return Array.from(this.fileItems.values()) }
public reset(workspaceFolders: vscode.WorkspaceFolder[]) { this.folderItems.clear() this.testItemsByFile.clear() this.fileItems.clear() this.flatTestItems.clear() this.watcher.reset()
this.loaderItem.busy = true
if (workspaceFolders.length === 1) { const rootItem = this.getOrCreateInlineFolderItem(workspaceFolders[0].uri) rootItem.children.replace([this.loaderItem]) } else { const folderItems = workspaceFolders.map((x) => { const item = this.getOrCreateWorkspaceFolderItem(x.uri) item.children.replace([]) item.busy = true return item }) this.controller.items.replace(folderItems) } }
async discoverAllTestFiles(api: VitestFolderAPI, files: [project: string, file: string][]) { const folderItem = this.folderItems.get(normalize(api.workspaceFolder.uri.fsPath)) if (folderItem) folderItem.busy = false
for (const [project, file] of files) this.getOrCreateFileTestItem(api, project, file)
return files }
// in cases where there is only a single workspace, we don't show it as a folder // this inline folder is required for "createFolderItem" to properly resolve the parent, // otherwise it will go into an infinite loop getOrCreateInlineFolderItem(folderUri: vscode.Uri) { let id = normalize(folderUri.fsPath) const cached = this.folderItems.get(id) if (cached) return cached const stats = lstatSync(folderUri.fsPath) if (stats.isSymbolicLink()) { const actualPath = readlinkSync(folderUri.fsPath) const dir = dirname(folderUri.fsPath) id = resolve(dir, actualPath) folderUri = vscode.Uri.file(id) } const cachedSymlink = this.folderItems.get(id) if (cachedSymlink) return cachedSymlink const item: vscode.TestItem = { id: folderUri.toString(), children: this.controller.items, uri: folderUri, label: '<root>', canResolveChildren: false, busy: false, parent: undefined, tags: [], range: undefined, error: undefined, } TestFolder.register(item) this.folderItems.set(id, item) return item }
getOrCreateWorkspaceFolderItem(folderUri: vscode.Uri) { const folderId = normalize(folderUri.fsPath) const cached = this.folderItems.get(folderId) if (cached) return cached
const folderItem = this._createFolderItem(folderUri) this.folderItems.set(folderId, folderItem) return folderItem }
getOrCreateFileTestItem(api: VitestFolderAPI, project: string, file: string) { const normalizedFile = normalize(file) const fileId = `${normalizedFile}${project}` const cached = this.fileItems.get(fileId) if (cached) return cached
const fileUri = vscode.Uri.file(resolve(file)) const parentItem = this.getOrCreateFolderTestItem(api, dirname(file)) const label = `${basename(file)}${project ? ` [${project}]` : ''}` const testFileItem = this.controller.createTestItem( fileId, label, fileUri, ) // "description" looks nicer in the test explorer, // but it's not displayed in the gutter icon // testFileItem.description = project testFileItem.tags = [api.tag] testFileItem.canResolveChildren = true TestFile.register( testFileItem, parentItem, normalizedFile, api, project, ) parentItem.children.add(testFileItem) this.fileItems.set(fileId, testFileItem) const cachedItems = this.testItemsByFile.get(normalizedFile) || [] cachedItems.push(testFileItem) this.testItemsByFile.set(normalizedFile, cachedItems) this.testFiles.add(fileUri.fsPath) vscode.commands.executeCommand( 'setContext', 'vitest.testFiles', Array.from(this.testFiles), )
return testFileItem }
getOrCreateFolderTestItem(api: VitestFolderAPI, normalizedFolder: string) { const cached = this.folderItems.get(normalizedFolder) if (cached) { if (!cached.tags.includes(api.tag)) cached.tags = [...cached.tags, api.tag] return cached }
const folderUri = vscode.Uri.file(normalizedFolder) const parentItem = this.getOrCreateFolderTestItem(api, dirname(normalizedFolder)) const folderItem = this._createFolderItem(folderUri, parentItem) folderItem.tags = [api.tag] parentItem.children.add(folderItem) this.folderItems.set(normalizedFolder, folderItem) return folderItem }
private _createFolderItem(folderUri: vscode.Uri, parentItem?: vscode.TestItem) { const folderFsPath = normalize(folderUri.fsPath) const folderItem = this.controller.createTestItem( folderFsPath, basename(folderFsPath), folderUri, ) TestFolder.register(folderItem, parentItem) folderItem.canResolveChildren = false return folderItem }
async watchTestFilesInWorkspace(api: VitestFolderAPI, testFiles: [prject: string, file: string][]) { await this.discoverAllTestFiles(api, testFiles) this.watcher.watchTestFilesInWorkspace(api) }
public removeFile(filepath: string) { const items = this.testItemsByFile.get(normalize(filepath)) items?.forEach(item => this.recursiveDelete(item)) }
private recursiveDelete(item: vscode.TestItem) { if (!item.parent) return item.parent.children.delete(item.id) this.flatTestItems.delete(item.id) const data = getTestData(item)
if (data instanceof TestFile) { this.testItemsByFile.delete(data.filepath) this.fileItems.delete(item.id) } if (data instanceof TestFolder) this.folderItems.delete(item.id)
if (!item.parent.children.size) this.recursiveDelete(item.parent) }
public getAPIFromTestItem(testItem: vscode.TestItem) { return getAPIFromTestItem(testItem) }
async discoverFileTests(testItem: vscode.TestItem) { const data = getTestData(testItem) if (!(data instanceof TestFile)) return const api = data.api if (!api) { log.error(`Cannot find collector for ${testItem.uri?.fsPath}`) return null } testItem.busy = true try { await api.collectTests(data.project, testItem.uri!.fsPath) return testItem } finally { testItem.busy = false } }
public getTestItemByTaskId(taskId: string): vscode.TestItem | undefined { const testItem = this.flatTestItems.get(taskId) if (!testItem) return undefined return testItem || undefined }
public getTestItemByTask(task: RunnerTask): vscode.TestItem | null { const cachedItem = this.flatTestItems.get(task.id) if (cachedItem) return cachedItem if ('filepath' in task && task.filepath) { const testItem = this.fileItems.get(`${task.filepath}${task.projectName || ''}`) return testItem || null } return null }
public getFolderFiles(folder: vscode.TestItem): vscode.TestItem[] { const files: vscode.TestItem[] = [] for (const [_, item] of folder.children) { const data = getTestData(item) if (data instanceof TestFile) files.push(item) else if (data instanceof TestFolder) files.push(...this.getFolderFiles(item)) } return files }
collectFile(api: VitestFolderAPI, file: RunnerTestFile) { const fileTestItem = this.getOrCreateFileTestItem(api, file.projectName || '', file.filepath) fileTestItem.error = undefined this.flatTestItems.set(file.id, fileTestItem) const data = getTestData(fileTestItem) as TestFile this.collectTasks(api.tag, data, file.tasks, fileTestItem) if (file.result?.errors) { const error = file.result.errors.map(error => error.stack || error.message).join('\n') fileTestItem.error = error log.error(`Error in ${file.filepath}`, error) } else if (!file.tasks.length) { fileTestItem.error = `No tests found in ${file.filepath}` } fileTestItem.canResolveChildren = false }
collectTasks(tag: vscode.TestTag, fileData: TestFile, tasks: RunnerTask[], parent: vscode.TestItem) { for (const task of tasks) { const cachedItem = this.flatTestItems.get(task.id) // suite became a test or vice versa if (cachedItem) { const data = getTestData(cachedItem) const taskType = task.type === 'custom' ? 'test' : task.type if (data.type !== taskType) { parent.children.delete(cachedItem.id) this.flatTestItems.delete(task.id) } }
const testItem = this.flatTestItems.get(task.id) || this.controller.createTestItem( task.id, task.name, parent.uri, ) testItem.tags = Array.from(new Set([...parent.tags, tag])) testItem.error = undefined testItem.label = task.name const location = task.location if (location) { const position = new vscode.Position(location.line - 1, location.column) testItem.range = new vscode.Range(position, position) } else { log.error(`Cannot find location for "${testItem.label}". Using "id" to sort instead.`) testItem.sortText = task.id } // dynamic exists only during browser collection // see src/worker/collect.ts:172 const isDynamic = (task as any).dynamic if (task.type === 'suite') TestSuite.register(testItem, parent, fileData, isDynamic) else if (task.type === 'test' || task.type === 'custom') TestCase.register(testItem, parent, fileData, isDynamic)
this.flatTestItems.set(task.id, testItem) parent.children.add(testItem)
// errors during collection are not test failures, they need to be // displayed as errors in the tree if (task.result?.errors) { const error = task.result.errors.map(error => error.stack).join('\n') testItem.error = error }
if ('tasks' in task) this.collectTasks(tag, fileData, task.tasks, testItem) }
// remove tasks that are no longer present const ids = new Set(tasks.map(x => x.id)) parent.children.forEach((child) => { if (!ids.has(child.id)) parent.children.delete(child.id) }) }}
function getAPIFromFolder(folder: vscode.TestItem): VitestFolderAPI | null { const data = getTestData(folder) if (data instanceof TestFile) return data.api if (!(data instanceof TestFolder)) return null for (const [, child] of folder.children) { const api = getAPIFromTestItem(child) if (api) return api } return null}
function getAPIFromTestItem(testItem: vscode.TestItem): VitestFolderAPI | null { const data = getTestData(testItem) // API is stored in test files - if this is a folder, try to find a file inside, // otherwise go up until we find a file, this should never be a folder if (data instanceof TestFolder) return getAPIFromFolder(testItem)
if (data instanceof TestFile) return data.api return data.file.api}