fix: print stdout/stderr logs, not just console.*

Fixes #585

Author: sheremet-va

src/api.ts

@@ -228,7 +228,11 @@ export class VitestFolderAPI {
     this.handlers.clearListeners()
   }
 
-  private createHandler<K extends Exclude<keyof ResolvedMeta['handlers'], 'clearListeners' | 'removeListener'>>(name: K) {
+  onStdout(callback: (log: string) => void) {
+    this.handlers.onStdout(callback)
+  }
+
+  private createHandler<K extends Exclude<keyof ResolvedMeta['handlers'], 'clearListeners' | 'removeListener' | 'onStdout'>>(name: K) {
     return (callback: VitestEvents[K]) => {
       this.handlers[name](callback as any)
     }
@@ -345,6 +349,7 @@ export interface ResolvedMeta {
   pkg: VitestPackage
   configs: string[]
   handlers: {
+    onStdout: (listener: (log: string) => void) => void
     onConsoleLog: (listener: VitestEvents['onConsoleLog']) => void
     onTaskUpdate: (listener: VitestEvents['onTaskUpdate']) => void
     onFinished: (listener: VitestEvents['onFinished']) => void

src/api/child_process.ts

@@ -51,16 +51,27 @@ export async function createVitestProcess(pkg: VitestPackage) {
     cwd: pkg.cwd,
   })
 
-  vitest.stdout?.on('data', d => log.worker('info', d.toString()))
+  const stdoutCallbacks = new Set<(data: string) => void>()
+
+  vitest.stdout?.on('data', (d) => {
+    const content = d.toString()
+    stdoutCallbacks.forEach(cb => cb(content))
+    log.worker('info', content)
+  })
   vitest.stderr?.on('data', (chunk) => {
     const string = chunk.toString()
     log.worker('error', string)
+    stdoutCallbacks.forEach(cb => cb(string))
     if (string.startsWith(' MISSING DEPENDENCY')) {
       const error = string.split(/\r?\n/, 1)[0].slice(' MISSING DEPENDENCY'.length)
       showVitestError(error)
     }
   })
 
+  vitest.on('exit', () => {
+    stdoutCallbacks.clear()
+  })
+
   return new Promise<ResolvedMeta>((resolve, reject) => {
     function onExit(code: number | null) {
       reject(new Error(`Vitest process exited with code ${code}`))
@@ -69,7 +80,17 @@ export async function createVitestProcess(pkg: VitestPackage) {
     vitest.on('exit', onExit)
 
     waitForWsResolvedMeta(wss, pkg, false, 'child_process', vitest)
-      .then(resolve, reject)
+      .then((resolved) => {
+        resolved.handlers.onStdout = (callback: (data: string) => void) => {
+          stdoutCallbacks.add(callback)
+        }
+        const clearListeners = resolved.handlers.clearListeners
+        resolved.handlers.clearListeners = () => {
+          clearListeners()
+          stdoutCallbacks.clear()
+        }
+        resolve(resolved)
+      }, reject)
       .finally(() => {
         vitest.off('exit', onExit)
       })

src/api/ws.ts

@@ -36,7 +36,12 @@ export function waitForWsResolvedMeta(
           })
           resolve({
             rpc: api,
-            handlers,
+            handlers: {
+              ...handlers,
+              onStdout() {
+                // do nothing by default
+              },
+            },
             configs: message.configs,
             process: new VitestWebSocketProcess(Math.random(), wss, ws, child),
             pkg,

src/log.ts

@@ -6,7 +6,21 @@ import { getConfig } from './config'
 
 const logFile = process.env.VITEST_VSCODE_E2E_LOG_FILE!
 const channel = window.createOutputChannel('Vitest')
+const callbacks: Set<((message: string) => void)> = new Set()
+
+function logToCallbacks(message: string) {
+  for (const callback of callbacks) {
+    callback(message)
+  }
+}
+
 export const log = {
+  onWorkerLog(callback: (message: string) => void) {
+    callbacks.add(callback)
+  },
+  offWorkerLog(callback: (message: string) => void) {
+    callbacks.delete(callback)
+  },
   worker: (type: 'info' | 'error', ...args: any[]) => {
     if (typeof args.at(-1) === 'string' && args.at(-1).endsWith('\n'))
       args[args.length - 1] = args.at(-1).slice(0, process.platform === 'win32' ? -2 : -1)
@@ -19,6 +33,7 @@ export const log = {
     if (logFile) {
       appendFile(message)
     }
+    logToCallbacks(message)
     channel.appendLine(message)
   },
   info: (...args: any[]) => {

src/runner/runner.ts

@@ -43,6 +43,12 @@ export class TestRunner extends vscode.Disposable {
       this.disposables = []
     })
 
+    api.onStdout((content) => {
+      if (this.testRun) {
+        this.testRun.appendOutput(formatTestOutput(content))
+      }
+    })
+
     api.onWatcherRerun((files, _trigger, collecting) => {
       if (collecting) {
         log.verbose?.('Not starting the runner because tests are being collected for', ...files.map(f => this.relative(f)))