test: bump test coverage

Author: rrewll

src/api/providers/__tests__/vscode-lm.spec.ts

@@ -432,6 +432,47 @@ describe("VsCodeLmHandler", () => {
 				expect.anything(),
 			)
 		})
+
+		it("should throw a Zoo Code branded error when request is cancelled", async () => {
+			const systemPrompt = "You are a helpful assistant"
+			const messages: Anthropic.Messages.MessageParam[] = [
+				{
+					role: "user" as const,
+					content: "Hello",
+				},
+			]
+
+			mockLanguageModelChat.sendRequest.mockRejectedValueOnce(new vscode.CancellationError())
+
+			await expect(handler.createMessage(systemPrompt, messages).next()).rejects.toThrow(
+				"Zoo Code <Language Model API>: Request cancelled by user",
+			)
+		})
+
+		it("should throw a Zoo Code branded error on stream error with error-like object", async () => {
+			const systemPrompt = "You are a helpful assistant"
+			const messages: Anthropic.Messages.MessageParam[] = [
+				{
+					role: "user" as const,
+					content: "Hello",
+				},
+			]
+
+			const consoleErrorSpy = vi.spyOn(console, "error").mockImplementation(() => {})
+
+			mockLanguageModelChat.sendRequest.mockRejectedValueOnce({ code: "STREAM_ERROR", details: "broken" })
+
+			await expect(handler.createMessage(systemPrompt, messages).next()).rejects.toThrow(
+				"Zoo Code <Language Model API>: Response stream error:",
+			)
+
+			expect(consoleErrorSpy).toHaveBeenCalledWith(
+				"Zoo Code <Language Model API>: Stream error object:",
+				expect.stringContaining("STREAM_ERROR"),
+			)
+
+			consoleErrorSpy.mockRestore()
+		})
 	})
 
 	describe("initializeClient", () => {
@@ -461,11 +502,18 @@ describe("VsCodeLmHandler", () => {
 		})
 
 		it("should return fallback model info when no client exists", () => {
+			const consoleDebugSpy = vi.spyOn(console, "debug").mockImplementation(() => {})
+
 			// Clear the client first
 			handler["client"] = null
 			const model = handler.getModel()
 			expect(model.id).toBe("test-vendor/test-family")
 			expect(model.info).toBeDefined()
+			expect(consoleDebugSpy).toHaveBeenCalledWith(
+				"Zoo Code <Language Model API>: No client available, using fallback model info",
+			)
+
+			consoleDebugSpy.mockRestore()
 		})
 
 		it("should return basic model info when client exists", async () => {
@@ -609,10 +657,17 @@ describe("VsCodeLmHandler", () => {
 			handler["client"] = null
 			handler["currentRequestCancellation"] = null
 
+			const consoleWarnSpy = vi.spyOn(console, "warn").mockImplementation(() => {})
+
 			const content: Anthropic.Messages.ContentBlockParam[] = [{ type: "text", text: "Hello" }]
 			const result = await handler.countTokens(content)
 
 			expect(result).toBe(0)
+			expect(consoleWarnSpy).toHaveBeenCalledWith(
+				"Zoo Code <Language Model API>: No client available for token counting",
+			)
+
+			consoleWarnSpy.mockRestore()
 		})
 
 		it("should handle image blocks with placeholder", async () => {

src/api/transform/__tests__/vscode-lm-format.spec.ts

@@ -434,4 +434,36 @@ describe("extractTextCountFromMessage", () => {
 		const result = extractTextCountFromMessage(message)
 		expect(result).toBe("result-id")
 	})
+
+	it("should log Zoo Code branded warning when tool call input stringify fails", () => {
+		const consoleErrorSpy = vi.spyOn(console, "error").mockImplementation(() => {})
+
+		// Create an object with a circular reference that will throw on JSON.stringify
+		// eslint-disable-next-line @typescript-eslint/no-explicit-any
+		const circularInput: any = { name: "circular" }
+		circularInput.self = circularInput
+
+		const mockToolCallPart = new (vitest.mocked(vscode).LanguageModelToolCallPart)(
+			"call-id",
+			"broken-tool",
+			circularInput,
+		)
+
+		// eslint-disable-next-line @typescript-eslint/no-explicit-any
+		const message: any = {
+			role: vscode.LanguageModelChatMessageRole.Assistant,
+			content: [mockToolCallPart],
+		}
+
+		const result = extractTextCountFromMessage(message)
+
+		// Should still return the tool name and callId even when input stringify fails
+		expect(result).toBe("broken-toolcall-id")
+		expect(consoleErrorSpy).toHaveBeenCalledWith(
+			"Zoo Code <Language Model API>: Failed to stringify tool call input:",
+			expect.any(Error),
+		)
+
+		consoleErrorSpy.mockRestore()
+	})
 })

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

@@ -544,6 +544,7 @@ describe("ClineProvider", () => {
 		})
 
 		expect(mockWebviewView.webview.html).toContain("<!DOCTYPE html>")
+		expect(mockWebviewView.webview.html).toContain("<title>Zoo Code</title>")
 	})
 
 	describe("logWebviewHiddenDiagnostics", () => {