Skip to content
This repository was archived by the owner on May 15, 2026. It is now read-only.

Commit e8f83a0

Browse files
committed
fix(test): update tests for firmware provider integration
- Add firmware provider to router model expectations in ClineProvider.spec.ts (3 tests) - Add firmware provider to router model expectations in webviewMessageHandler.spec.ts (4 tests) - Update mock call sequences to include firmware provider (placed after chutes, before litellm) All tests now pass (5161 passed). The firmware provider was recently added to the dynamic providers list and these tests needed to be updated to include it in their expected routerModels objects.
1 parent 6e9911f commit e8f83a0

2 files changed

Lines changed: 16 additions & 0 deletions

File tree

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

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2643,6 +2643,7 @@ describe("ClineProvider - Router Models", () => {
26432643
unbound: mockModels,
26442644
roo: mockModels,
26452645
chutes: mockModels,
2646+
firmware: mockModels,
26462647
litellm: mockModels,
26472648
ollama: {},
26482649
lmstudio: {},
@@ -2682,6 +2683,7 @@ describe("ClineProvider - Router Models", () => {
26822683
.mockResolvedValueOnce(mockModels) // deepinfra success
26832684
.mockResolvedValueOnce(mockModels) // roo success
26842685
.mockRejectedValueOnce(new Error("Chutes API error")) // chutes fail
2686+
.mockResolvedValueOnce(mockModels) // firmware success
26852687
.mockRejectedValueOnce(new Error("LiteLLM connection failed")) // litellm fail
26862688

26872689
await messageHandler({ type: "requestRouterModels" })
@@ -2696,6 +2698,7 @@ describe("ClineProvider - Router Models", () => {
26962698
unbound: {},
26972699
roo: mockModels,
26982700
chutes: {},
2701+
firmware: mockModels,
26992702
ollama: {},
27002703
lmstudio: {},
27012704
litellm: {},
@@ -2817,6 +2820,7 @@ describe("ClineProvider - Router Models", () => {
28172820
unbound: mockModels,
28182821
roo: mockModels,
28192822
chutes: mockModels,
2823+
firmware: mockModels,
28202824
litellm: {},
28212825
ollama: {},
28222826
lmstudio: {},

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

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -325,6 +325,7 @@ describe("webviewMessageHandler - requestRouterModels", () => {
325325
litellm: mockModels,
326326
roo: mockModels,
327327
chutes: mockModels,
328+
firmware: mockModels,
328329
ollama: {},
329330
lmstudio: {},
330331
"vercel-ai-gateway": mockModels,
@@ -415,6 +416,7 @@ describe("webviewMessageHandler - requestRouterModels", () => {
415416
unbound: mockModels,
416417
roo: mockModels,
417418
chutes: mockModels,
419+
firmware: mockModels,
418420
litellm: {},
419421
ollama: {},
420422
lmstudio: {},
@@ -445,6 +447,7 @@ describe("webviewMessageHandler - requestRouterModels", () => {
445447
.mockResolvedValueOnce(mockModels) // deepinfra
446448
.mockResolvedValueOnce(mockModels) // roo
447449
.mockRejectedValueOnce(new Error("Chutes API error")) // chutes
450+
.mockResolvedValueOnce(mockModels) // firmware
448451
.mockRejectedValueOnce(new Error("LiteLLM connection failed")) // litellm
449452

450453
await webviewMessageHandler(mockClineProvider, {
@@ -490,6 +493,7 @@ describe("webviewMessageHandler - requestRouterModels", () => {
490493
unbound: {},
491494
roo: mockModels,
492495
chutes: {},
496+
firmware: mockModels,
493497
litellm: {},
494498
ollama: {},
495499
lmstudio: {},
@@ -511,6 +515,7 @@ describe("webviewMessageHandler - requestRouterModels", () => {
511515
.mockRejectedValueOnce(new Error("DeepInfra API error")) // deepinfra
512516
.mockRejectedValueOnce(new Error("Roo API error")) // roo
513517
.mockRejectedValueOnce(new Error("Chutes API error")) // chutes
518+
.mockRejectedValueOnce(new Error("Firmware API error")) // firmware
514519
.mockRejectedValueOnce(new Error("LiteLLM connection failed")) // litellm
515520

516521
await webviewMessageHandler(mockClineProvider, {
@@ -567,6 +572,13 @@ describe("webviewMessageHandler - requestRouterModels", () => {
567572
values: { provider: "chutes" },
568573
})
569574

575+
expect(mockClineProvider.postMessageToWebview).toHaveBeenCalledWith({
576+
type: "singleRouterModelFetchResponse",
577+
success: false,
578+
error: "Firmware API error",
579+
values: { provider: "firmware" },
580+
})
581+
570582
expect(mockClineProvider.postMessageToWebview).toHaveBeenCalledWith({
571583
type: "singleRouterModelFetchResponse",
572584
success: false,

0 commit comments

Comments
 (0)