Skip to content

Commit 1c2b210

Browse files
sheremet-vasheremet-va
authored
fix(api): don't report events during vitest list (#7257)
1 parent 80ce0e1 commit 1c2b210

File tree

16 files changed

+87
-33
lines changed

16 files changed

+87
-33
lines changed

‎examples/lit/package.json‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
"coverage": "vitest run --coverage",
1111
"dev": "vite",
1212
"test": "vitest",
13+
"list": "vitest list",
1314
"test:ui": "vitest --ui"
1415
},
1516
"dependencies": {

‎examples/lit/vite.config.ts‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,10 @@ export default defineConfig({
99
// https://lit.dev/docs/tools/testing/#testing-in-the-browser
1010
browser: {
1111
enabled: true,
12-
name: 'chromium',
1312
provider: 'playwright',
13+
instances: [
14+
{ browser: 'chromium' },
15+
],
1416
},
1517
},
1618
})

‎packages/browser/src/client/client.ts‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,10 @@ export const RPC_ID
1212
= PAGE_TYPE === 'orchestrator'
1313
? getBrowserState().sessionId
1414
: getBrowserState().testerId
15+
const METHOD = getBrowserState().method
1516
export const ENTRY_URL = `${
1617
location.protocol === 'https:' ? 'wss:' : 'ws:'
17-
}//${HOST}/__vitest_browser_api__?type=${PAGE_TYPE}&rpcId=${RPC_ID}&sessionId=${getBrowserState().sessionId}&projectName=${getBrowserState().config.name || ''}`
18+
}//${HOST}/__vitest_browser_api__?type=${PAGE_TYPE}&rpcId=${RPC_ID}&sessionId=${getBrowserState().sessionId}&projectName=${getBrowserState().config.name || ''}&method=${METHOD}`
1819

1920
let setCancel = (_: CancelReason) => {}
2021
export const onCancel = new Promise<CancelReason>((resolve) => {

‎packages/browser/src/client/public/esm-client-injector.js‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
sessionId: { __VITEST_SESSION_ID__ },
2828
testerId: { __VITEST_TESTER_ID__ },
2929
provider: { __VITEST_PROVIDER__ },
30+
method: { __VITEST_METHOD__ },
3031
providedContext: { __VITEST_PROVIDED_CONTEXT__ },
3132
};
3233

‎packages/browser/src/client/utils.ts‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,7 @@ export interface BrowserRunnerState {
7474
iframeId?: string
7575
sessionId: string
7676
testerId: string
77+
method: 'run' | 'collect'
7778
runTests?: (tests: string[]) => Promise<void>
7879
createTesters?: (files: string[]) => Promise<void>
7980
cdp?: {

‎packages/browser/src/node/rpc.ts‎

Lines changed: 34 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,13 @@ export function setupBrowserRpc(globalServer: ParentBrowserProject) {
5050
)
5151
}
5252

53+
const method = searchParams.get('method') as 'run' | 'collect'
54+
if (method !== 'run' && method !== 'collect') {
55+
return error(
56+
new Error(`[vitest] Method query in ${request.url} is invalid. Method should be either "run" or "collect".`),
57+
)
58+
}
59+
5360
if (type === 'orchestrator') {
5461
const session = vitest._browserSessions.getSession(sessionId)
5562
// it's possible the session was already resolved by the preview provider
@@ -67,7 +74,7 @@ export function setupBrowserRpc(globalServer: ParentBrowserProject) {
6774
wss.handleUpgrade(request, socket, head, (ws) => {
6875
wss.emit('connection', ws, request)
6976

70-
const rpc = setupClient(project, rpcId, ws)
77+
const rpc = setupClient(project, rpcId, ws, method)
7178
const state = project.browser!.state as BrowserServerState
7279
const clients = type === 'tester' ? state.testers : state.orchestrators
7380
clients.set(rpcId, rpc)
@@ -96,7 +103,7 @@ export function setupBrowserRpc(globalServer: ParentBrowserProject) {
96103
}
97104
}
98105

99-
function setupClient(project: TestProject, rpcId: string, ws: WebSocket) {
106+
function setupClient(project: TestProject, rpcId: string, ws: WebSocket, method: 'run' | 'collect') {
100107
const mockResolver = new ServerMockResolver(globalServer.vite, {
101108
moduleDirectories: project.config.server?.deps?.moduleDirectories,
102109
})
@@ -111,19 +118,39 @@ export function setupBrowserRpc(globalServer: ParentBrowserProject) {
111118
vitest.state.catchError(error, type)
112119
},
113120
async onQueued(file) {
114-
await vitest._testRun.enqueued(project, file)
121+
if (method === 'collect') {
122+
vitest.state.collectFiles(project, [file])
123+
}
124+
else {
125+
await vitest._testRun.enqueued(project, file)
126+
}
115127
},
116128
async onCollected(files) {
117-
await vitest._testRun.collected(project, files)
129+
if (method === 'collect') {
130+
vitest.state.collectFiles(project, files)
131+
}
132+
else {
133+
await vitest._testRun.collected(project, files)
134+
}
118135
},
119136
async onTaskUpdate(packs, events) {
120-
await vitest._testRun.updated(packs, events)
137+
if (method === 'collect') {
138+
vitest.state.updateTasks(packs)
139+
}
140+
else {
141+
await vitest._testRun.updated(packs, events)
142+
}
121143
},
122144
onAfterSuiteRun(meta) {
123145
vitest.coverageProvider?.onAfterSuiteRun(meta)
124146
},
125-
sendLog(log) {
126-
return vitest._testRun.log(log)
147+
async sendLog(log) {
148+
if (method === 'collect') {
149+
vitest.state.updateUserLog(log)
150+
}
151+
else {
152+
await vitest._testRun.log(log)
153+
}
127154
},
128155
resolveSnapshotPath(testPath) {
129156
return vitest.snapshot.resolvePath<ResolveSnapshotPathHandlerContext>(testPath, {

‎packages/browser/src/node/serverOrchestrator.ts‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@ export async function resolveOrchestrator(
3636
__VITEST_VITE_CONFIG__: JSON.stringify({
3737
root: browserProject.vite.config.root,
3838
}),
39+
__VITEST_METHOD__: JSON.stringify(session?.method || 'run'),
3940
__VITEST_FILES__: JSON.stringify(files),
4041
__VITEST_TYPE__: '"orchestrator"',
4142
__VITEST_SESSION_ID__: JSON.stringify(sessionId),

‎packages/browser/src/node/serverTester.ts‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,13 +57,14 @@ export async function resolveTester(
5757
: await globalServer.injectorJs
5858

5959
const injector = replacer(injectorJs, {
60-
__VITEST_PROVIDER__: JSON.stringify(project.browser!.provider!.name),
60+
__VITEST_PROVIDER__: JSON.stringify(project.browser!.provider.name),
6161
__VITEST_CONFIG__: JSON.stringify(browserProject.wrapSerializedConfig()),
6262
__VITEST_FILES__: JSON.stringify(files),
6363
__VITEST_VITE_CONFIG__: JSON.stringify({
6464
root: browserProject.vite.config.root,
6565
}),
6666
__VITEST_TYPE__: '"tester"',
67+
__VITEST_METHOD__: JSON.stringify(method),
6768
__VITEST_SESSION_ID__: JSON.stringify(sessionId),
6869
__VITEST_TESTER_ID__: JSON.stringify(crypto.randomUUID()),
6970
__VITEST_PROVIDED_CONTEXT__: JSON.stringify(stringify(project.getProvidedContext())),

‎packages/vitest/src/node/pools/forks.ts‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ import { wrapSerializableConfig } from '../../utils/config-helpers'
1717
import { envsOrder, groupFilesByEnv } from '../../utils/test-helpers'
1818
import { createMethodsRPC } from './rpc'
1919

20-
function createChildProcessChannel(project: TestProject) {
20+
function createChildProcessChannel(project: TestProject, collect = false) {
2121
const emitter = new EventEmitter()
2222
const cleanup = () => emitter.removeAllListeners()
2323

@@ -27,7 +27,7 @@ function createChildProcessChannel(project: TestProject) {
2727
postMessage: message => emitter.emit(events.response, message),
2828
}
2929

30-
const rpc = createBirpc<RunnerRPC, RuntimeRPC>(createMethodsRPC(project, { cacheFs: true }), {
30+
const rpc = createBirpc<RunnerRPC, RuntimeRPC>(createMethodsRPC(project, { cacheFs: true, collect }), {
3131
eventNames: ['onCancel'],
3232
serialize: v8.serialize,
3333
deserialize: v => v8.deserialize(Buffer.from(v)),
@@ -109,7 +109,7 @@ export function createForksPool(
109109
const paths = files.map(f => f.filepath)
110110
ctx.state.clearFiles(project, paths)
111111

112-
const { channel, cleanup } = createChildProcessChannel(project)
112+
const { channel, cleanup } = createChildProcessChannel(project, name === 'collect')
113113
const workerId = ++id
114114
const data: ContextRPC = {
115115
pool: 'forks',

‎packages/vitest/src/node/pools/rpc.ts‎

Lines changed: 26 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,8 @@ const promises = new Map<string, Promise<void>>()
1111

1212
interface MethodsOptions {
1313
cacheFs?: boolean
14+
// do not report files
15+
collect?: boolean
1416
}
1517

1618
export function createMethodsRPC(project: TestProject, options: MethodsOptions = {}): RuntimeRPC {
@@ -74,24 +76,40 @@ export function createMethodsRPC(project: TestProject, options: MethodsOptions =
7476
transform(id, environment) {
7577
return project.vitenode.transformModule(id, environment).catch(handleRollupError)
7678
},
77-
onPathsCollected(paths) {
78-
ctx.state.collectPaths(paths)
79-
return ctx.report('onPathsCollected', paths)
80-
},
8179
async onQueued(file) {
82-
await ctx._testRun.enqueued(project, file)
80+
if (options.collect) {
81+
ctx.state.collectFiles(project, [file])
82+
}
83+
else {
84+
await ctx._testRun.enqueued(project, file)
85+
}
8386
},
8487
async onCollected(files) {
85-
await ctx._testRun.collected(project, files)
88+
if (options.collect) {
89+
ctx.state.collectFiles(project, files)
90+
}
91+
else {
92+
await ctx._testRun.collected(project, files)
93+
}
8694
},
8795
onAfterSuiteRun(meta) {
8896
ctx.coverageProvider?.onAfterSuiteRun(meta)
8997
},
9098
async onTaskUpdate(packs, events) {
91-
await ctx._testRun.updated(packs, events)
99+
if (options.collect) {
100+
ctx.state.updateTasks(packs)
101+
}
102+
else {
103+
await ctx._testRun.updated(packs, events)
104+
}
92105
},
93106
async onUserConsoleLog(log) {
94-
await ctx._testRun.log(log)
107+
if (options.collect) {
108+
ctx.state.updateUserLog(log)
109+
}
110+
else {
111+
await ctx._testRun.log(log)
112+
}
95113
},
96114
onUnhandledError(err, type) {
97115
ctx.state.catchError(err, type)

0 commit comments

Comments
 (0)