fix: fix failed test

Author: rrewll

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

@@ -89,6 +89,9 @@ describe("VsCodeLmHandler", () => {
 
 	beforeEach(() => {
 		vi.clearAllMocks()
+		// Set up a default successful mock for selectChatModels before creating the handler
+		const mockModel = { ...mockLanguageModelChat }
+		;(vscode.lm.selectChatModels as Mock).mockResolvedValue([mockModel])
 		handler = new VsCodeLmHandler(defaultOptions)
 	})
 
@@ -733,13 +736,13 @@ describe("VsCodeLmHandler", () => {
 
 		it("should throw a Zoo Code branded error when client initialization fails", async () => {
 			const consoleErrorSpy = vi.spyOn(console, "error").mockImplementation(() => {})
-			// Use mockRejectedValue (not Once) because beforeEach already called selectChatModels once
 			;(vscode.lm.selectChatModels as Mock).mockRejectedValue(new Error("select failed"))
 			handler["client"] = null
 
-			await expect(handler.initializeClient()).rejects.toThrow(
-				"Zoo Code <Language Model API>: Failed to initialize client:",
-			)
+			// Catch the unhandled rejection that may occur from the constructor's async call
+			const initPromise = handler.initializeClient()
+
+			await expect(initPromise).rejects.toThrow("Zoo Code <Language Model API>: Failed to initialize client:")
 
 			expect(consoleErrorSpy).toHaveBeenCalledWith(
 				"Zoo Code <Language Model API>: Client initialization failed:",