fix(browser): stop the browser rpc when the pool is closed (#6858)

sheremet-va · web-flow · commit 9a0c93d7876d · 2024-11-13T11:54:53.000+01:00
diff --git a/packages/browser/src/node/pool.ts b/packages/browser/src/node/pool.ts
@@ -86,8 +86,13 @@ export function createBrowserPool(ctx: Vitest): ProcessPool {
           [...files.map(f => relative(project.config.root, f))].join(', '),
         )
         const promise = waitForTests(method, contextId, project, files)
-        promises.push(promise)
-        orchestrator.createTesters(files)
+        const tester = orchestrator.createTesters(files).catch((error) => {
+          if (error instanceof Error && error.message.startsWith('[birpc] rpc is closed')) {
+            return
+          }
+          return Promise.reject(error)
+        })
+        promises.push(promise, tester)
       }
       else {
         const contextId = crypto.randomUUID()
@@ -156,6 +161,11 @@ export function createBrowserPool(ctx: Vitest): ProcessPool {
     async close() {
       await Promise.all([...providers].map(provider => provider.close()))
       providers.clear()
+      ctx.resolvedProjects.forEach((project) => {
+        project.browser?.state.orchestrators.forEach((orchestrator) => {
+          orchestrator.$close()
+        })
+      })
     },
     runTests: files => runWorkspaceTests('run', files),
     collectTests: files => runWorkspaceTests('collect', files),
diff --git a/packages/vitest/src/node/core.ts b/packages/vitest/src/node/core.ts
@@ -77,7 +77,8 @@ export class Vitest {
 
   private coreWorkspaceProject!: WorkspaceProject
 
-  private resolvedProjects: WorkspaceProject[] = []
+  /** @private */
+  public resolvedProjects: WorkspaceProject[] = []
   public projects: WorkspaceProject[] = []
 
   public distPath = distDir
diff --git a/packages/vitest/src/node/types/browser.ts b/packages/vitest/src/node/types/browser.ts
@@ -194,6 +194,7 @@ export interface BrowserServerStateContext {
 export interface BrowserOrchestrator {
   createTesters: (files: string[]) => Promise<void>
   onCancel: (reason: CancelReason) => Promise<void>
+  $close: () => void
 }
 
 export interface BrowserServerState {

Original file line number	Diff line number	Diff line change
`@@ -194,6 +194,7 @@ export interface BrowserServerStateContext {`
`194`	`194`	`export interface BrowserOrchestrator {`
`195`	`195`	`createTesters: (files: string[]) => Promise<void>`
`196`	`196`	`onCancel: (reason: CancelReason) => Promise<void>`
	`197`	`+ $close: () => void`
`197`	`198`	`}`
`198`	`199`
`199`	`200`	`export interface BrowserServerState {`