Skip to content

Commit 61546b5

Browse files
committed
test(api): add cognima provider error expectations to requestRouterModels test
1 parent ab0d7c2 commit 61546b5

File tree

1 file changed

+20
-4
lines changed

1 file changed

+20
-4
lines changed

src/core/webview/__tests__/webviewMessageHandler.spec.ts

Lines changed: 20 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -372,6 +372,7 @@ describe("webviewMessageHandler - requestRouterModels", () => {
372372
.mockResolvedValueOnce(mockModels) // vercel-ai-gateway
373373
.mockResolvedValueOnce(mockModels) // deepinfra
374374
.mockResolvedValueOnce(mockModels) // roo
375+
.mockRejectedValueOnce(new Error("Cognima API error")) // cognima
375376
.mockRejectedValueOnce(new Error("LiteLLM connection failed")) // litellm
376377

377378
await webviewMessageHandler(mockClineProvider, {
@@ -413,6 +414,13 @@ describe("webviewMessageHandler - requestRouterModels", () => {
413414
values: { provider: "unbound" },
414415
})
415416

417+
expect(mockClineProvider.postMessageToWebview).toHaveBeenCalledWith({
418+
type: "singleRouterModelFetchResponse",
419+
success: false,
420+
error: "Cognima API error",
421+
values: { provider: "cognima" },
422+
})
423+
416424
expect(mockClineProvider.postMessageToWebview).toHaveBeenCalledWith({
417425
type: "singleRouterModelFetchResponse",
418426
success: false,
@@ -431,6 +439,7 @@ describe("webviewMessageHandler - requestRouterModels", () => {
431439
.mockRejectedValueOnce(new Error("Vercel AI Gateway error")) // vercel-ai-gateway
432440
.mockRejectedValueOnce(new Error("DeepInfra API error")) // deepinfra
433441
.mockRejectedValueOnce(new Error("Roo API error")) // roo
442+
.mockRejectedValueOnce(new Error("Cognima API error")) // cognima
434443
.mockRejectedValueOnce(new Error("LiteLLM connection failed")) // litellm
435444

436445
await webviewMessageHandler(mockClineProvider, {
@@ -469,15 +478,15 @@ describe("webviewMessageHandler - requestRouterModels", () => {
469478
expect(mockClineProvider.postMessageToWebview).toHaveBeenCalledWith({
470479
type: "singleRouterModelFetchResponse",
471480
success: false,
472-
error: "DeepInfra API error",
473-
values: { provider: "deepinfra" },
481+
error: "Vercel AI Gateway error",
482+
values: { provider: "vercel-ai-gateway" },
474483
})
475484

476485
expect(mockClineProvider.postMessageToWebview).toHaveBeenCalledWith({
477486
type: "singleRouterModelFetchResponse",
478487
success: false,
479-
error: "Vercel AI Gateway error",
480-
values: { provider: "vercel-ai-gateway" },
488+
error: "DeepInfra API error",
489+
values: { provider: "deepinfra" },
481490
})
482491

483492
expect(mockClineProvider.postMessageToWebview).toHaveBeenCalledWith({
@@ -487,6 +496,13 @@ describe("webviewMessageHandler - requestRouterModels", () => {
487496
values: { provider: "roo" },
488497
})
489498

499+
expect(mockClineProvider.postMessageToWebview).toHaveBeenCalledWith({
500+
type: "singleRouterModelFetchResponse",
501+
success: false,
502+
error: "Cognima API error",
503+
values: { provider: "cognima" },
504+
})
505+
490506
expect(mockClineProvider.postMessageToWebview).toHaveBeenCalledWith({
491507
type: "singleRouterModelFetchResponse",
492508
success: false,

0 commit comments

Comments
 (0)