m4
c2743251e9
Self-evolving AI scientist framework built on LangGraph/LangChain with CLI/TUI core, FastAPI gateway, and Next.js frontend. Co-Authored-By: Claude Opus 4 <noreply@anthropic.com>
96 lines
3.2 KiB
Python
96 lines
3.2 KiB
Python
"""Tests for MCP tool caching in EvoScientist.EvoScientist."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import EvoScientist.EvoScientist as agent_module
|
|
|
|
|
|
def _reset_mcp_cache() -> None:
|
|
agent_module._MCP_TOOLS_CACHE_KEY = None
|
|
agent_module._MCP_TOOLS_CACHE_VALUE = None
|
|
|
|
|
|
class TestMcpToolCaching:
|
|
def setup_method(self) -> None:
|
|
_reset_mcp_cache()
|
|
|
|
def test_reuses_cached_tools_when_config_unchanged(self, monkeypatch):
|
|
calls = {"load": 0}
|
|
tool = object()
|
|
|
|
monkeypatch.setattr(
|
|
"EvoScientist.mcp.client.load_mcp_config",
|
|
lambda: {"srv": {"transport": "stdio", "command": "demo"}},
|
|
)
|
|
|
|
def fake_load_mcp_tools(config=None):
|
|
calls["load"] += 1
|
|
return {"main": [tool]}
|
|
|
|
monkeypatch.setattr("EvoScientist.mcp.load_mcp_tools", fake_load_mcp_tools)
|
|
|
|
first = agent_module._load_mcp_tools_cached()
|
|
second = agent_module._load_mcp_tools_cached()
|
|
|
|
assert calls["load"] == 1
|
|
assert first == second
|
|
assert first is not second
|
|
assert first["main"] is not second["main"]
|
|
|
|
def test_reload_when_config_changes(self, monkeypatch):
|
|
calls = {"load": 0}
|
|
state = {"cfg": {"srv": {"transport": "stdio", "command": "v1"}}}
|
|
|
|
def fake_load_config():
|
|
return state["cfg"]
|
|
|
|
def fake_load_mcp_tools(config=None):
|
|
calls["load"] += 1
|
|
return {"main": [f"tool-v{calls['load']}"]}
|
|
|
|
monkeypatch.setattr("EvoScientist.mcp.client.load_mcp_config", fake_load_config)
|
|
monkeypatch.setattr("EvoScientist.mcp.load_mcp_tools", fake_load_mcp_tools)
|
|
|
|
first = agent_module._load_mcp_tools_cached()
|
|
state["cfg"] = {"srv": {"transport": "stdio", "command": "v2"}}
|
|
second = agent_module._load_mcp_tools_cached()
|
|
|
|
assert calls["load"] == 2
|
|
assert first != second
|
|
|
|
def test_load_mcp_config_called_once_per_cache_miss(self, monkeypatch):
|
|
"""load_mcp_config should be called exactly once per cache miss,
|
|
not twice (once for the signature and once inside load_mcp_tools)."""
|
|
calls = {"config": 0}
|
|
|
|
def counting_load_config():
|
|
calls["config"] += 1
|
|
return {"srv": {"transport": "stdio", "command": "demo"}}
|
|
|
|
monkeypatch.setattr(
|
|
"EvoScientist.mcp.client.load_mcp_config", counting_load_config
|
|
)
|
|
monkeypatch.setattr(
|
|
"EvoScientist.mcp.load_mcp_tools", lambda config=None: {"main": []}
|
|
)
|
|
|
|
agent_module._load_mcp_tools_cached()
|
|
assert calls["config"] == 1
|
|
|
|
def test_cached_config_passed_to_load_mcp_tools(self, monkeypatch):
|
|
"""load_mcp_tools should receive the pre-loaded config dict."""
|
|
received = {}
|
|
|
|
def fake_load_config():
|
|
return {"srv": {"transport": "stdio", "command": "demo"}}
|
|
|
|
def fake_load_mcp_tools(config=None):
|
|
received["config"] = config
|
|
return {"main": []}
|
|
|
|
monkeypatch.setattr("EvoScientist.mcp.client.load_mcp_config", fake_load_config)
|
|
monkeypatch.setattr("EvoScientist.mcp.load_mcp_tools", fake_load_mcp_tools)
|
|
|
|
agent_module._load_mcp_tools_cached()
|
|
assert received["config"] == {"srv": {"transport": "stdio", "command": "demo"}}
|