chore(deps): bump to vitest 4 and adjust tests after the upgrade (#1198)

* chore(deps): update vitest monorepo to v4

* test: adjust test for vitest 4

* chore: format

* tests: use find

---------

Co-authored-by: renovate[bot] <29139614+renovate[bot]@users.noreply.github.com>

Author: samuv

__mocks__/fs.cjs

@@ -0,0 +1,5 @@
+// Mock for node:fs using memfs
+// See: https://vitest.dev/guide/mocking/file-system.html
+
+const { fs } = require('memfs')
+module.exports = fs

__mocks__/fs/promises.cjs

@@ -0,0 +1,5 @@
+// Mock for node:fs/promises using memfs
+// See: https://vitest.dev/guide/mocking/file-system.html
+
+const { fs } = require('memfs')
+module.exports = fs.promises

main/src/tests/auto-update.test.ts

@@ -104,7 +104,9 @@ vi.mock('electron-store', () => {
   }
 
   return {
-    default: vi.fn().mockImplementation(() => mockStoreInstance),
+    default: vi.fn(function ElectronStore() {
+      return mockStoreInstance
+    }),
   }
 })
 

main/src/tests/graceful-exit.test.ts

@@ -48,7 +48,9 @@ vi.mock('electron-store', () => {
   ).__mockStoreInstance = mockStoreInstance
 
   return {
-    default: vi.fn().mockImplementation(() => mockStoreInstance),
+    default: vi.fn(function ElectronStore() {
+      return mockStoreInstance
+    }),
   }
 })
 

main/src/tests/toolhive-manager.test.ts

@@ -1,26 +1,48 @@
 import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest'
+import { EventEmitter } from 'node:events'
+import { vol } from 'memfs'
 import { spawn } from 'node:child_process'
-import { existsSync } from 'node:fs'
 import net from 'node:net'
 import { app } from 'electron'
-import { EventEmitter } from 'node:events'
 import {
   startToolhive,
   getToolhivePort,
   getToolhiveMcpPort,
   isToolhiveRunning,
   stopToolhive,
   restartToolhive,
+  binPath,
 } from '../toolhive-manager'
 import { updateTrayStatus } from '../system-tray'
 import log from '../logger'
 import * as Sentry from '@sentry/electron/main'
 import { getQuittingState } from '../app-state'
 
-// Mock dependencies
-vi.mock('node:child_process')
+vi.mock('node:child_process', async (importOriginal) => {
+  const actual = await importOriginal<typeof import('node:child_process')>()
+  const mockSpawnFn = vi.fn()
+  return {
+    ...actual,
+    spawn: mockSpawnFn,
+    default: {
+      ...actual,
+      spawn: mockSpawnFn,
+    },
+  }
+})
 vi.mock('node:fs')
-vi.mock('node:net')
+vi.mock('node:net', async (importOriginal) => {
+  const actual = await importOriginal<typeof import('node:net')>()
+  const mockCreateServerFn = vi.fn()
+  return {
+    ...actual,
+    createServer: mockCreateServerFn,
+    default: {
+      ...actual,
+      createServer: mockCreateServerFn,
+    },
+  }
+})
 vi.mock('electron', () => ({
   app: {
     isPackaged: false,
@@ -53,12 +75,14 @@ vi.mock('electron-store', () => {
   }
 
   return {
-    default: vi.fn().mockImplementation(() => mockStoreInstance),
+    default: vi.fn(function ElectronStore() {
+      return mockStoreInstance
+    }),
   }
 })
 
+// Get mocked functions
 const mockSpawn = vi.mocked(spawn)
-const mockExistsSync = vi.mocked(existsSync)
 const mockNet = vi.mocked(net)
 const mockApp = vi.mocked(app)
 const mockUpdateTrayStatus = vi.mocked(updateTrayStatus)
@@ -112,26 +136,35 @@ describe('toolhive-manager', () => {
     vi.clearAllMocks()
     vi.useFakeTimers()
 
+    // Reset the in-memory file system
+    vol.reset()
+
+    // Setup the file system state - make binary exist by default
+    vol.fromJSON({
+      [binPath]: '', // Mock the binary file exists
+    })
+
     // Reset module state
     stopToolhive()
 
     // Setup mocks
     mockProcess = new MockProcess()
 
-    mockSpawn.mockReturnValue(
-      mockProcess as unknown as ReturnType<typeof spawn>
-    )
-    mockExistsSync.mockReturnValue(true)
+    mockSpawn.mockImplementation(function spawn() {
+      return mockProcess as unknown as ReturnType<
+        typeof import('node:child_process').spawn
+      >
+    })
     Object.defineProperty(mockApp, 'isPackaged', {
       value: false,
       configurable: true,
     })
     vi.mocked(mockApp.getPath).mockReturnValue('/test/userData')
 
     // Mock net.createServer to return a new MockServer instance each time
-    mockNet.createServer.mockImplementation(
-      () => new MockServer() as unknown as net.Server
-    )
+    mockNet.createServer.mockImplementation(function createServer() {
+      return new MockServer() as unknown as net.Server
+    })
 
     // Mock logger methods
     mockLog.info = vi.fn()
@@ -147,7 +180,8 @@ describe('toolhive-manager', () => {
 
   describe('startToolhive', () => {
     it('returns early if binary does not exist', async () => {
-      mockExistsSync.mockReturnValue(false)
+      // Remove the binary file from the in-memory file system
+      vol.reset()
 
       await startToolhive()
 
@@ -382,28 +416,30 @@ describe('toolhive-manager', () => {
   describe('port finding with fallback', () => {
     it('falls back to random port when preferred range is unavailable', async () => {
       // Mock all ports in range to be unavailable, then allow random port
-      mockNet.createServer.mockImplementation(() => {
+      mockNet.createServer.mockImplementation(function createServer() {
         const server = new MockServer() as unknown as net.Server
         const originalListen = server.listen.bind(server)
 
-        server.listen = vi
-          .fn()
-          .mockImplementation((port: number, callback?: () => void) => {
-            if (port >= 50000 && port <= 50100) {
-              // Simulate all ports in range being unavailable
-              setTimeout(() => {
-                server.emit('error', { code: 'EADDRINUSE' })
-              }, 5)
-            } else if (port === 0) {
-              // Allow OS assignment (fallback)
-              setTimeout(() => {
-                originalListen(port, callback)
-              }, 5)
-            } else {
-              // Any other specific port
+        server.listen = vi.fn(function listen(
+          port: number,
+          callback?: () => void
+        ) {
+          if (port >= 50000 && port <= 50100) {
+            // Simulate all ports in range being unavailable
+            setTimeout(() => {
+              server.emit('error', { code: 'EADDRINUSE' })
+            }, 5)
+          } else if (port === 0) {
+            // Allow OS assignment (fallback)
+            setTimeout(() => {
               originalListen(port, callback)
-            }
-          })
+            }, 5)
+          } else {
+            // Any other specific port
+            originalListen(port, callback)
+          }
+          return server
+        }) as unknown as typeof server.listen
 
         return server
       })
@@ -566,10 +602,12 @@ describe('toolhive-manager', () => {
 
     it('handles kill errors and attempts force kill as fallback', () => {
       const killSpy = vi.spyOn(mockProcess, 'kill')
-      killSpy.mockImplementationOnce(() => {
+      killSpy.mockImplementationOnce(function killImpl() {
         throw new Error('Kill failed')
       })
-      killSpy.mockImplementationOnce(() => true)
+      killSpy.mockImplementationOnce(function killImpl() {
+        return true
+      })
 
       stopToolhive()
 

package.json

@@ -61,7 +61,7 @@
     "@types/regexp.escape": "^2.0.0",
     "@types/unzipper": "^0.10.11",
     "@vitejs/plugin-react-swc": "^4.0.0",
-    "@vitest/coverage-istanbul": "^3.1.4",
+    "@vitest/coverage-istanbul": "^4.0.0",
     "ajv": "^8.17.1",
     "autoprefixer": "^10.4.21",
     "dotenv": "^17.0.0",
@@ -76,6 +76,7 @@
     "json-schema-faker": "^0.5.9",
     "knip": "^5.58.1",
     "lint-staged": "^16.0.0",
+    "memfs": "^4.50.0",
     "msw": "^2.8.4",
     "prettier": "3.6.2",
     "prettier-plugin-classnames": "^0.8.0",
@@ -87,7 +88,7 @@
     "typescript-eslint": "^8.30.1",
     "unzipper": "^0.12.3",
     "vite": "^7.0.0",
-    "vitest": "^3.1.4",
+    "vitest": "^4.0.0",
     "vitest-fail-on-console": "^0.10.0"
   },
   "dependencies": {

parallel-requests.sh

@@ -0,0 +1,70 @@
+#!/bin/bash
+
+# Colors for output
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+BLUE='\033[0;34m'
+RED='\033[0;31m'
+NC='\033[0m' # No Color
+
+echo "================================"
+echo "   Starting API Requests"
+echo "================================"
+echo ""
+
+# Function to run curl and track its status
+run_request() {
+    local name=$1
+    local url=$2
+    local method=$3
+    local data=$4
+    
+    echo -e "${YELLOW}⏳ [$method]${NC} $name - ${BLUE}PENDING${NC}"
+    
+    if [ "$method" == "POST" ]; then
+        response=$(curl -s -w "\n%{http_code}" "$url" --data-raw "$data" 2>&1)
+    else
+        response=$(curl -s -w "\n%{http_code}" "$url" 2>&1)
+    fi
+    
+    http_code=$(echo "$response" | tail -n1)
+    body=$(echo "$response" | head -n-1)
+    
+    if [ "$http_code" == "200" ] || [ "$http_code" == "201" ]; then
+        echo -e "${GREEN}✓ [$method]${NC} $name - ${GREEN}COMPLETED${NC} (HTTP $http_code)"
+    else
+        echo -e "${RED}✗ [$method]${NC} $name - ${RED}FAILED${NC} (HTTP $http_code)"
+    fi
+    
+    # Optionally print response body (truncated)
+    # echo "$body" | head -c 200
+}
+
+# Launch POST request in background
+run_request "Create Workload" \
+    "http://localhost:50018/api/v1beta/workloads" \
+    "POST" \
+    '{"name":"everything-2","image":"docker.io/mcp/everything:latest","transport":"stdio","env_vars":{},"secrets":[],"cmd_arguments":[],"network_isolation":false,"volumes":[],"group":"default"}' &
+pid1=$!
+
+# Launch GET requests in background
+run_request "Health Check" \
+    "http://localhost:50018/health" \
+    "GET" \
+    "" &
+pid2=$!
+
+run_request "List Workloads" \
+    "http://localhost:50018/api/v1beta/workloads?all=true&group=default" \
+    "GET" \
+    "" &
+pid3=$!
+
+# Wait for all background jobs to complete
+wait $pid1 $pid2 $pid3
+
+echo ""
+echo "================================"
+echo "   All Requests Completed"
+echo "================================"
+