oclaw/tests/test_catalog_mcp_integration.py
oliver 4a23b715a2 重构仓库目录为统一的 runtime 分层并清理历史 openclaw 残留。
本次迁移将网关/通道/工具/技能/脚本与协议资源集中到新结构,统一路径常量与脚本转发机制,减少顶层噪音并保证运行与测试行为一致。

Made-with: Cursor
2026-04-25 01:24:23 +08:00

42 lines
1.4 KiB
Python

from __future__ import annotations
from pathlib import Path
from oclaw.platform.persistence.sqlite_store import SqliteStore
from oclaw.runtime.tools.catalog import materialize_tool_specs
def _seed_single_mcp_tool(store: SqliteStore) -> None:
store.upsert_mcp_server(
server_id="echo",
source_type="github",
source_ref="https://github.com/acme/echo",
entry_command="python",
entry_args=["-m", "echo_server"],
enabled=True,
)
store.replace_mcp_server_tools(
server_id="echo",
tools=[{"tool_name": "ping", "description": "Ping", "parameters": {"type": "object", "properties": {}}}],
)
def test_catalog_materializes_mcp_tools_when_enabled(tmp_path: Path) -> None:
store = SqliteStore(str(tmp_path / "ops.sqlite"))
_seed_single_mcp_tool(store)
store.set_setting("AIA_ENABLE_MCP_TOOLS", "1")
specs = materialize_tool_specs(store=store, specialist="generalist")
names = {x.name for x in specs}
assert "system_time" in names
assert "mcp__echo__ping" in names
def test_catalog_skips_mcp_tools_when_disabled(tmp_path: Path) -> None:
store = SqliteStore(str(tmp_path / "ops.sqlite"))
_seed_single_mcp_tool(store)
store.set_setting("AIA_ENABLE_MCP_TOOLS", "0")
specs = materialize_tool_specs(store=store, specialist="generalist")
names = {x.name for x in specs}
assert "system_time" in names
assert "mcp__echo__ping" not in names