fix tests

Author: adilhafeez

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

@@ -2416,10 +2416,10 @@ describe("ClineProvider - Router Models", () => {
 				litellm: mockModels,
 				ollama: {},
 				lmstudio: {},
+				archgw: mockModels,
 			},
 		})
 	})
-
 	test("handles requestRouterModels with individual provider failures", async () => {
 		await provider.resolveWebviewView(mockWebviewView)
 		const messageHandler = (mockWebviewView.webview.onDidReceiveMessage as any).mock.calls[0][0]
@@ -2432,6 +2432,8 @@ describe("ClineProvider - Router Models", () => {
 				unboundApiKey: "unbound-key",
 				litellmApiKey: "litellm-key",
 				litellmBaseUrl: "http://localhost:4000",
+				// archgwApiKey: "archgw-key",
+				// archgwBaseUrl: "http://localhost:12000/v1",
 			},
 		} as any)
 
@@ -2447,6 +2449,7 @@ describe("ClineProvider - Router Models", () => {
 			.mockResolvedValueOnce(mockModels) // glama success
 			.mockRejectedValueOnce(new Error("Unbound API error")) // unbound fail
 			.mockRejectedValueOnce(new Error("LiteLLM connection failed")) // litellm fail
+			.mockResolvedValueOnce(mockModels) // archgw success
 
 		await messageHandler({ type: "requestRouterModels" })
 
@@ -2461,6 +2464,7 @@ describe("ClineProvider - Router Models", () => {
 				ollama: {},
 				lmstudio: {},
 				litellm: {},
+				archgw: mockModels,
 			},
 		})
 
@@ -2571,6 +2575,7 @@ describe("ClineProvider - Router Models", () => {
 				litellm: {},
 				ollama: {},
 				lmstudio: {},
+				archgw: mockModels,
 			},
 		})
 	})

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

@@ -153,6 +153,7 @@ describe("webviewMessageHandler - requestRouterModels", () => {
 				litellm: mockModels,
 				ollama: {},
 				lmstudio: {},
+				archgw: mockModels,
 			},
 		})
 	})
@@ -240,6 +241,7 @@ describe("webviewMessageHandler - requestRouterModels", () => {
 				litellm: {},
 				ollama: {},
 				lmstudio: {},
+				archgw: mockModels,
 			},
 		})
 	})
@@ -277,6 +279,7 @@ describe("webviewMessageHandler - requestRouterModels", () => {
 				litellm: {},
 				ollama: {},
 				lmstudio: {},
+				archgw: mockModels,
 			},
 		})