82 lines
3.7 KiB
Python
82 lines
3.7 KiB
Python
"""Tests for hermes tools disable/enable/list command (backend)."""
|
|
from argparse import Namespace
|
|
from unittest.mock import patch
|
|
|
|
from hermes_cli.tools_config import tools_disable_enable_command
|
|
|
|
|
|
# ── Built-in toolset disable ────────────────────────────────────────────────
|
|
|
|
|
|
class TestToolsDisableBuiltin:
|
|
|
|
def test_disable_removes_toolset_from_platform(self):
|
|
config = {"platform_toolsets": {"cli": ["web", "memory", "terminal"]}}
|
|
with patch("hermes_cli.tools_config.load_config", return_value=config), \
|
|
patch("hermes_cli.tools_config.save_config") as mock_save:
|
|
tools_disable_enable_command(Namespace(tools_action="disable", names=["web"], platform="cli"))
|
|
saved = mock_save.call_args[0][0]
|
|
assert "web" not in saved["platform_toolsets"]["cli"]
|
|
assert "memory" in saved["platform_toolsets"]["cli"]
|
|
|
|
|
|
# ── Built-in toolset enable ─────────────────────────────────────────────────
|
|
|
|
|
|
# ── MCP tool disable ────────────────────────────────────────────────────────
|
|
|
|
|
|
class TestToolsDisableMcp:
|
|
|
|
|
|
def test_disable_unknown_server_prints_error(self, capsys):
|
|
config = {"mcp_servers": {}}
|
|
with patch("hermes_cli.tools_config.load_config", return_value=config), \
|
|
patch("hermes_cli.tools_config.save_config"):
|
|
tools_disable_enable_command(
|
|
Namespace(tools_action="disable", names=["unknown:tool"], platform="cli")
|
|
)
|
|
out = capsys.readouterr().out
|
|
assert "MCP server 'unknown' not found in config" in out
|
|
|
|
|
|
# ── MCP tool enable ──────────────────────────────────────────────────────────
|
|
|
|
|
|
# ── Mixed targets ────────────────────────────────────────────────────────────
|
|
|
|
|
|
# ── List output ──────────────────────────────────────────────────────────────
|
|
|
|
|
|
class TestToolsList:
|
|
|
|
|
|
def test_list_shows_mcp_excluded_tools(self, capsys):
|
|
config = {
|
|
"mcp_servers": {"github": {"tools": {"exclude": ["create_issue"]}}},
|
|
}
|
|
with patch("hermes_cli.tools_config.load_config", return_value=config):
|
|
tools_disable_enable_command(Namespace(tools_action="list", platform="cli"))
|
|
out = capsys.readouterr().out
|
|
assert "github" in out
|
|
assert "create_issue" in out
|
|
|
|
|
|
# ── Validation ───────────────────────────────────────────────────────────────
|
|
|
|
|
|
class TestToolsValidation:
|
|
|
|
|
|
def test_mixed_valid_and_invalid_applies_valid_only(self):
|
|
config = {"platform_toolsets": {"cli": ["web", "memory"]}}
|
|
with patch("hermes_cli.tools_config.load_config", return_value=config), \
|
|
patch("hermes_cli.tools_config.save_config") as mock_save:
|
|
tools_disable_enable_command(
|
|
Namespace(tools_action="disable", names=["web", "bad_toolset"], platform="cli")
|
|
)
|
|
saved = mock_save.call_args[0][0]
|
|
assert "web" not in saved["platform_toolsets"]["cli"]
|
|
assert "memory" in saved["platform_toolsets"]["cli"]
|