mirror of
https://github.com/angr/angr
synced 2026-08-17 12:23:11 -04:00
189d3057e2
* Rewrite tests/mcprtest_cli.py to use unittest instead of pytest * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
import unittest
|
|
from unittest import mock
|
|
|
|
from angr.mcp import __main__ as mcp_main
|
|
|
|
|
|
class DummyMCP:
|
|
"""Test double for the MCP server object."""
|
|
|
|
def __init__(self) -> None:
|
|
self.calls: list[dict[str, object]] = []
|
|
|
|
def run(self, **kwargs: object) -> None:
|
|
self.calls.append(kwargs)
|
|
|
|
|
|
class TestMCPCLI(unittest.TestCase):
|
|
def test_main_dispatches_transport(self):
|
|
cases = [
|
|
([], {"transport": "stdio"}),
|
|
(
|
|
["--transport", "sse", "--host", "0.0.0.0", "--port", "8123"],
|
|
{"transport": "sse", "host": "0.0.0.0", "port": 8123},
|
|
),
|
|
(
|
|
["--transport", "http", "--host", "0.0.0.0", "--port", "8123", "--path", "/custom"],
|
|
{"transport": "http", "host": "0.0.0.0", "port": 8123, "path": "/custom"},
|
|
),
|
|
]
|
|
for argv, expected in cases:
|
|
with self.subTest(argv=argv):
|
|
dummy_mcp = DummyMCP()
|
|
with mock.patch.object(mcp_main, "mcp", dummy_mcp):
|
|
mcp_main.main(argv)
|
|
assert dummy_mcp.calls == [expected]
|
|
|
|
def test_http_path_must_be_absolute(self):
|
|
dummy_mcp = DummyMCP()
|
|
with mock.patch.object(mcp_main, "mcp", dummy_mcp), self.assertRaises(SystemExit) as exc_info:
|
|
mcp_main.main(["--transport", "http", "--path", "mcp"])
|
|
|
|
assert exc_info.exception.code == 2
|
|
assert not dummy_mcp.calls
|