summaryrefslogtreecommitdiff
path: root/packages/core/src/tools/edit.test.ts
diff options
context:
space:
mode:
authorN. Taylor Mullen <[email protected]>2025-06-02 22:30:52 -0700
committerGitHub <[email protected]>2025-06-03 05:30:52 +0000
commit8ab74ef1bb4bd1c475596088d9d3b52e0a9c5ca7 (patch)
treeb5847fc0cf03d491a3bb0bb18f8cbbf002d096ea /packages/core/src/tools/edit.test.ts
parentcf84f1af6854a99975903192b5770c790521be55 (diff)
Refactor: Use config.getGeminiClient() for GeminiClient instantiation (#715)
Diffstat (limited to 'packages/core/src/tools/edit.test.ts')
-rw-r--r--packages/core/src/tools/edit.test.ts8
1 files changed, 8 insertions, 0 deletions
diff --git a/packages/core/src/tools/edit.test.ts b/packages/core/src/tools/edit.test.ts
index 3b93708a..87ae3b6f 100644
--- a/packages/core/src/tools/edit.test.ts
+++ b/packages/core/src/tools/edit.test.ts
@@ -39,7 +39,15 @@ describe('EditTool', () => {
rootDir = path.join(tempDir, 'root');
fs.mkdirSync(rootDir);
+ // The client instance that EditTool will use
+ const mockClientInstanceWithGenerateJson = {
+ generateJson: mockGenerateJson, // mockGenerateJson is already defined and hoisted
+ };
+
mockConfig = {
+ getGeminiClient: vi
+ .fn()
+ .mockReturnValue(mockClientInstanceWithGenerateJson),
getTargetDir: () => rootDir,
getApprovalMode: vi.fn(() => false),
setApprovalMode: vi.fn(),