Something went wrong. Try again.
The agent that grows with you
Something went wrong. Try again.
2.2 kB · 71 lines
Python
at bb/theme
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172"""Tests for CLI /copy command."""
from unittest.mock import MagicMock, patch
from cli import HermesCLI
def _make_cli() -> HermesCLI: cli_obj = HermesCLI.__new__(HermesCLI) cli_obj.config = {} cli_obj.console = MagicMock() cli_obj.agent = None cli_obj.conversation_history = [] cli_obj.session_id = "sess-copy-test" cli_obj._pending_input = MagicMock() cli_obj._app = None return cli_obj
def test_copy_copies_latest_assistant_message(): cli_obj = _make_cli() cli_obj.conversation_history = [ {"role": "user", "content": "hi"}, {"role": "assistant", "content": "first"}, {"role": "assistant", "content": "latest"}, ]
with patch.object(cli_obj, "_write_osc52_clipboard") as mock_copy: result = cli_obj.process_command("/copy")
assert result is True mock_copy.assert_called_once_with("latest")
def test_copy_with_index_uses_requested_assistant_message(): cli_obj = _make_cli() cli_obj.conversation_history = [ {"role": "assistant", "content": "one"}, {"role": "assistant", "content": "two"}, ]
with patch.object(cli_obj, "_write_osc52_clipboard") as mock_copy: cli_obj.process_command("/copy 1")
mock_copy.assert_called_once_with("one")
def test_copy_strips_reasoning_blocks_before_copy(): cli_obj = _make_cli() cli_obj.conversation_history = [ { "role": "assistant", "content": "<REASONING_SCRATCHPAD>internal</REASONING_SCRATCHPAD>\nVisible answer", } ]
with patch.object(cli_obj, "_write_osc52_clipboard") as mock_copy: cli_obj.process_command("/copy")
mock_copy.assert_called_once_with("Visible answer")
def test_copy_invalid_index_does_not_copy(): cli_obj = _make_cli() cli_obj.conversation_history = [{"role": "assistant", "content": "only"}]
with patch.object(cli_obj, "_write_osc52_clipboard") as mock_copy, patch("cli._cprint") as mock_print: cli_obj.process_command("/copy 99")
mock_copy.assert_not_called() assert any("Invalid response number" in str(call) for call in mock_print.call_args_list)