fix: update test files to use Vitest instead of Jest for LMStudio and Ollama fetchers

Author: daniel-lxs

src/api/providers/fetchers/__tests__/lmstudio.test.ts

@@ -1,30 +1,29 @@
 import axios from "axios"
+import { vi, describe, it, expect, beforeEach } from "vitest"
 import { LMStudioClient, LLMInfo } from "@lmstudio/sdk" // LLMInfo is a type
 import { getLMStudioModels, parseLMStudioModel } from "../lmstudio"
 import { ModelInfo, lMStudioDefaultModelInfo } from "@roo-code/types" // ModelInfo is a type
 
 // Mock axios
-jest.mock("axios")
-const mockedAxios = axios as jest.Mocked<typeof axios>
+vi.mock("axios")
+const mockedAxios = axios as any
 
 // Mock @lmstudio/sdk
-const mockListDownloadedModels = jest.fn()
-jest.mock("@lmstudio/sdk", () => {
-	const originalModule = jest.requireActual("@lmstudio/sdk")
+const mockListDownloadedModels = vi.fn()
+vi.mock("@lmstudio/sdk", () => {
 	return {
-		...originalModule,
-		LMStudioClient: jest.fn().mockImplementation(() => ({
+		LMStudioClient: vi.fn().mockImplementation(() => ({
 			system: {
 				listDownloadedModels: mockListDownloadedModels,
 			},
 		})),
 	}
 })
-const MockedLMStudioClientConstructor = LMStudioClient as jest.MockedClass<typeof LMStudioClient>
+const MockedLMStudioClientConstructor = LMStudioClient as any
 
 describe("LMStudio Fetcher", () => {
 	beforeEach(() => {
-		jest.clearAllMocks()
+		vi.clearAllMocks()
 		MockedLMStudioClientConstructor.mockClear()
 	})
 
@@ -133,7 +132,7 @@ describe("LMStudio Fetcher", () => {
 		})
 
 		it("should return an empty object and log error if axios.get fails with a generic error", async () => {
-			const consoleErrorSpy = jest.spyOn(console, "error").mockImplementation(() => {})
+			const consoleErrorSpy = vi.spyOn(console, "error").mockImplementation(() => {})
 			const networkError = new Error("Network connection failed")
 			mockedAxios.get.mockRejectedValueOnce(networkError)
 
@@ -151,7 +150,7 @@ describe("LMStudio Fetcher", () => {
 		})
 
 		it("should return an empty object and log info if axios.get fails with ECONNREFUSED", async () => {
-			const consoleInfoSpy = jest.spyOn(console, "warn").mockImplementation(() => {})
+			const consoleInfoSpy = vi.spyOn(console, "warn").mockImplementation(() => {})
 			const econnrefusedError = new Error("Connection refused")
 			;(econnrefusedError as any).code = "ECONNREFUSED"
 			mockedAxios.get.mockRejectedValueOnce(econnrefusedError)
@@ -168,7 +167,7 @@ describe("LMStudio Fetcher", () => {
 		})
 
 		it("should return an empty object and log error if listDownloadedModels fails", async () => {
-			const consoleErrorSpy = jest.spyOn(console, "error").mockImplementation(() => {})
+			const consoleErrorSpy = vi.spyOn(console, "error").mockImplementation(() => {})
 			const listError = new Error("LMStudio SDK internal error")
 
 			mockedAxios.get.mockResolvedValueOnce({ data: {} })

src/api/providers/fetchers/__tests__/ollama.test.ts

@@ -1,32 +1,35 @@
 import axios from "axios"
 import path from "path"
+import { vi, describe, it, expect, beforeEach } from "vitest"
 import { getOllamaModels, parseOllamaModel } from "../ollama"
-import * as ollamaModelsData from "./fixtures/ollama-model-details.json"
+import ollamaModelsData from "./fixtures/ollama-model-details.json"
 
 // Mock axios
-jest.mock("axios")
-const mockedAxios = axios as jest.Mocked<typeof axios>
+vi.mock("axios")
+const mockedAxios = axios as any
 
 describe("Ollama Fetcher", () => {
 	beforeEach(() => {
-		jest.clearAllMocks()
+		vi.clearAllMocks()
 	})
 
 	describe("parseOllamaModel", () => {
-		const ollamaModels = ollamaModelsData as Record<string, any>
-		const parsedModel = parseOllamaModel(ollamaModels["qwen3-2to16:latest"])
-
-		expect(parsedModel).toEqual({
-			maxTokens: 40960,
-			contextWindow: 40960,
-			supportsImages: false,
-			supportsComputerUse: false,
-			supportsPromptCache: true,
-			inputPrice: 0,
-			outputPrice: 0,
-			cacheWritesPrice: 0,
-			cacheReadsPrice: 0,
-			description: "Family: qwen3, Context: 40960, Size: 32.8B",
+		it("should correctly parse Ollama model info", () => {
+			const modelData = ollamaModelsData["qwen3-2to16:latest"]
+			const parsedModel = parseOllamaModel(modelData)
+
+			expect(parsedModel).toEqual({
+				maxTokens: 40960,
+				contextWindow: 40960,
+				supportsImages: false,
+				supportsComputerUse: false,
+				supportsPromptCache: true,
+				inputPrice: 0,
+				outputPrice: 0,
+				cacheWritesPrice: 0,
+				cacheReadsPrice: 0,
+				description: "Family: qwen3, Context: 40960, Size: 32.8B",
+			})
 		})
 	})
 
@@ -98,7 +101,7 @@ describe("Ollama Fetcher", () => {
 		it("should return an empty list if the initial /api/tags call fails", async () => {
 			const baseUrl = "http://localhost:11434"
 			mockedAxios.get.mockRejectedValueOnce(new Error("Network error"))
-			const consoleInfoSpy = jest.spyOn(console, "error").mockImplementation(() => {}) // Spy and suppress output
+			const consoleInfoSpy = vi.spyOn(console, "error").mockImplementation(() => {}) // Spy and suppress output
 
 			const result = await getOllamaModels(baseUrl)
 
@@ -110,7 +113,7 @@ describe("Ollama Fetcher", () => {
 
 		it("should log an info message and return an empty object on ECONNREFUSED", async () => {
 			const baseUrl = "http://localhost:11434"
-			const consoleInfoSpy = jest.spyOn(console, "warn").mockImplementation(() => {}) // Spy and suppress output
+			const consoleInfoSpy = vi.spyOn(console, "warn").mockImplementation(() => {}) // Spy and suppress output
 
 			const econnrefusedError = new Error("Connection refused") as any
 			econnrefusedError.code = "ECONNREFUSED"

webview-ui/src/utils/__tests__/validate.test.ts

@@ -36,6 +36,8 @@ describe("Model Validation Functions", () => {
 		requesty: {},
 		unbound: {},
 		litellm: {},
+		ollama: {},
+		lmstudio: {},
 	}
 
 	const allowAllOrganization: OrganizationAllowList = {