Add tests

Author: tysonthomas9

config/gni/devtools_grd_files.gni

@@ -661,6 +661,11 @@ grd_files_bundled_sources = [
   "front_end/panels/ai_chat/ui/TodoListDisplay.js",
   "front_end/panels/ai_chat/ui/FileListDisplay.js",
   "front_end/panels/ai_chat/ui/FileContentViewer.js",
+  "front_end/panels/ai_chat/ui/AgentStudioBridge.js",
+  "front_end/panels/ai_chat/ui/AgentStudioController.js",
+  "front_end/panels/ai_chat/ui/AgentStudioView.js",
+  "front_end/panels/ai_chat/ui/SchemaEditor.js",
+  "front_end/panels/ai_chat/ui/agent_studio/AgentStudioSPA.js",
   "front_end/panels/ai_chat/ui/settings/advanced/BrowsingHistorySettings.js",
   "front_end/panels/ai_chat/ui/settings/advanced/EvaluationSettings.js",
   "front_end/panels/ai_chat/ui/settings/advanced/MCPSettings.js",
@@ -703,6 +708,9 @@ grd_files_bundled_sources = [
   "front_end/panels/ai_chat/core/VersionChecker.js",
   "front_end/panels/ai_chat/core/LLMConfigurationManager.js",
   "front_end/panels/ai_chat/core/CustomProviderManager.js",
+  "front_end/panels/ai_chat/core/AgentStorageManager.js",
+  "front_end/panels/ai_chat/core/AgentStudioIntegration.js",
+  "front_end/panels/ai_chat/core/AgentTestRunner.js",
   "front_end/panels/ai_chat/LLM/LLMTypes.js",
   "front_end/panels/ai_chat/LLM/LLMProvider.js",
   "front_end/panels/ai_chat/LLM/LLMProviderRegistry.js",
@@ -748,10 +756,25 @@ grd_files_bundled_sources = [
   "front_end/panels/ai_chat/tools/UpdateTodoTool.js",
   "front_end/panels/ai_chat/tools/VisualIndicatorTool.js",
   "front_end/panels/ai_chat/tools/ReadabilityExtractorTool.js",
+  "front_end/panels/ai_chat/tools/CallCustomAgentTool.js",
+  "front_end/panels/ai_chat/tools/SearchCustomAgentsTool.js",
+  "front_end/panels/ai_chat/tools/mini_app/CloseMiniAppTool.js",
+  "front_end/panels/ai_chat/tools/mini_app/ExecuteMiniAppActionTool.js",
+  "front_end/panels/ai_chat/tools/mini_app/GetMiniAppStateTool.js",
+  "front_end/panels/ai_chat/tools/mini_app/LaunchMiniAppTool.js",
+  "front_end/panels/ai_chat/tools/mini_app/ListMiniAppsTool.js",
+  "front_end/panels/ai_chat/tools/mini_app/UpdateMiniAppStateTool.js",
   "front_end/panels/ai_chat/common/utils.js",
   "front_end/panels/ai_chat/common/log.js",
   "front_end/panels/ai_chat/common/context.js",
   "front_end/panels/ai_chat/common/page.js",
+  "front_end/panels/ai_chat/mini_apps/GenericMiniAppBridge.js",
+  "front_end/panels/ai_chat/mini_apps/MiniAppEventBus.js",
+  "front_end/panels/ai_chat/mini_apps/MiniAppInitialization.js",
+  "front_end/panels/ai_chat/mini_apps/MiniAppRegistry.js",
+  "front_end/panels/ai_chat/mini_apps/MiniAppStorageManager.js",
+  "front_end/panels/ai_chat/mini_apps/apps/agent_studio/AgentStudioMiniApp.js",
+  "front_end/panels/ai_chat/mini_apps/types/MiniAppTypes.js",
     "front_end/panels/ai_chat/models/ChatTypes.js",
   "front_end/panels/ai_chat/ui/input/ChatInput.js",
   "front_end/panels/ai_chat/ui/input/InputBar.js",

front_end/panels/ai_chat/BUILD.gn

@@ -537,10 +537,15 @@ ts_library("unittests") {
     "ui/__tests__/WebAppCodeViewer.test.ts",
     "ui/input/__tests__/InputBarClear.test.ts",
     "ui/message/__tests__/MessageCombiner.test.ts",
-        "LLM/__tests__/MessageSanitizer.test.ts",
+    "LLM/__tests__/MessageSanitizer.test.ts",
     "agent_framework/__tests__/AgentRunner.sanitizeToolResult.test.ts",
     "agent_framework/__tests__/AgentRunner.computeToolResultText.test.ts",
     "agent_framework/__tests__/AgentRunner.run.flows.test.ts",
+    "agent_framework/__tests__/AgentRunner.core.test.ts",
+    "agent_framework/__tests__/AgentRunner.handoff.test.ts",
+    "agent_framework/__tests__/ConfigurableAgentTool.test.ts",
+    "agent_framework/__tests__/ToolRegistry.test.ts",
+    "agent_framework/__tests__/AgentRunnerEventBus.test.ts",
     "mcp/__tests__/MCPClientSDK.test.ts",
     "mcp/__tests__/MCPConfig.test.ts",
     "mcp/__tests__/MCPIntegration.test.ts",
@@ -551,6 +556,7 @@ ts_library("unittests") {
     "core/__tests__/ToolNameMap.test.ts",
     "core/__tests__/ToolNameMapping.test.ts",
     "core/__tests__/ToolSurfaceProvider.test.ts",
+    "core/__tests__/StateGraph.core.test.ts",
     "ui/__tests__/AIChatPanel.test.ts",
     "ui/__tests__/LiveAgentSessionComponent.test.ts",
     "ui/message/__tests__/MessageList.test.ts",
@@ -559,10 +565,15 @@ ts_library("unittests") {
     "tools/__tests__/ReadFileTool.test.ts",
     "tools/__tests__/ListFilesTool.test.ts",
     "tools/__tests__/FileStorageManager.test.ts",
+    "tools/mini_app/__tests__/MiniAppTools.test.ts",
+    "mini_apps/__tests__/MiniAppRegistry.test.ts",
+    "mini_apps/__tests__/GenericMiniAppBridge.test.ts",
+    "mini_apps/__tests__/MiniAppEventBus.test.ts",
   ]
 
   deps = [
     ":ai_chat",
+    "testing:testing",
     "../../testing",
     "../../core/sdk:bundle",
     "../../generated:protocol",

front_end/panels/ai_chat/agent_framework/AgentRunner.ts

@@ -402,19 +402,21 @@ export class AgentRunner {
           ...(actualResult.intermediateSteps || []) // History *from* the recursive call (should exist if flag is true)
       ];
       // Return the result from the target agent, but with combined history
+      // Always use 'handed_off' since this IS a handoff completion
       return {
           ...actualResult,
           intermediateSteps: combinedIntermediateSteps,
-          terminationReason: actualResult.terminationReason || 'handed_off',
+          terminationReason: 'handed_off' as const,
           agentSession: childSession
       };
     }
     // Otherwise (default), omit the target's intermediate steps
     logger.info(`Omitting intermediateSteps from ${targetAgentTool.name} based on its config (default or flag set to false).`);
     // Return result from target, ensuring intermediateSteps are omitted
+    // Always use 'handed_off' since this IS a handoff completion
     const finalResult = {
       ...actualResult,
-      terminationReason: actualResult.terminationReason || 'handed_off',
+      terminationReason: 'handed_off' as const,
       agentSession: childSession
     };
     // Explicitly delete intermediateSteps if they somehow exist on actualResult (shouldn't due to target config)