test(webview): cover Kimi Code model selection

Author: WebMad

webview-ui/src/components/ui/hooks/__tests__/useSelectedModel.spec.ts

@@ -23,6 +23,7 @@ import {
 	vscodeLlmDefaultModelId,
 	moonshotDefaultModelId,
 	moonshotModels,
+	kimiCodeDefaultModelInfo,
 	providerIdentifiers,
 } from "@roo-code/types"
 
@@ -1079,6 +1080,50 @@ describe("useSelectedModel", () => {
 		})
 	})
 
+	describe("Kimi Code provider", () => {
+		it("should resolve the configured model from router models", () => {
+			const modelInfo: ModelInfo = {
+				...kimiCodeDefaultModelInfo,
+				description: "Configured Kimi Code model",
+			}
+
+			mockUseRouterModels.mockReturnValue({
+				data: { "kimi-code": { "kimi-for-coding": modelInfo } },
+				isLoading: false,
+				isError: false,
+			} as any)
+
+			const apiConfiguration: ProviderSettings = {
+				apiProvider: providerIdentifiers.kimiCode,
+				apiModelId: "kimi-for-coding",
+			}
+
+			const wrapper = createWrapper()
+			const { result } = renderHook(() => useSelectedModel(apiConfiguration), { wrapper })
+
+			expect(result.current.provider).toBe(providerIdentifiers.kimiCode)
+			expect(result.current.id).toBe("kimi-for-coding")
+			expect(result.current.info).toEqual(modelInfo)
+		})
+	})
+
+	it("should reject providers unsupported by model selection", () => {
+		const consoleError = vi.spyOn(console, "error").mockImplementation(() => undefined)
+		const preventExpectedError = (event: ErrorEvent) => event.preventDefault()
+		window.addEventListener("error", preventExpectedError)
+		const apiConfiguration = {
+			apiProvider: "unsupported-provider",
+		} as unknown as ProviderSettings
+
+		const wrapper = createWrapper()
+
+		expect(() => renderHook(() => useSelectedModel(apiConfiguration), { wrapper })).toThrow(
+			"Unsupported provider: unsupported-provider",
+		)
+		window.removeEventListener("error", preventExpectedError)
+		consoleError.mockRestore()
+	})
+
 	describe("moonshot provider", () => {
 		beforeEach(() => {
 			mockUseRouterModels.mockReturnValue({