scriptoriumadmin/Roo-Code
1
0
Fork 0
mirror of https://github.com/RooVetGit/Roo-Code.git synced 2026-09-10 22:41:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: update tests for new ollama model discovery interface

Browse source
This commit is contained in:
randomizedcoder dave.seddon.ca@gmail.com 2026-02-19 19:40:27 -08:00
parent 4e5230d714
commit 3c09ec5bd5
2 changed files with 31 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/api/providers/fetchers/__tests__/ollama.test.ts
View file

@ -23,6 +23,7 @@ describe("Ollama Fetcher", () => {
contextWindow: 40960,
supportsImages: false,
supportsPromptCache: true,
supportsNativeTools: true,
inputPrice: 0,
outputPrice: 0,
cacheWritesPrice: 0,
@ -50,6 +51,7 @@ describe("Ollama Fetcher", () => {
contextWindow: 40960,
supportsImages: false,
supportsPromptCache: true,
supportsNativeTools: true,
inputPrice: 0,
outputPrice: 0,
cacheWritesPrice: 0,

46
src/core/webview/__tests__/webviewMessageHandler.spec.ts
View file

@ -4,6 +4,7 @@ import type { Mock } from "vitest"
// Mock dependencies - must come before imports
vi.mock("../../../api/providers/fetchers/modelCache")
vi.mock("../../../api/providers/fetchers/ollama")
vi.mock("../../../integrations/openai-codex/oauth", () => ({
openAiCodexOAuthManager: {
@ -25,11 +26,14 @@ import type { ModelRecord } from "@roo-code/types"
import { webviewMessageHandler } from "../webviewMessageHandler"
import type { ClineProvider } from "../ClineProvider"
import { getModels } from "../../../api/providers/fetchers/modelCache"
import { getModels, flushModels } from "../../../api/providers/fetchers/modelCache"
import { discoverOllamaModelsWithSorting } from "../../../api/providers/fetchers/ollama"
const { openAiCodexOAuthManager } = await import("../../../integrations/openai-codex/oauth")
const { fetchOpenAiCodexRateLimitInfo } = await import("../../../integrations/openai-codex/rate-limits")
const mockGetModels = getModels as Mock<typeof getModels>
const mockFlushModels = flushModels as Mock<typeof flushModels>
const mockDiscoverOllamaModels = discoverOllamaModelsWithSorting as Mock<typeof discoverOllamaModelsWithSorting>
const mockGetAccessToken = vi.mocked(openAiCodexOAuthManager.getAccessToken)
const mockGetAccountId = vi.mocked(openAiCodexOAuthManager.getAccountId)
const mockFetchOpenAiCodexRateLimitInfo = vi.mocked(fetchOpenAiCodexRateLimitInfo)
@ -228,32 +232,40 @@ describe("webviewMessageHandler - requestOllamaModels", () => {
})
it("successfully fetches models from Ollama", async () => {
const mockModels: ModelRecord = {
"model-1": {
maxTokens: 4096,
contextWindow: 8192,
supportsPromptCache: false,
description: "Test model 1",
},
"model-2": {
maxTokens: 8192,
contextWindow: 16384,
supportsPromptCache: false,
description: "Test model 2",
},
const mockModelInfo = {
maxTokens: 4096,
contextWindow: 8192,
supportsPromptCache: false,
description: "Test model 1",
}
mockGetModels.mockResolvedValue(mockModels)
mockFlushModels.mockResolvedValue(undefined)
mockDiscoverOllamaModels.mockResolvedValue({
modelsWithTools: [
{ name: "model-1", modelInfo: mockModelInfo, contextWindow: 8192, supportsImages: false },
],
modelsWithoutTools: [],
totalCount: 1,
})
await webviewMessageHandler(mockClineProvider, {
type: "requestOllamaModels",
})
expect(mockGetModels).toHaveBeenCalledWith({ provider: "ollama", baseUrl: "http://localhost:1234" })
expect(mockFlushModels).toHaveBeenCalledWith({ provider: "ollama", baseUrl: "http://localhost:1234" }, true)
expect(mockDiscoverOllamaModels).toHaveBeenCalledWith(
"http://localhost:1234",
undefined,
expect.objectContaining({}),
)
expect(mockClineProvider.postMessageToWebview).toHaveBeenCalledWith({
type: "ollamaModels",
ollamaModels: mockModels,
ollamaModels: { "model-1": mockModelInfo },
ollamaModelsWithTools: [
{ name: "model-1", modelInfo: mockModelInfo, contextWindow: 8192, supportsImages: false },
],
modelsWithoutTools: [],
})
})
})