Fix tests

Author: EhabY

test/unit/webviews/tasks/tasksPanel.test.ts

@@ -1,4 +1,4 @@
-import { beforeEach, describe, expect, it, vi } from "vitest";
+import { beforeEach, describe, expect, it, vi, type Mock } from "vitest";
 import * as vscode from "vscode";
 
 import { TasksPanel } from "@/webviews/tasks/tasksPanel";
@@ -31,21 +31,21 @@ type TasksPanelClient = Pick<
 	| "getHost"
 >;
 
-type MockClient = { [K in keyof TasksPanelClient]: ReturnType<typeof vi.fn> };
+type MockClient = { [K in keyof TasksPanelClient]: Mock<TasksPanelClient[K]> };
 
 function createClient(baseUrl = "https://coder.example.com"): MockClient {
 	return {
 		getTasks: vi.fn().mockResolvedValue([]),
 		getTask: vi.fn(),
-		getTaskLogs: vi.fn().mockResolvedValue([]),
+		getTaskLogs: vi.fn().mockResolvedValue({ logs: [] }),
 		createTask: vi.fn(),
 		deleteTask: vi.fn().mockResolvedValue(undefined),
 		getTemplates: vi.fn().mockResolvedValue([]),
 		getTemplateVersionPresets: vi.fn().mockResolvedValue([]),
 		startWorkspace: vi.fn().mockResolvedValue(undefined),
 		stopWorkspace: vi.fn().mockResolvedValue(undefined),
 		getHost: vi.fn().mockReturnValue(baseUrl),
-	};
+	} as MockClient;
 }
 
 interface ApiDef {
@@ -247,9 +247,9 @@ describe("TasksPanel", () => {
 		it("returns task with logs", async () => {
 			const h = createHarness();
 			h.client.getTask.mockResolvedValue(task());
-			h.client.getTaskLogs.mockResolvedValue([
-				logEntry({ content: "Starting" }),
-			]);
+			h.client.getTaskLogs.mockResolvedValue({
+				logs: [logEntry({ content: "Starting" })],
+			});
 
 			const res = await h.request(TasksApi.getTaskDetails, {
 				taskId: "task-1",
@@ -306,7 +306,7 @@ describe("TasksPanel", () => {
 			h.client.getTask.mockResolvedValue(
 				task({ current_state: { timestamp: "", state, message: "", uri: "" } }),
 			);
-			h.client.getTaskLogs.mockResolvedValue([logEntry()]);
+			h.client.getTaskLogs.mockResolvedValue({ logs: [logEntry()] });
 
 			await h.request(TasksApi.getTaskDetails, { taskId: "task-1" });
 			await h.request(TasksApi.getTaskDetails, { taskId: "task-1" });
@@ -475,7 +475,7 @@ describe("TasksPanel", () => {
 	describe("downloadLogs", () => {
 		it("saves logs to file", async () => {
 			const h = createHarness();
-			h.client.getTaskLogs.mockResolvedValue([logEntry()]);
+			h.client.getTaskLogs.mockResolvedValue({ logs: [logEntry()] });
 			const saveUri = vscode.Uri.file("/downloads/logs.txt");
 			vi.mocked(vscode.window.showSaveDialog).mockResolvedValue(saveUri);
 			h.ui.setResponse(`Logs saved to ${saveUri.fsPath}`, "Open File");
@@ -494,7 +494,7 @@ describe("TasksPanel", () => {
 
 		it("does not open file when notification is dismissed", async () => {
 			const h = createHarness();
-			h.client.getTaskLogs.mockResolvedValue([logEntry()]);
+			h.client.getTaskLogs.mockResolvedValue({ logs: [logEntry()] });
 			const saveUri = vscode.Uri.file("/downloads/logs.txt");
 			vi.mocked(vscode.window.showSaveDialog).mockResolvedValue(saveUri);
 			h.ui.setResponse(`Logs saved to ${saveUri.fsPath}`, undefined);
@@ -513,7 +513,7 @@ describe("TasksPanel", () => {
 
 		it("shows warning when no logs", async () => {
 			const h = createHarness();
-			h.client.getTaskLogs.mockResolvedValue([]);
+			h.client.getTaskLogs.mockResolvedValue({ logs: [] });
 
 			const res = await h.request(TasksApi.downloadLogs, {
 				taskId: "task-1",
@@ -545,7 +545,7 @@ describe("TasksPanel", () => {
 
 		it("does nothing when user cancels", async () => {
 			const h = createHarness();
-			h.client.getTaskLogs.mockResolvedValue([logEntry()]);
+			h.client.getTaskLogs.mockResolvedValue({ logs: [logEntry()] });
 			vi.mocked(vscode.window.showSaveDialog).mockResolvedValue(undefined);
 
 			const res = await h.request(TasksApi.downloadLogs, {