Something went wrong. Try again.
The agent that grows with you
Something went wrong. Try again.
3.2 kB · 92 lines
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293import importlibimport osimport sysfrom unittest.mock import MagicMock, patch
sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
_cli_mod = None
def _make_cli(user_message_preview=None): global _cli_mod clean_config = { "model": { "default": "anthropic/claude-opus-4.6", "base_url": "https://openrouter.ai/api/v1", "provider": "auto", }, "display": { "compact": False, "tool_progress": "all", "user_message_preview": user_message_preview or {"first_lines": 2, "last_lines": 2}, }, "agent": {}, "terminal": {"env_type": "local"}, } clean_env = {"LLM_MODEL": "", "HERMES_MAX_ITERATIONS": ""} prompt_toolkit_stubs = { "prompt_toolkit": MagicMock(), "prompt_toolkit.history": MagicMock(), "prompt_toolkit.styles": MagicMock(), "prompt_toolkit.patch_stdout": MagicMock(), "prompt_toolkit.application": MagicMock(), "prompt_toolkit.layout": MagicMock(), "prompt_toolkit.layout.processors": MagicMock(), "prompt_toolkit.filters": MagicMock(), "prompt_toolkit.layout.dimension": MagicMock(), "prompt_toolkit.layout.menus": MagicMock(), "prompt_toolkit.widgets": MagicMock(), "prompt_toolkit.key_binding": MagicMock(), "prompt_toolkit.completion": MagicMock(), "prompt_toolkit.formatted_text": MagicMock(), "prompt_toolkit.auto_suggest": MagicMock(), } with patch.dict(sys.modules, prompt_toolkit_stubs), patch.dict("os.environ", clean_env, clear=False): import cli as mod
mod = importlib.reload(mod) _cli_mod = mod with patch.object(mod, "get_tool_definitions", return_value=[]), patch.dict(mod.__dict__, {"CLI_CONFIG": clean_config}): return mod.HermesCLI()
class TestSubmittedUserMessagePreview: def test_default_preview_shows_first_two_lines_and_last_two_lines(self): cli = _make_cli()
rendered = cli._format_submitted_user_message_preview( "line1\nline2\nline3\nline4\nline5\nline6" )
assert "line1" in rendered assert "line2" in rendered assert "line5" in rendered assert "line6" in rendered assert "line3" not in rendered assert "line4" not in rendered assert "(+2 more lines)" in rendered
def test_preview_can_hide_last_lines(self): cli = _make_cli({"first_lines": 2, "last_lines": 0})
rendered = cli._format_submitted_user_message_preview( "line1\nline2\nline3\nline4\nline5\nline6" )
assert "line1" in rendered assert "line2" in rendered assert "line5" not in rendered assert "line6" not in rendered assert "(+4 more lines)" in rendered
def test_invalid_first_lines_value_falls_back_to_one(self): cli = _make_cli({"first_lines": 0, "last_lines": 2})
rendered = cli._format_submitted_user_message_preview("line1\nline2\nline3\nline4")
assert "line1" in rendered assert "line3" in rendered assert "line4" in rendered assert "(+1 more line)" in rendered