mirror of
https://github.com/tiennm99/serena.git
synced 2026-10-14 06:17:58 +00:00
1 parent
f0693a10c1
commit
c4dc91a7da
7 files changed
+91
-68
No files matched your search
+2
-22
@@ -1,13 +1,11 @@
|
||||
"""Tests for the mcp.py module in serena."""
|
||||
|
||||
import pytest
|
||||
from mcp.server.fastmcp import Context
|
||||
from mcp.server.fastmcp.tools.base import Tool as MCPTool
|
||||
from mcp.server.mcpserver import Context
|
||||
from mcp.server.mcpserver.tools.base import Tool as MCPTool
|
||||
|
||||
from serena import __version__
|
||||
from serena.agent import Tool, ToolRegistry
|
||||
from serena.config.context_mode import SerenaAgentContext
|
||||
from serena.config.serena_config import SerenaConfig
|
||||
from serena.mcp import SerenaMCPFactory
|
||||
from serena.repl.facade import ApiScope
|
||||
from serena.repl.repl import SerenaRepl
|
||||
@@ -64,24 +62,6 @@ class BasicTool(BaseMockTool):
|
||||
return self.apply(**kwargs)
|
||||
|
||||
|
||||
def test_create_mcp_server_reports_serena_version(monkeypatch: pytest.MonkeyPatch) -> None:
|
||||
"""MCP initialize must report Serena's version, not the installed mcp SDK version."""
|
||||
|
||||
class MinimalAgent:
|
||||
def create_connection_prompt(self) -> str:
|
||||
return ""
|
||||
|
||||
monkeypatch.setattr(SerenaConfig, "from_config_file", classmethod(lambda cls: SerenaConfig()))
|
||||
factory = SerenaMCPFactory(transport="stdio")
|
||||
monkeypatch.setattr(factory, "_create_serena_agent", lambda *args, **kwargs: MinimalAgent())
|
||||
|
||||
mcp_server = factory.create_mcp_server()
|
||||
initialization_options = mcp_server._mcp_server.create_initialization_options()
|
||||
|
||||
assert initialization_options.server_name == "Serena"
|
||||
assert initialization_options.server_version == __version__
|
||||
|
||||
|
||||
def test_make_tool_basic() -> None:
|
||||
"""Test that make_tool correctly creates an MCP tool from a Tool object."""
|
||||
mock_tool = BasicTool()
|
||||
|
||||
Reference in new issue
Block a user