From 7eba1c463541a40b1a2f1a58a81b3e69f42339a4 Mon Sep 17 00:00:00 2001
From: Reid <61492567+reidliu41@users.noreply.github.com>
Date: Sat, 18 Apr 2026 13:46:46 +0800
Subject: [PATCH]   test(core): update scheduler registry mock (#3415)

Update the CoreToolScheduler retry-loop test registry mock to match the current
  ToolRegistry interface. Add ensureTool and getAllToolNames so the tests exercise
  the scheduler path used in production.
---
 packages/core/src/core/coreToolScheduler.test.ts | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/packages/core/src/core/coreToolScheduler.test.ts b/packages/core/src/core/coreToolScheduler.test.ts
index 8c0b0edbe..f7be11c5a 100644
--- a/packages/core/src/core/coreToolScheduler.test.ts
+++ b/packages/core/src/core/coreToolScheduler.test.ts
@@ -3919,16 +3919,22 @@ describe('CoreToolScheduler validation retry loop detection', () => {
 
   function createSchedulerWithTool(tool: StrictStringTool) {
     const mockToolRegistry = {
-      getTool: () => tool,
+      ensureTool: async (name: string) =>
+        name === StrictStringTool.Name ? tool : undefined,
+      getTool: (name: string) =>
+        name === StrictStringTool.Name ? tool : undefined,
       getFunctionDeclarations: () => [],
       tools: new Map(),
       discovery: {},
       registerTool: () => {},
-      getToolByName: () => tool,
-      getToolByDisplayName: () => tool,
+      getToolByName: (name: string) =>
+        name === StrictStringTool.Name ? tool : undefined,
+      getToolByDisplayName: (name: string) =>
+        name === 'StrictStringTool' ? tool : undefined,
       getTools: () => [],
       discoverTools: async () => {},
       getAllTools: () => [],
+      getAllToolNames: () => [StrictStringTool.Name],
       getToolsByServer: () => [],
     } as unknown as ToolRegistry;