feat(writeToFileTool): update diff view provider to use view column and reset with listeners

Author: felixAnhalt

src/core/tools/__tests__/writeToFileTool.test.ts

@@ -84,6 +84,8 @@ jest.mock("../../ignore/RooIgnoreController", () => ({
 	},
 }))
 
+const MOCK_VIEW_COLUMN = 1
+
 describe("writeToFileTool", () => {
 	// Test data
 	const testFilePath = "test/file.txt"
@@ -141,6 +143,7 @@ describe("writeToFileTool", () => {
 				finalContent: "final content",
 			}),
 			scrollToFirstDiff: jest.fn(),
+			resetWithListeners: jest.fn().mockResolvedValue(undefined),
 		}
 		mockCline.api = {
 			getModel: jest.fn().mockReturnValue({ id: "claude-3" }),
@@ -152,6 +155,11 @@ describe("writeToFileTool", () => {
 		mockCline.ask = jest.fn().mockResolvedValue(undefined)
 		mockCline.recordToolError = jest.fn()
 		mockCline.sayAndCreateMissingParamError = jest.fn().mockResolvedValue("Missing param error")
+		mockCline.providerRef = {
+			deref: jest.fn().mockReturnValue({
+				getViewColumn: jest.fn().mockReturnValue(MOCK_VIEW_COLUMN),
+			}),
+		}
 
 		mockAskApproval = jest.fn().mockResolvedValue(true)
 		mockHandleError = jest.fn().mockResolvedValue(undefined)
@@ -211,7 +219,7 @@ describe("writeToFileTool", () => {
 			await executeWriteFileTool({}, { accessAllowed: true })
 
 			expect(mockCline.rooIgnoreController.validateAccess).toHaveBeenCalledWith(testFilePath)
-			expect(mockCline.diffViewProvider.open).toHaveBeenCalledWith(testFilePath)
+			expect(mockCline.diffViewProvider.open).toHaveBeenCalledWith(testFilePath, MOCK_VIEW_COLUMN)
 		})
 	})
 
@@ -286,7 +294,7 @@ describe("writeToFileTool", () => {
 			await executeWriteFileTool({}, { fileExists: false })
 
 			expect(mockCline.consecutiveMistakeCount).toBe(0)
-			expect(mockCline.diffViewProvider.open).toHaveBeenCalledWith(testFilePath)
+			expect(mockCline.diffViewProvider.open).toHaveBeenCalledWith(testFilePath, MOCK_VIEW_COLUMN)
 			expect(mockCline.diffViewProvider.update).toHaveBeenCalledWith(testContent, true)
 			expect(mockAskApproval).toHaveBeenCalled()
 			expect(mockCline.diffViewProvider.saveChanges).toHaveBeenCalled()
@@ -327,7 +335,7 @@ describe("writeToFileTool", () => {
 			await executeWriteFileTool({}, { isPartial: true })
 
 			expect(mockCline.ask).toHaveBeenCalled()
-			expect(mockCline.diffViewProvider.open).toHaveBeenCalledWith(testFilePath)
+			expect(mockCline.diffViewProvider.open).toHaveBeenCalledWith(testFilePath, MOCK_VIEW_COLUMN)
 			expect(mockCline.diffViewProvider.update).toHaveBeenCalledWith(testContent, false)
 		})
 	})
@@ -365,7 +373,7 @@ describe("writeToFileTool", () => {
 			await executeWriteFileTool({})
 
 			expect(mockHandleError).toHaveBeenCalledWith("writing file", expect.any(Error))
-			expect(mockCline.diffViewProvider.reset).toHaveBeenCalled()
+			expect(mockCline.diffViewProvider.resetWithListeners).toHaveBeenCalled()
 		})
 
 		it("handles partial streaming errors", async () => {
@@ -374,7 +382,7 @@ describe("writeToFileTool", () => {
 			await executeWriteFileTool({}, { isPartial: true })
 
 			expect(mockHandleError).toHaveBeenCalledWith("writing file", expect.any(Error))
-			expect(mockCline.diffViewProvider.reset).toHaveBeenCalled()
+			expect(mockCline.diffViewProvider.resetWithListeners).toHaveBeenCalled()
 		})
 	})
 })