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.
This commit is contained in:
Mohamed Boudra
2026-01-30 12:29:01 +07:00
parent f8da5ada71
commit 5b32d0a56c
2 changed files with 8 additions and 12 deletions

View File

@@ -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,
});

View File

@@ -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",
});