@office-ai/aioncli-core 0.2.3 → 0.18.4

package/dist/src/tools/edit.test.js

@@ -7,7 +7,12 @@
 const mockEnsureCorrectEdit = vi.hoisted(() => vi.fn());
 const mockGenerateJson = vi.hoisted(() => vi.fn());
 const mockOpenDiff = vi.hoisted(() => vi.fn());
-import { IDEConnectionStatus } from '../ide/ide-client.js';
+import { IdeClient } from '../ide/ide-client.js';
+vi.mock('../ide/ide-client.js', () => ({
+    IdeClient: {
+        getInstance: vi.fn(),
+    },
+}));
 vi.mock('../utils/editCorrector.js', () => ({
     ensureCorrectEdit: mockEnsureCorrectEdit,
 }));
@@ -19,22 +24,26 @@ vi.mock('../core/client.js', () => ({
 vi.mock('../utils/editor.js', () => ({
     openDiff: mockOpenDiff,
 }));
+vi.mock('../telemetry/loggers.js', () => ({
+    logFileOperation: vi.fn(),
+}));
 import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest';
 import { applyReplacement, EditTool } from './edit.js';
 import { ToolConfirmationOutcome } from './tools.js';
 import { ToolErrorType } from './tool-error.js';
-import path from 'path';
-import fs from 'fs';
-import os from 'os';
-import { ApprovalMode } from '../config/config.js';
-import { createMockWorkspaceContext } from '../test-utils/mockWorkspaceContext.js';
+import path from 'node:path';
+import fs from 'node:fs';
+import os from 'node:os';
+import { ApprovalMode } from '../policy/types.js';
 import { StandardFileSystemService } from '../services/fileSystemService.js';
+import { WorkspaceContext } from '../utils/workspaceContext.js';
 describe('EditTool', () => {
     let tool;
     let tempDir;
     let rootDir;
     let mockConfig;
     let geminiClient;
+    let baseLlmClient;
     beforeEach(() => {
         vi.restoreAllMocks();
         tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'edit-tool-test-'));
@@ -43,14 +52,17 @@ describe('EditTool', () => {
         geminiClient = {
             generateJson: mockGenerateJson, // mockGenerateJson is already defined and hoisted
         };
+        baseLlmClient = {
+            generateJson: vi.fn(),
+        };
         mockConfig = {
             getGeminiClient: vi.fn().mockReturnValue(geminiClient),
+            getBaseLlmClient: vi.fn().mockReturnValue(baseLlmClient),
             getTargetDir: () => rootDir,
             getApprovalMode: vi.fn(),
             setApprovalMode: vi.fn(),
-            getWorkspaceContext: () => createMockWorkspaceContext(rootDir),
+            getWorkspaceContext: () => new WorkspaceContext(rootDir),
             getFileSystemService: () => new StandardFileSystemService(),
-            getIdeClient: () => undefined,
             getIdeMode: () => false,
             // getGeminiConfig: () => ({ apiKey: 'test-api-key' }), // This was not a real Config method
             // Add other properties/methods of Config if EditTool uses them
@@ -60,7 +72,6 @@ describe('EditTool', () => {
             getSandbox: () => false,
             getDebugMode: () => false,
             getQuestion: () => undefined,
-            getFullContext: () => false,
             getToolDiscoveryCommand: () => undefined,
             getToolCallCommand: () => undefined,
             getMcpServerCommand: () => undefined,
@@ -71,6 +82,7 @@ describe('EditTool', () => {
             getGeminiMdFileCount: () => 0,
             setGeminiMdFileCount: vi.fn(),
             getToolRegistry: () => ({}), // Minimal mock for ToolRegistry
+            isInteractive: () => false,
         };
         // Reset mocks before each test
         mockConfig.getApprovalMode.mockClear();
@@ -145,6 +157,81 @@ describe('EditTool', () => {
         it('should return currentContent if oldString is empty and not a new file', () => {
             expect(applyReplacement('hello world', '', 'new', false)).toBe('hello world');
         });
+        it.each([
+            {
+                name: '$ literal',
+                current: "price is $100 and pattern end is ' '",
+                oldStr: 'price is $100',
+                newStr: 'price is $200',
+                expected: "price is $200 and pattern end is ' '",
+            },
+            {
+                name: "$' literal",
+                current: 'foo',
+                oldStr: 'foo',
+                newStr: "bar$'baz",
+                expected: "bar$'baz",
+            },
+            {
+                name: '$& literal',
+                current: 'hello world',
+                oldStr: 'hello',
+                newStr: '$&-replacement',
+                expected: '$&-replacement world',
+            },
+            {
+                name: '$` literal',
+                current: 'prefix-middle-suffix',
+                oldStr: 'middle',
+                newStr: 'new$`content',
+                expected: 'prefix-new$`content-suffix',
+            },
+            {
+                name: '$1, $2 capture groups literal',
+                current: 'test string',
+                oldStr: 'test',
+                newStr: '$1$2replacement',
+                expected: '$1$2replacement string',
+            },
+            {
+                name: 'normal strings without problematic $',
+                current: 'normal text replacement',
+                oldStr: 'text',
+                newStr: 'string',
+                expected: 'normal string replacement',
+            },
+            {
+                name: 'multiple occurrences with $ sequences',
+                current: 'foo bar foo baz',
+                oldStr: 'foo',
+                newStr: "test$'end",
+                expected: "test$'end bar test$'end baz",
+            },
+            {
+                name: 'complex regex patterns with $ at end',
+                current: "| select('match', '^[sv]d[a-z]$')",
+                oldStr: "'^[sv]d[a-z]$'",
+                newStr: "'^[sv]d[a-z]$' # updated",
+                expected: "| select('match', '^[sv]d[a-z]$' # updated)",
+            },
+            {
+                name: 'empty replacement with problematic $',
+                current: 'test content',
+                oldStr: 'nothing',
+                newStr: "replacement$'text",
+                expected: 'test content',
+            },
+            {
+                name: '$$ (escaped dollar)',
+                current: 'price value',
+                oldStr: 'value',
+                newStr: '$$100',
+                expected: 'price $$100',
+            },
+        ])('should handle $name', ({ current, oldStr, newStr, expected }) => {
+            const result = applyReplacement(current, oldStr, newStr, false);
+            expect(result).toBe(expected);
+        });
     });
     describe('validateToolParams', () => {
         it('should return null for valid params', () => {
@@ -155,14 +242,6 @@ describe('EditTool', () => {
             };
             expect(tool.validateToolParams(params)).toBeNull();
         });
-        it('should return error for relative path', () => {
-            const params = {
-                file_path: 'test.txt',
-                old_string: 'old',
-                new_string: 'new',
-            };
-            expect(tool.validateToolParams(params)).toMatch(/File path must be absolute/);
-        });
         it('should return error for path outside root', () => {
             const params = {
                 file_path: path.join(tempDir, 'outside-root.txt'),
@@ -179,13 +258,25 @@ describe('EditTool', () => {
         beforeEach(() => {
             filePath = path.join(rootDir, testFile);
         });
-        it('should throw an error if params are invalid', async () => {
+        it('should resolve relative path and request confirmation', async () => {
+            fs.writeFileSync(filePath, 'some old content here');
             const params = {
-                file_path: 'relative.txt',
+                file_path: testFile, // relative path
                 old_string: 'old',
                 new_string: 'new',
             };
-            expect(() => tool.build(params)).toThrow();
+            // ensureCorrectEdit will be called by shouldConfirmExecute
+            mockEnsureCorrectEdit.mockResolvedValueOnce({
+                params: { ...params, file_path: filePath },
+                occurrences: 1,
+            });
+            const invocation = tool.build(params);
+            const confirmation = await invocation.shouldConfirmExecute(new AbortController().signal);
+            expect(confirmation).toEqual(expect.objectContaining({
+                title: `Confirm Edit: ${testFile}`,
+                fileName: testFile,
+                fileDiff: expect.any(String),
+            }));
         });
         it('should request confirmation for valid edit', async () => {
             fs.writeFileSync(filePath, 'some old content here');
@@ -264,11 +355,12 @@ describe('EditTool', () => {
             // The main beforeEach already calls mockEnsureCorrectEdit.mockReset()
             // Set a specific mock for this test case
             let mockCalled = false;
-            mockEnsureCorrectEdit.mockImplementationOnce(async (_, content, p, client) => {
+            mockEnsureCorrectEdit.mockImplementationOnce(async (_, content, p, client, baseClient) => {
                 mockCalled = true;
                 expect(content).toBe(originalContent);
                 expect(p).toBe(params);
                 expect(client).toBe(geminiClient);
+                expect(baseClient).toBe(baseLlmClient);
                 return {
                     params: {
                         file_path: filePath,
@@ -295,6 +387,27 @@ describe('EditTool', () => {
             correctedNewString);
             expect(patchedContent).toBe(expectedFinalContent);
         });
+        it('should rethrow calculateEdit errors when the abort signal is triggered', async () => {
+            const filePath = path.join(rootDir, 'abort-confirmation.txt');
+            const params = {
+                file_path: filePath,
+                old_string: 'old',
+                new_string: 'new',
+            };
+            const invocation = tool.build(params);
+            const abortController = new AbortController();
+            const abortError = new Error('Abort requested');
+            const calculateSpy = vi
+                .spyOn(invocation, 'calculateEdit')
+                .mockImplementation(async () => {
+                if (!abortController.signal.aborted) {
+                    abortController.abort();
+                }
+                throw abortError;
+            });
+            await expect(invocation.shouldConfirmExecute(abortController.signal)).rejects.toBe(abortError);
+            calculateSpy.mockRestore();
+        });
     });
     describe('execute', () => {
         const testFile = 'execute_me.txt';
@@ -317,13 +430,19 @@ describe('EditTool', () => {
                 return { params, occurrences };
             });
         });
-        it('should throw error if file path is not absolute', async () => {
+        it('should resolve relative path and execute successfully', async () => {
+            const initialContent = 'This is some old text.';
+            const newContent = 'This is some new text.';
+            fs.writeFileSync(filePath, initialContent, 'utf8');
             const params = {
-                file_path: 'relative.txt',
+                file_path: testFile, // relative path
                 old_string: 'old',
                 new_string: 'new',
             };
-            expect(() => tool.build(params)).toThrow(/File path must be absolute/);
+            const invocation = tool.build(params);
+            const result = await invocation.execute(new AbortController().signal);
+            expect(result.llmContent).toMatch(/Successfully modified file/);
+            expect(fs.readFileSync(filePath, 'utf8')).toBe(newContent);
         });
         it('should throw error if file path is empty', async () => {
             const params = {
@@ -333,6 +452,26 @@ describe('EditTool', () => {
             };
             expect(() => tool.build(params)).toThrow(/The 'file_path' parameter must be non-empty./);
         });
+        it('should reject when calculateEdit fails after an abort signal', async () => {
+            const params = {
+                file_path: path.join(rootDir, 'abort-execute.txt'),
+                old_string: 'old',
+                new_string: 'new',
+            };
+            const invocation = tool.build(params);
+            const abortController = new AbortController();
+            const abortError = new Error('Abort requested during execute');
+            const calculateSpy = vi
+                .spyOn(invocation, 'calculateEdit')
+                .mockImplementation(async () => {
+                if (!abortController.signal.aborted) {
+                    abortController.abort();
+                }
+                throw abortError;
+            });
+            await expect(invocation.execute(abortController.signal)).rejects.toBe(abortError);
+            calculateSpy.mockRestore();
+        });
         it('should edit an existing file and return diff with fileName', async () => {
             const initialContent = 'This is some old text.';
             const newContent = 'This is some new text.'; // old -> new
@@ -369,7 +508,19 @@ describe('EditTool', () => {
             expect(result.llmContent).toMatch(/Created new file/);
             expect(fs.existsSync(newFilePath)).toBe(true);
             expect(fs.readFileSync(newFilePath, 'utf8')).toBe(fileContent);
-            expect(result.returnDisplay).toBe(`Created ${newFileName}`);
+            const display = result.returnDisplay;
+            expect(display.fileDiff).toMatch(/\+Content for the new file\./);
+            expect(display.fileName).toBe(newFileName);
+            expect(result.returnDisplay.diffStat).toStrictEqual({
+                model_added_lines: 1,
+                model_removed_lines: 0,
+                model_added_chars: 25,
+                model_removed_chars: 0,
+                user_added_lines: 0,
+                user_removed_lines: 0,
+                user_added_chars: 0,
+                user_removed_chars: 0,
+            });
         });
         it('should return error if old_string is not found in file', async () => {
             fs.writeFileSync(filePath, 'Some content.', 'utf8');
@@ -398,7 +549,7 @@ describe('EditTool', () => {
             expect(result.returnDisplay).toMatch(/Failed to edit, expected 1 occurrence but found 2/);
         });
         it('should successfully replace multiple occurrences when expected_replacements specified', async () => {
-            fs.writeFileSync(filePath, 'old text old text old text', 'utf8');
+            fs.writeFileSync(filePath, 'old text\nold text\nold text', 'utf8');
             const params = {
                 file_path: filePath,
                 old_string: 'old',
@@ -408,11 +559,21 @@ describe('EditTool', () => {
             const invocation = tool.build(params);
             const result = await invocation.execute(new AbortController().signal);
             expect(result.llmContent).toMatch(/Successfully modified file/);
-            expect(fs.readFileSync(filePath, 'utf8')).toBe('new text new text new text');
+            expect(fs.readFileSync(filePath, 'utf8')).toBe('new text\nnew text\nnew text');
             const display = result.returnDisplay;
-            expect(display.fileDiff).toMatch(/old text old text old text/);
-            expect(display.fileDiff).toMatch(/new text new text new text/);
+            expect(display.fileDiff).toMatch(/-old text\n-old text\n-old text/);
+            expect(display.fileDiff).toMatch(/\+new text\n\+new text\n\+new text/);
             expect(display.fileName).toBe(testFile);
+            expect(result.returnDisplay.diffStat).toStrictEqual({
+                model_added_lines: 3,
+                model_removed_lines: 3,
+                model_added_chars: 24,
+                model_removed_chars: 24,
+                user_added_lines: 0,
+                user_removed_lines: 0,
+                user_added_chars: 0,
+                user_removed_chars: 0,
+            });
         });
         it('should return error if expected_replacements does not match actual occurrences', async () => {
             fs.writeFileSync(filePath, 'old text old text', 'utf8');
@@ -440,40 +601,49 @@ describe('EditTool', () => {
             expect(result.returnDisplay).toMatch(/Attempted to create a file that already exists/);
         });
         it('should include modification message when proposed content is modified', async () => {
-            const initialContent = 'This is some old text.';
+            const initialContent = 'Line 1\nold line\nLine 3\nLine 4\nLine 5\n';
             fs.writeFileSync(filePath, initialContent, 'utf8');
             const params = {
                 file_path: filePath,
                 old_string: 'old',
                 new_string: 'new',
                 modified_by_user: true,
+                ai_proposed_content: 'Line 1\nAI line\nLine 3\nLine 4\nLine 5\n',
             };
             mockConfig.getApprovalMode.mockReturnValueOnce(ApprovalMode.AUTO_EDIT);
             const invocation = tool.build(params);
             const result = await invocation.execute(new AbortController().signal);
             expect(result.llmContent).toMatch(/User modified the `new_string` content/);
+            expect(result.returnDisplay.diffStat).toStrictEqual({
+                model_added_lines: 1,
+                model_removed_lines: 1,
+                model_added_chars: 7,
+                model_removed_chars: 8,
+                user_added_lines: 1,
+                user_removed_lines: 1,
+                user_added_chars: 8,
+                user_removed_chars: 7,
+            });
         });
-        it('should not include modification message when proposed content is not modified', async () => {
-            const initialContent = 'This is some old text.';
-            fs.writeFileSync(filePath, initialContent, 'utf8');
-            const params = {
-                file_path: filePath,
-                old_string: 'old',
-                new_string: 'new',
-                modified_by_user: false,
-            };
-            mockConfig.getApprovalMode.mockReturnValueOnce(ApprovalMode.AUTO_EDIT);
-            const invocation = tool.build(params);
-            const result = await invocation.execute(new AbortController().signal);
-            expect(result.llmContent).not.toMatch(/User modified the `new_string` content/);
-        });
-        it('should not include modification message when modified_by_user is not provided', async () => {
+        it.each([
+            {
+                name: 'modified_by_user is false',
+                modifiedByUser: false,
+            },
+            {
+                name: 'modified_by_user is not provided',
+                modifiedByUser: undefined,
+            },
+        ])('should not include modification message when $name', async ({ modifiedByUser }) => {
             const initialContent = 'This is some old text.';
             fs.writeFileSync(filePath, initialContent, 'utf8');
             const params = {
                 file_path: filePath,
                 old_string: 'old',
                 new_string: 'new',
+                ...(modifiedByUser !== undefined && {
+                    modified_by_user: modifiedByUser,
+                }),
             };
             mockConfig.getApprovalMode.mockReturnValueOnce(ApprovalMode.AUTO_EDIT);
             const invocation = tool.build(params);
@@ -493,133 +663,154 @@ describe('EditTool', () => {
             expect(result.llmContent).toMatch(/No changes to apply/);
             expect(result.returnDisplay).toMatch(/No changes to apply/);
         });
-    });
-    describe('Error Scenarios', () => {
-        const testFile = 'error_test.txt';
-        let filePath;
-        beforeEach(() => {
-            filePath = path.join(rootDir, testFile);
-        });
-        it('should return FILE_NOT_FOUND error', async () => {
-            const params = {
-                file_path: filePath,
-                old_string: 'any',
-                new_string: 'new',
-            };
-            const invocation = tool.build(params);
-            const result = await invocation.execute(new AbortController().signal);
-            expect(result.error?.type).toBe(ToolErrorType.FILE_NOT_FOUND);
-        });
-        it('should return ATTEMPT_TO_CREATE_EXISTING_FILE error', async () => {
-            fs.writeFileSync(filePath, 'existing content', 'utf8');
-            const params = {
-                file_path: filePath,
-                old_string: '',
-                new_string: 'new content',
-            };
-            const invocation = tool.build(params);
-            const result = await invocation.execute(new AbortController().signal);
-            expect(result.error?.type).toBe(ToolErrorType.ATTEMPT_TO_CREATE_EXISTING_FILE);
-        });
-        it('should return NO_OCCURRENCE_FOUND error', async () => {
-            fs.writeFileSync(filePath, 'content', 'utf8');
-            const params = {
-                file_path: filePath,
-                old_string: 'not-found',
-                new_string: 'new',
-            };
-            const invocation = tool.build(params);
-            const result = await invocation.execute(new AbortController().signal);
-            expect(result.error?.type).toBe(ToolErrorType.EDIT_NO_OCCURRENCE_FOUND);
-        });
-        it('should return EXPECTED_OCCURRENCE_MISMATCH error', async () => {
-            fs.writeFileSync(filePath, 'one one two', 'utf8');
-            const params = {
-                file_path: filePath,
-                old_string: 'one',
-                new_string: 'new',
-                expected_replacements: 3,
-            };
-            const invocation = tool.build(params);
-            const result = await invocation.execute(new AbortController().signal);
-            expect(result.error?.type).toBe(ToolErrorType.EDIT_EXPECTED_OCCURRENCE_MISMATCH);
-        });
-        it('should return NO_CHANGE error', async () => {
-            fs.writeFileSync(filePath, 'content', 'utf8');
+        it('should return EDIT_NO_CHANGE error if replacement results in identical content', async () => {
+            // This can happen if ensureCorrectEdit finds a fuzzy match, but the literal
+            // string replacement with `replaceAll` results in no change.
+            const initialContent = 'line 1\nline  2\nline 3'; // Note the double space
+            fs.writeFileSync(filePath, initialContent, 'utf8');
             const params = {
                 file_path: filePath,
-                old_string: 'content',
-                new_string: 'content',
+                // old_string has a single space, so it won't be found by replaceAll
+                old_string: 'line 1\nline 2\nline 3',
+                new_string: 'line 1\nnew line 2\nline 3',
             };
+            // Mock ensureCorrectEdit to simulate it finding a match (e.g., via fuzzy matching)
+            // but it doesn't correct the old_string to the literal content.
+            mockEnsureCorrectEdit.mockResolvedValueOnce({ params, occurrences: 1 });
             const invocation = tool.build(params);
             const result = await invocation.execute(new AbortController().signal);
             expect(result.error?.type).toBe(ToolErrorType.EDIT_NO_CHANGE);
+            expect(result.returnDisplay).toMatch(/No changes to apply. The new content is identical to the current content./);
+            // Ensure the file was not actually changed
+            expect(fs.readFileSync(filePath, 'utf8')).toBe(initialContent);
         });
-        it('should throw INVALID_PARAMETERS error for relative path', async () => {
-            const params = {
-                file_path: 'relative/path.txt',
-                old_string: 'a',
-                new_string: 'b',
-            };
-            expect(() => tool.build(params)).toThrow();
+    });
+    describe('Error Scenarios', () => {
+        const testFile = 'error_test.txt';
+        let filePath;
+        beforeEach(() => {
+            filePath = path.join(rootDir, testFile);
         });
-        it('should return FILE_WRITE_FAILURE on write error', async () => {
-            fs.writeFileSync(filePath, 'content', 'utf8');
-            // Make file readonly to trigger a write error
-            fs.chmodSync(filePath, '444');
-            const params = {
-                file_path: filePath,
-                old_string: 'content',
-                new_string: 'new content',
-            };
-            const invocation = tool.build(params);
-            const result = await invocation.execute(new AbortController().signal);
-            expect(result.error?.type).toBe(ToolErrorType.FILE_WRITE_FAILURE);
+        it.each([
+            {
+                name: 'FILE_NOT_FOUND error',
+                setup: () => { },
+                params: { file_path: '', old_string: 'any', new_string: 'new' },
+                expectedError: ToolErrorType.FILE_NOT_FOUND,
+                isAsyncTest: true,
+            },
+            {
+                name: 'ATTEMPT_TO_CREATE_EXISTING_FILE error',
+                setup: (fp) => fs.writeFileSync(fp, 'existing content', 'utf8'),
+                params: { file_path: '', old_string: '', new_string: 'new content' },
+                expectedError: ToolErrorType.ATTEMPT_TO_CREATE_EXISTING_FILE,
+                isAsyncTest: true,
+            },
+            {
+                name: 'NO_OCCURRENCE_FOUND error',
+                setup: (fp) => fs.writeFileSync(fp, 'content', 'utf8'),
+                params: { file_path: '', old_string: 'not-found', new_string: 'new' },
+                expectedError: ToolErrorType.EDIT_NO_OCCURRENCE_FOUND,
+                isAsyncTest: true,
+            },
+            {
+                name: 'EXPECTED_OCCURRENCE_MISMATCH error',
+                setup: (fp) => fs.writeFileSync(fp, 'one one two', 'utf8'),
+                params: {
+                    file_path: '',
+                    old_string: 'one',
+                    new_string: 'new',
+                    expected_replacements: 3,
+                },
+                expectedError: ToolErrorType.EDIT_EXPECTED_OCCURRENCE_MISMATCH,
+                isAsyncTest: true,
+            },
+            {
+                name: 'NO_CHANGE error',
+                setup: (fp) => fs.writeFileSync(fp, 'content', 'utf8'),
+                params: { file_path: '', old_string: 'content', new_string: 'content' },
+                expectedError: ToolErrorType.EDIT_NO_CHANGE,
+                isAsyncTest: true,
+            },
+            {
+                name: 'relative path (should not throw)',
+                setup: () => { },
+                params: {
+                    file_path: 'relative/path.txt',
+                    old_string: 'a',
+                    new_string: 'b',
+                },
+                expectedError: null,
+                isAsyncTest: false,
+            },
+            {
+                name: 'FILE_WRITE_FAILURE on write error',
+                setup: (fp) => {
+                    fs.writeFileSync(fp, 'content', 'utf8');
+                    fs.chmodSync(fp, '444');
+                },
+                params: {
+                    file_path: '',
+                    old_string: 'content',
+                    new_string: 'new content',
+                },
+                expectedError: ToolErrorType.FILE_WRITE_FAILURE,
+                isAsyncTest: true,
+            },
+        ])('should return $name', async ({ setup, params, expectedError, isAsyncTest }) => {
+            const testParams = {
+                ...params,
+                file_path: params.file_path || filePath,
+            };
+            setup(filePath);
+            if (!isAsyncTest) {
+                expect(() => tool.build(testParams)).not.toThrow();
+            }
+            else {
+                const invocation = tool.build(testParams);
+                const result = await invocation.execute(new AbortController().signal);
+                expect(result.error?.type).toBe(expectedError);
+            }
         });
     });
     describe('getDescription', () => {
-        it('should return "No file changes to..." if old_string and new_string are the same', () => {
-            const testFileName = 'test.txt';
-            const params = {
-                file_path: path.join(rootDir, testFileName),
-                old_string: 'identical_string',
-                new_string: 'identical_string',
-            };
-            const invocation = tool.build(params);
-            // shortenPath will be called internally, resulting in just the file name
-            expect(invocation.getDescription()).toBe(`No file changes to ${testFileName}`);
-        });
-        it('should return a snippet of old and new strings if they are different', () => {
-            const testFileName = 'test.txt';
-            const params = {
-                file_path: path.join(rootDir, testFileName),
-                old_string: 'this is the old string value',
-                new_string: 'this is the new string value',
-            };
-            const invocation = tool.build(params);
-            // shortenPath will be called internally, resulting in just the file name
-            // The snippets are truncated at 30 chars + '...'
-            expect(invocation.getDescription()).toBe(`${testFileName}: this is the old string value => this is the new string value`);
-        });
-        it('should handle very short strings correctly in the description', () => {
-            const testFileName = 'short.txt';
-            const params = {
-                file_path: path.join(rootDir, testFileName),
-                old_string: 'old',
-                new_string: 'new',
+        it.each([
+            {
+                name: 'identical strings (no change)',
+                fileName: 'test.txt',
+                oldStr: 'identical_string',
+                newStr: 'identical_string',
+                expected: 'No file changes to test.txt',
+            },
+            {
+                name: 'different strings (full)',
+                fileName: 'test.txt',
+                oldStr: 'this is the old string value',
+                newStr: 'this is the new string value',
+                expected: 'test.txt: this is the old string value => this is the new string value',
+            },
+            {
+                name: 'very short strings',
+                fileName: 'short.txt',
+                oldStr: 'old',
+                newStr: 'new',
+                expected: 'short.txt: old => new',
+            },
+            {
+                name: 'long strings (truncated)',
+                fileName: 'long.txt',
+                oldStr: 'this is a very long old string that will definitely be truncated',
+                newStr: 'this is a very long new string that will also be truncated',
+                expected: 'long.txt: this is a very long old string... => this is a very long new string...',
+            },
+        ])('should handle $name', ({ fileName, oldStr, newStr, expected }) => {
+            const params = {
+                file_path: path.join(rootDir, fileName),
+                old_string: oldStr,
+                new_string: newStr,
             };
             const invocation = tool.build(params);
-            expect(invocation.getDescription()).toBe(`${testFileName}: old => new`);
-        });
-        it('should truncate long strings in the description', () => {
-            const testFileName = 'long.txt';
-            const params = {
-                file_path: path.join(rootDir, testFileName),
-                old_string: 'this is a very long old string that will definitely be truncated',
-                new_string: 'this is a very long new string that will also be truncated',
-            };
-            const invocation = tool.build(params);
-            expect(invocation.getDescription()).toBe(`${testFileName}: this is a very long old string... => this is a very long new string...`);
+            expect(invocation.getDescription()).toBe(expected);
         });
     });
     describe('workspace boundary validation', () => {
@@ -642,6 +833,25 @@ describe('EditTool', () => {
             expect(error).toContain(rootDir);
         });
     });
+    describe('constructor', () => {
+        afterEach(() => {
+            vi.restoreAllMocks();
+        });
+        it('should use windows-style path examples on windows', () => {
+            vi.spyOn(process, 'platform', 'get').mockReturnValue('win32');
+            const tool = new EditTool({});
+            const schema = tool.schema;
+            expect(schema.parametersJsonSchema.properties
+                .file_path.description).toBe('The path to the file to modify.');
+        });
+        it('should use unix-style path examples on non-windows platforms', () => {
+            vi.spyOn(process, 'platform', 'get').mockReturnValue('linux');
+            const tool = new EditTool({});
+            const schema = tool.schema;
+            expect(schema.parametersJsonSchema.properties
+                .file_path.description).toBe('The path to the file to modify.');
+        });
+    });
     describe('IDE mode', () => {
         const testFile = 'edit_me.txt';
         let filePath;
@@ -650,12 +860,10 @@ describe('EditTool', () => {
             filePath = path.join(rootDir, testFile);
             ideClient = {
                 openDiff: vi.fn(),
-                getConnectionStatus: vi.fn().mockReturnValue({
-                    status: IDEConnectionStatus.Connected,
-                }),
+                isDiffingEnabled: vi.fn().mockReturnValue(true),
             };
+            vi.mocked(IdeClient.getInstance).mockResolvedValue(ideClient);
             mockConfig.getIdeMode = () => true;
-            mockConfig.getIdeClient = () => ideClient;
         });
         it('should call ideClient.openDiff and update params on confirmation', async () => {
             const initialContent = 'some old content here';
@@ -685,5 +893,63 @@ describe('EditTool', () => {
             expect(params.new_string).toBe(modifiedContent);
         });
     });
+    describe('multiple file edits', () => {
+        it('should perform multiple removals and report correct diff stats', async () => {
+            const numFiles = 10;
+            const files = [];
+            const expectedLinesRemoved = [];
+            const actualLinesRemoved = [];
+            // 1. Create 10 files with 5-10 lines each
+            for (let i = 0; i < numFiles; i++) {
+                const fileName = `test-file-${i}.txt`;
+                const filePath = path.join(rootDir, fileName);
+                const numLines = Math.floor(Math.random() * 6) + 5; // 5 to 10 lines
+                const lines = Array.from({ length: numLines }, (_, j) => `File ${i}, Line ${j + 1}`);
+                const content = lines.join('\n') + '\n';
+                // Determine which lines to remove (2 or 3 lines)
+                const numLinesToRemove = Math.floor(Math.random() * 2) + 2; // 2 or 3
+                expectedLinesRemoved.push(numLinesToRemove);
+                const startLineToRemove = 1; // Start removing from the second line
+                const linesToRemove = lines.slice(startLineToRemove, startLineToRemove + numLinesToRemove);
+                const toRemove = linesToRemove.join('\n') + '\n';
+                fs.writeFileSync(filePath, content, 'utf8');
+                files.push({
+                    path: filePath,
+                    initialContent: content,
+                    toRemove,
+                });
+            }
+            // 2. Create and execute 10 tool calls for removal
+            for (const file of files) {
+                const params = {
+                    file_path: file.path,
+                    old_string: file.toRemove,
+                    new_string: '', // Removing the content
+                };
+                const invocation = tool.build(params);
+                const result = await invocation.execute(new AbortController().signal);
+                if (result.returnDisplay &&
+                    typeof result.returnDisplay === 'object' &&
+                    'diffStat' in result.returnDisplay &&
+                    result.returnDisplay.diffStat) {
+                    actualLinesRemoved.push(result.returnDisplay.diffStat?.model_removed_lines);
+                }
+                else if (result.error) {
+                    console.error(`Edit failed for ${file.path}:`, result.error);
+                }
+            }
+            // 3. Assert that the content was removed from each file
+            for (const file of files) {
+                const finalContent = fs.readFileSync(file.path, 'utf8');
+                const expectedContent = file.initialContent.replace(file.toRemove, '');
+                expect(finalContent).toBe(expectedContent);
+                expect(finalContent).not.toContain(file.toRemove);
+            }
+            // 4. Assert that the total number of removed lines matches the diffStat total
+            const totalExpectedRemoved = expectedLinesRemoved.reduce((sum, current) => sum + current, 0);
+            const totalActualRemoved = actualLinesRemoved.reduce((sum, current) => sum + current, 0);
+            expect(totalActualRemoved).toBe(totalExpectedRemoved);
+        });
+    });
 });
 //# sourceMappingURL=edit.test.js.map