test: bump coverage for Roo-to-Zoo user-facing string changes (#971)

Author: roomote

src/api/providers/__tests__/lmstudio.spec.ts

@@ -125,7 +125,9 @@ describe("LmStudioHandler", () => {
 				for await (const _chunk of stream) {
 					// Should not reach here
 				}
-			}).rejects.toThrow("Please check the LM Studio developer logs to debug what went wrong")
+			}).rejects.toThrow(
+				"Please check the LM Studio developer logs to debug what went wrong. You may need to load the model with a larger context length to work with Zoo Code's prompts.",
+			)
 		})
 	})
 
@@ -144,7 +146,7 @@ describe("LmStudioHandler", () => {
 		it("should handle API errors", async () => {
 			mockCreate.mockRejectedValueOnce(new Error("API Error"))
 			await expect(handler.completePrompt("Test prompt")).rejects.toThrow(
-				"Please check the LM Studio developer logs to debug what went wrong",
+				"Please check the LM Studio developer logs to debug what went wrong. You may need to load the model with a larger context length to work with Zoo Code's prompts.",
 			)
 		})
 

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

@@ -397,6 +397,52 @@ describe("VsCodeLmHandler", () => {
 
 			await expect(handler.createMessage(systemPrompt, messages).next()).rejects.toThrow("API Error")
 		})
+
+		it("should brand the LM authorization justification as Zoo Code", async () => {
+			const systemPrompt = "You are a helpful assistant"
+			const messages: Anthropic.Messages.MessageParam[] = [
+				{
+					role: "user" as const,
+					content: "Hello",
+				},
+			]
+
+			mockLanguageModelChat.sendRequest.mockResolvedValueOnce({
+				stream: (async function* () {
+					yield new vscode.LanguageModelTextPart("Hi")
+					return
+				})(),
+				text: (async function* () {
+					yield "Hi"
+					return
+				})(),
+			})
+
+			const stream = handler.createMessage(systemPrompt, messages)
+			for await (const _chunk of stream) {
+				// drain
+			}
+
+			expect(mockLanguageModelChat.sendRequest).toHaveBeenCalledWith(
+				expect.any(Array),
+				expect.objectContaining({
+					justification:
+						"Zoo Code would like to use 'Test Model' from 'test-vendor', Click 'Allow' to proceed.",
+				}),
+				expect.anything(),
+			)
+		})
+	})
+
+	describe("initializeClient", () => {
+		it("should throw a Zoo Code branded error when client initialization fails", async () => {
+			;(vscode.lm.selectChatModels as Mock).mockRejectedValueOnce(new Error("select failed"))
+			handler["client"] = null
+
+			await expect(handler.initializeClient()).rejects.toThrow(
+				"Zoo Code <Language Model API>: Failed to initialize client: Zoo Code <Language Model API>: Failed to select model: select failed",
+			)
+		})
 	})
 
 	describe("getModel", () => {

src/i18n/__tests__/tools-missing-param.spec.ts

@@ -0,0 +1,46 @@
+import fs from "fs"
+import path from "path"
+
+const localesDir = path.join(__dirname, "..", "locales")
+
+const languages = fs
+	.readdirSync(localesDir, { withFileTypes: true })
+	.filter((dirent) => dirent.isDirectory() && !dirent.name.startsWith("."))
+	.map((dirent) => dirent.name)
+
+function loadTools(language: string): Record<string, string> {
+	return JSON.parse(fs.readFileSync(path.join(localesDir, language, "tools.json"), "utf8"))
+}
+
+describe("tools:missingToolParameter locales", () => {
+	it("covers every shipped locale", () => {
+		expect(languages.length).toBeGreaterThan(0)
+		for (const language of languages) {
+			const tools = loadTools(language)
+			expect(tools.missingToolParameter, `missingToolParameter missing in ${language}`).toBeTruthy()
+			expect(
+				tools.missingToolParameterWithPath,
+				`missingToolParameterWithPath missing in ${language}`,
+			).toBeTruthy()
+		}
+	})
+
+	it("keeps all interpolation placeholders in every locale", () => {
+		for (const language of languages) {
+			const tools = loadTools(language)
+			expect(tools.missingToolParameter, `placeholders missing in ${language}`).toContain("{{toolName}}")
+			expect(tools.missingToolParameter, `placeholders missing in ${language}`).toContain("{{paramName}}")
+			expect(tools.missingToolParameterWithPath, `placeholders missing in ${language}`).toContain("{{toolName}}")
+			expect(tools.missingToolParameterWithPath, `placeholders missing in ${language}`).toContain("{{relPath}}")
+			expect(tools.missingToolParameterWithPath, `placeholders missing in ${language}`).toContain("{{paramName}}")
+		}
+	})
+
+	it("brands the English messages as Zoo, not Roo", () => {
+		const tools = loadTools("en")
+		expect(tools.missingToolParameter).toContain("Zoo")
+		expect(tools.missingToolParameter).not.toContain("Roo")
+		expect(tools.missingToolParameterWithPath).toContain("Zoo")
+		expect(tools.missingToolParameterWithPath).not.toContain("Roo")
+	})
+})