Something went wrong. Try again.
The agent that grows with you
Something went wrong. Try again.
4.5 kB · 123 lines
Python
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124"""Tests for the /busy CLI command and busy-input-mode config handling."""
import unittestfrom types import SimpleNamespacefrom unittest.mock import patch
def _import_cli(): import hermes_cli.config as config_mod
if not hasattr(config_mod, "save_env_value_secure"): config_mod.save_env_value_secure = lambda key, value: { "success": True, "stored_as": key, "validated": False, }
import cli as cli_mod
return cli_mod
class TestHandleBusyCommand(unittest.TestCase): def _make_cli(self, busy_input_mode="interrupt"): return SimpleNamespace( busy_input_mode=busy_input_mode, agent=None, )
def test_no_args_shows_status(self): cli_mod = _import_cli() stub = self._make_cli("queue") with ( patch.object(cli_mod, "_cprint") as mock_cprint, patch.object(cli_mod, "save_config_value") as mock_save, ): cli_mod.HermesCLI._handle_busy_command(stub, "/busy")
mock_save.assert_not_called() printed = " ".join(str(c) for c in mock_cprint.call_args_list) self.assertIn("queue", printed) self.assertIn("interrupt", printed)
def test_queue_argument_sets_queue_mode_and_saves(self): cli_mod = _import_cli() stub = self._make_cli("interrupt") with ( patch.object(cli_mod, "_cprint"), patch.object(cli_mod, "save_config_value", return_value=True) as mock_save, ): cli_mod.HermesCLI._handle_busy_command(stub, "/busy queue")
self.assertEqual(stub.busy_input_mode, "queue") mock_save.assert_called_once_with("display.busy_input_mode", "queue")
def test_interrupt_argument_sets_interrupt_mode_and_saves(self): cli_mod = _import_cli() stub = self._make_cli("queue") with ( patch.object(cli_mod, "_cprint"), patch.object(cli_mod, "save_config_value", return_value=True) as mock_save, ): cli_mod.HermesCLI._handle_busy_command(stub, "/busy interrupt")
self.assertEqual(stub.busy_input_mode, "interrupt") mock_save.assert_called_once_with("display.busy_input_mode", "interrupt")
def test_steer_argument_sets_steer_mode_and_saves(self): cli_mod = _import_cli() stub = self._make_cli("interrupt") with ( patch.object(cli_mod, "_cprint") as mock_cprint, patch.object(cli_mod, "save_config_value", return_value=True) as mock_save, ): cli_mod.HermesCLI._handle_busy_command(stub, "/busy steer")
self.assertEqual(stub.busy_input_mode, "steer") mock_save.assert_called_once_with("display.busy_input_mode", "steer") printed = " ".join(str(c) for c in mock_cprint.call_args_list) self.assertIn("steer", printed.lower())
def test_status_reports_steer_behavior(self): cli_mod = _import_cli() stub = self._make_cli("steer") with ( patch.object(cli_mod, "_cprint") as mock_cprint, patch.object(cli_mod, "save_config_value") as mock_save, ): cli_mod.HermesCLI._handle_busy_command(stub, "/busy status")
mock_save.assert_not_called() printed = " ".join(str(c) for c in mock_cprint.call_args_list) self.assertIn("steer", printed.lower()) # The usage line should also advertise the steer option self.assertIn("steer", printed)
def test_invalid_argument_prints_usage(self): cli_mod = _import_cli() stub = self._make_cli() with ( patch.object(cli_mod, "_cprint") as mock_cprint, patch.object(cli_mod, "save_config_value") as mock_save, ): cli_mod.HermesCLI._handle_busy_command(stub, "/busy nonsense")
mock_save.assert_not_called() printed = " ".join(str(c) for c in mock_cprint.call_args_list) self.assertIn("Usage: /busy", printed)
class TestBusyCommandRegistry(unittest.TestCase): def test_busy_in_registry(self): from hermes_cli.commands import COMMAND_REGISTRY
names = [c.name for c in COMMAND_REGISTRY] assert "busy" in names
def test_busy_subcommands_documented(self): from hermes_cli.commands import COMMAND_REGISTRY
busy = next(c for c in COMMAND_REGISTRY if c.name == "busy") assert busy.args_hint == "[queue|steer|interrupt|status]" assert busy.category == "Configuration"