Enhance tool docstrings for LLM discoverability
- Add cross-references between related tools - Explain use cases and when to use alternatives - All 25 MCP tools now have descriptive docstrings
This commit is contained in:
parent
5595f1aef1
commit
36a181d131
File diff suppressed because it is too large
Load Diff
@ -2,7 +2,7 @@
|
||||
|
||||
import pytest
|
||||
|
||||
from mctelnet.server import _connections, mcp
|
||||
from mctelnet.server import _connections, _servers, _strip_ansi, mcp
|
||||
|
||||
|
||||
def test_mcp_server_exists():
|
||||
@ -10,15 +10,16 @@ def test_mcp_server_exists():
|
||||
assert mcp.name == "mctelnet"
|
||||
|
||||
|
||||
def test_tools_registered():
|
||||
"""Verify all expected tools are registered."""
|
||||
# Access tools through the tool manager's internal storage
|
||||
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",
|
||||
@ -26,6 +27,29 @@ def test_tools_registered():
|
||||
"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}"
|
||||
@ -33,6 +57,50 @@ def test_tools_registered():
|
||||
|
||||
def test_connection_storage_initially_empty():
|
||||
"""Verify no connections exist at startup."""
|
||||
# Clear any lingering connections
|
||||
_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"
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user