Ryan Malloy
36a181d131
- Add cross-references between related tools - Explain use cases and when to use alternatives - All 25 MCP tools now have descriptive docstrings
107 lines
3.0 KiB
Python
107 lines
3.0 KiB
Python
"""Tests for mctelnet server."""
|
|
|
|
import pytest
|
|
|
|
from mctelnet.server import _connections, _servers, _strip_ansi, mcp
|
|
|
|
|
|
def test_mcp_server_exists():
|
|
"""Verify MCP server is configured."""
|
|
assert mcp.name == "mctelnet"
|
|
|
|
|
|
def test_client_tools_registered():
|
|
"""Verify all expected client tools are registered."""
|
|
tool_manager = mcp._tool_manager
|
|
tool_names = set(tool_manager._tools.keys())
|
|
|
|
expected_tools = {
|
|
"connect",
|
|
"send",
|
|
"send_key",
|
|
"list_keys",
|
|
"read",
|
|
"expect",
|
|
"expect_send",
|
|
"run_script",
|
|
"list_connections",
|
|
"disconnect",
|
|
"disconnect_all",
|
|
"connection_info",
|
|
"get_transcript",
|
|
}
|
|
|
|
assert expected_tools.issubset(tool_names), f"Missing tools: {expected_tools - tool_names}"
|
|
|
|
|
|
def test_server_tools_registered():
|
|
"""Verify all expected server tools are registered."""
|
|
tool_manager = mcp._tool_manager
|
|
tool_names = set(tool_manager._tools.keys())
|
|
|
|
expected_tools = {
|
|
"start_server",
|
|
"stop_server",
|
|
"list_servers",
|
|
"server_info",
|
|
"list_clients",
|
|
"read_client",
|
|
"send_client",
|
|
"broadcast",
|
|
"disconnect_client",
|
|
"get_server_transcript",
|
|
}
|
|
|
|
assert expected_tools.issubset(tool_names), f"Missing tools: {expected_tools - tool_names}"
|
|
|
|
|
|
def test_connection_storage_initially_empty():
|
|
"""Verify no connections exist at startup."""
|
|
_connections.clear()
|
|
assert len(_connections) == 0
|
|
|
|
|
|
def test_server_storage_initially_empty():
|
|
"""Verify no servers exist at startup."""
|
|
_servers.clear()
|
|
assert len(_servers) == 0
|
|
|
|
|
|
class TestAnsiStripping:
|
|
"""Tests for ANSI escape sequence stripping."""
|
|
|
|
def test_strip_simple_color(self):
|
|
"""Strip basic color codes."""
|
|
text = "\x1b[32mgreen\x1b[0m text"
|
|
assert _strip_ansi(text) == "green text"
|
|
|
|
def test_strip_complex_sgr(self):
|
|
"""Strip multi-parameter SGR sequences."""
|
|
text = "\x1b[1;31;40mBold red on black\x1b[0m"
|
|
assert _strip_ansi(text) == "Bold red on black"
|
|
|
|
def test_strip_cursor_movement(self):
|
|
"""Strip cursor movement sequences."""
|
|
text = "\x1b[2J\x1b[H\x1b[5;10HHello"
|
|
assert _strip_ansi(text) == "Hello"
|
|
|
|
def test_strip_osc_title(self):
|
|
"""Strip OSC window title sequences (BEL terminated)."""
|
|
text = "\x1b]0;Window Title\x07Some text"
|
|
assert _strip_ansi(text) == "Some text"
|
|
|
|
def test_strip_osc_st_terminated(self):
|
|
"""Strip OSC sequences terminated with ST."""
|
|
text = "\x1b]0;Title\x1b\\Some text"
|
|
assert _strip_ansi(text) == "Some text"
|
|
|
|
def test_preserve_plain_text(self):
|
|
"""Plain text passes through unchanged."""
|
|
text = "Hello, World! 123"
|
|
assert _strip_ansi(text) == text
|
|
|
|
def test_mixed_content(self):
|
|
"""Handle mixed plain text and ANSI codes."""
|
|
text = "Start \x1b[1mBold\x1b[0m middle \x1b[4mUnderline\x1b[0m end"
|
|
assert _strip_ansi(text) == "Start Bold middle Underline end"
|