From 709a7029dace004cbe38a73e0072a47d95d3c33b Mon Sep 17 00:00:00 2001 From: Shawn <5414767+playcations@users.noreply.github.com> Date: Sun, 31 Aug 2025 16:15:11 -0400 Subject: [PATCH] Fix type issues in FCOMessageHandler MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - Change getCurrentCline() to getCurrentTask() - Fix Promise return type in test mock 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude --- src/services/file-changes/FCOMessageHandler.ts | 2 +- src/services/file-changes/__tests__/FCOMessageHandler.test.ts | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/src/services/file-changes/FCOMessageHandler.ts b/src/services/file-changes/FCOMessageHandler.ts index 95f550a079..ceb54064a5 100644 --- a/src/services/file-changes/FCOMessageHandler.ts +++ b/src/services/file-changes/FCOMessageHandler.ts @@ -413,7 +413,7 @@ export class FCOMessageHandler { // Detect enable event (transition from false to true) during active task if (!previousFilesChangedEnabled && filesChangedEnabled) { - const currentTask = this.provider.getCurrentCline() + const currentTask = this.provider.getCurrentTask() if (currentTask && currentTask.taskId) { try { await this.handleFCOEnableResetBaseline(currentTask) diff --git a/src/services/file-changes/__tests__/FCOMessageHandler.test.ts b/src/services/file-changes/__tests__/FCOMessageHandler.test.ts index 15d22fa5c0..5195ceb59e 100644 --- a/src/services/file-changes/__tests__/FCOMessageHandler.test.ts +++ b/src/services/file-changes/__tests__/FCOMessageHandler.test.ts @@ -75,7 +75,9 @@ describe("FCOMessageHandler", () => { vi.clearAllMocks() // Setup getCheckpointService mock - vi.mocked(getCheckpointService).mockImplementation((task) => task?.checkpointService || undefined) + vi.mocked(getCheckpointService).mockImplementation((task) => + Promise.resolve(task?.checkpointService || undefined), + ) // Reset checkpointSave mock vi.mocked(checkpointSave).mockReset()