From 5b32d0a56cbc5dd481b9a6e2a043d9f054b7cba9 Mon Sep 17 00:00:00 2001
From: Mohamed Boudra <boudra.moha@gmail.com>
Date: Fri, 30 Jan 2026 12:29:01 +0700
Subject: [PATCH] fix: update MCP tests to use createAgentSelfIdMcpServer

Tests for set_title and set_branch were looking for these tools in
mcp-server.ts but they were moved to agent-self-id-mcp.ts during
the MCP split.
---
 .../agent/agent-self-identification.e2e.test.ts    | 14 +++++---------
 .../server/src/server/agent/mcp-server.test.ts     |  6 +++---
 2 files changed, 8 insertions(+), 12 deletions(-)

diff --git a/packages/server/src/server/agent/agent-self-identification.e2e.test.ts b/packages/server/src/server/agent/agent-self-identification.e2e.test.ts
index 57c1e87ba..a34f08f0d 100644
--- a/packages/server/src/server/agent/agent-self-identification.e2e.test.ts
+++ b/packages/server/src/server/agent/agent-self-identification.e2e.test.ts
@@ -8,7 +8,7 @@ import { randomUUID } from "node:crypto";
 import { createTestLogger } from "../../test-utils/test-logger.js";
 import { AgentManager } from "./agent-manager.js";
 import { AgentStorage } from "./agent-storage.js";
-import { createAgentMcpServer } from "./mcp-server.js";
+import { createAgentSelfIdMcpServer } from "./agent-self-id-mcp.js";
 import { createWorktree } from "../../utils/worktree.js";
 import type {
   AgentClient,
@@ -148,9 +148,8 @@ describe("self-identification MCP tools", () => {
         cwd: worktree.worktreePath,
       });
 
-      const server = await createAgentMcpServer({
+      const server = await createAgentSelfIdMcpServer({
         agentManager: manager,
-        agentStorage: storage,
         paseoHome,
         callerAgentId: agent.id,
         logger,
@@ -202,9 +201,8 @@ describe("self-identification MCP tools", () => {
         cwd: nestedDir,
       });
 
-      const server = await createAgentMcpServer({
+      const server = await createAgentSelfIdMcpServer({
         agentManager: manager,
-        agentStorage: storage,
         paseoHome,
         callerAgentId: agent.id,
         logger,
@@ -245,9 +243,8 @@ describe("self-identification MCP tools", () => {
         cwd: repoDir,
       });
 
-      const server = await createAgentMcpServer({
+      const server = await createAgentSelfIdMcpServer({
         agentManager: manager,
-        agentStorage: storage,
         callerAgentId: agent.id,
         logger,
       });
@@ -279,9 +276,8 @@ describe("self-identification MCP tools", () => {
         cwd: repoDir,
       });
 
-      const server = await createAgentMcpServer({
+      const server = await createAgentSelfIdMcpServer({
         agentManager: manager,
-        agentStorage: storage,
         callerAgentId: agent.id,
         logger,
       });
diff --git a/packages/server/src/server/agent/mcp-server.test.ts b/packages/server/src/server/agent/mcp-server.test.ts
index 24bb0a765..ef9e9fa17 100644
--- a/packages/server/src/server/agent/mcp-server.test.ts
+++ b/packages/server/src/server/agent/mcp-server.test.ts
@@ -2,6 +2,7 @@ import { describe, expect, it, vi } from "vitest";
 
 import { createTestLogger } from "../../test-utils/test-logger.js";
 import { createAgentMcpServer } from "./mcp-server.js";
+import { createAgentSelfIdMcpServer } from "./agent-self-id-mcp.js";
 import type { AgentManager, ManagedAgent } from "./agent-manager.js";
 import type { AgentStorage } from "./agent-storage.js";
 
@@ -104,14 +105,13 @@ describe("create_agent MCP tool", () => {
   });
 
   it("set_title trims and persists titles for caller agent", async () => {
-    const { agentManager, agentStorage, spies } = createTestDeps();
+    const { agentManager, spies } = createTestDeps();
     spies.agentManager.getAgent.mockReturnValue({
       id: "agent-1",
     } as ManagedAgent);
 
-    const server = await createAgentMcpServer({
+    const server = await createAgentSelfIdMcpServer({
       agentManager,
-      agentStorage,
       logger,
       callerAgentId: "agent-1",
     });