oclaw/tests/test_mcp_e2e_probe.py
oliver 420abac9f1 refactor: root-package imports (svc/runtime/interfaces) and fix PYTHONPATH
- Rename platform/ to svc/ to avoid shadowing stdlib platform.
- Replace from oclaw.* with from svc/runtime/interfaces; update -m CLI paths.
- tests/conftest: prepend repo root to sys.path (no parent-folder package name).
- CI: paths and offline_eval script under repo root.
- Ops scripts: PYTHONPATH must be repo root for python -m runtime.* (fixes gateway/WhatsApp sidecar startup).
- Fix default oclaw.json path in tabular/file attachment limits; stabilize attachment test config.

Co-authored-by: Cursor <cursoragent@cursor.com>
2026-05-13 14:51:17 +08:00

63 lines
2.3 KiB
Python

from __future__ import annotations
import unittest
from interfaces.admin.mcp_e2e_probe import build_mcp_e2e_probe_plans, parse_mcp_bound_tool_name
from runtime.tools.base import ToolRegistry, ToolSpec
class McpE2eProbeTests(unittest.TestCase):
def test_parse_mcp_bound_tool_name(self) -> None:
self.assertEqual(parse_mcp_bound_tool_name("mcp__srv__ping"), ("srv", "ping"))
self.assertIsNone(parse_mcp_bound_tool_name("read_file"))
self.assertIsNone(parse_mcp_bound_tool_name("mcp__only"))
def test_build_plans_empty_registry(self) -> None:
reg = ToolRegistry([])
plans, skipped = build_mcp_e2e_probe_plans(reg, workspace_root="/tmp")
self.assertEqual(plans, [])
self.assertEqual(skipped, [])
def test_build_one_probe_per_server(self) -> None:
def _h(_a):
return {"ok": True}
reg = ToolRegistry(
[
ToolSpec(
name="mcp__alpha__zebra",
description="z",
parameters={"type": "object", "properties": {}, "required": []},
handler=_h,
tags=frozenset({"mcp"}),
),
ToolSpec(
name="mcp__alpha__apple",
description="a",
parameters={"type": "object", "properties": {}, "required": []},
handler=_h,
tags=frozenset({"mcp"}),
),
ToolSpec(
name="mcp__beta__read_graph",
description="g",
parameters={"type": "object", "properties": {}, "required": []},
handler=_h,
tags=frozenset({"mcp"}),
),
]
)
plans, skipped = build_mcp_e2e_probe_plans(reg, workspace_root="/tmp")
self.assertEqual(skipped, [])
self.assertEqual(len(plans), 2)
by_sid = {p[0]: p for p in plans}
self.assertIn("alpha", by_sid)
self.assertIn("beta", by_sid)
# read_graph wins priority over apple/zebra for beta
self.assertEqual(by_sid["beta"][1], "mcp__beta__read_graph")
# alpha: apple before zebra lexicographically among equal priority
self.assertEqual(by_sid["alpha"][1], "mcp__alpha__apple")
if __name__ == "__main__":
unittest.main()