Skip to content

Commit 477409b

Browse files
committed
test(webview-handler): mock resolveMessageTask on mockClineProvider
1 parent e0768d7 commit 477409b

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

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

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,7 @@ const mockClineProvider = {
8080
log: vi.fn(),
8181
postStateToWebview: vi.fn(),
8282
getCurrentTask: vi.fn(),
83+
resolveMessageTask: vi.fn(),
8384
getTaskWithId: vi.fn(),
8485
createTaskWithHistoryItem: vi.fn(),
8586
getSkillsManager: vi.fn(),
@@ -221,11 +222,14 @@ describe("webviewMessageHandler - image mentions", () => {
221222

222223
it("should resolve image mentions for askResponse payloads", async () => {
223224
const mockHandleWebviewAskResponse = vi.fn()
224-
vi.mocked(mockClineProvider.getCurrentTask).mockReturnValue({
225+
const mockTask = {
226+
taskId: "task-1",
225227
cwd: "/mock/workspace",
226228
rooIgnoreController: undefined,
227229
handleWebviewAskResponse: mockHandleWebviewAskResponse,
228-
} as any)
230+
} as any
231+
vi.mocked(mockClineProvider.getCurrentTask).mockReturnValue(mockTask)
232+
vi.mocked(mockClineProvider.resolveMessageTask).mockReturnValue(mockTask)
229233

230234
await webviewMessageHandler(mockClineProvider, {
231235
type: "askResponse",

0 commit comments

Comments
 (0)