Something went wrong. Try again.
The agent that grows with you
Something went wrong. Try again.
9.6 kB · 249 lines
Python
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250"""Tests for the /branch (/fork) command — session branching.
Verifies that:- Branching creates a new session with copied conversation history- The original session is preserved (ended with "branched" reason)- Auto-generated titles use lineage numbering- Custom branch names are used when provided- parent_session_id links are set correctly- Edge cases: empty conversation, missing session DB"""
import osimport uuidfrom datetime import datetimefrom pathlib import Pathfrom unittest.mock import MagicMock, patch, PropertyMock
import pytest
@pytest.fixturedef session_db(tmp_path): """Create a real SessionDB for testing.""" os.environ["HERMES_HOME"] = str(tmp_path / ".hermes") os.makedirs(tmp_path / ".hermes", exist_ok=True) from hermes_state import SessionDB db = SessionDB(db_path=tmp_path / ".hermes" / "test_sessions.db") yield db db.close()
@pytest.fixturedef cli_instance(tmp_path, session_db): """Create a minimal HermesCLI-like object for testing _handle_branch_command.""" # We'll mock the CLI enough to test the branch logic without full init from unittest.mock import MagicMock
cli = MagicMock() cli._session_db = session_db cli.session_id = "20260403_120000_abc123" cli.model = "anthropic/claude-sonnet-4.6" cli.max_turns = 90 cli.reasoning_config = {"enabled": True, "effort": "medium"} cli.session_start = datetime.now() cli._pending_title = None cli._resumed = False cli.agent = None cli.conversation_history = [ {"role": "user", "content": "Hello, can you help me?"}, {"role": "assistant", "content": "Of course! How can I help?"}, {"role": "user", "content": "Write a Python function to sort a list."}, {"role": "assistant", "content": "def sort_list(lst): return sorted(lst)"}, ]
# Create the original session in the DB session_db.create_session( session_id=cli.session_id, source="cli", model=cli.model, ) session_db.set_session_title(cli.session_id, "My Coding Session")
return cli
class TestBranchCommandCLI: """Test the /branch command logic for the CLI."""
def test_branch_creates_new_session(self, cli_instance, session_db): """Branching should create a new session in the DB.""" from cli import HermesCLI
# Call the real method on the mock, using the real implementation HermesCLI._handle_branch_command(cli_instance, "/branch")
# Verify a new session was created assert cli_instance.session_id != "20260403_120000_abc123" new_session = session_db.get_session(cli_instance.session_id) assert new_session is not None
def test_branch_copies_history(self, cli_instance, session_db): """Branching should copy all messages to the new session.""" from cli import HermesCLI
HermesCLI._handle_branch_command(cli_instance, "/branch")
messages = session_db.get_messages_as_conversation(cli_instance.session_id) assert len(messages) == 4 # All 4 messages copied
def test_branch_preserves_parent_link(self, cli_instance, session_db): """The new session should reference the original as parent.""" from cli import HermesCLI original_id = cli_instance.session_id
HermesCLI._handle_branch_command(cli_instance, "/branch")
new_session = session_db.get_session(cli_instance.session_id) assert new_session["parent_session_id"] == original_id
def test_branch_ends_original_session(self, cli_instance, session_db): """The original session should be marked as ended with 'branched' reason.""" from cli import HermesCLI original_id = cli_instance.session_id
HermesCLI._handle_branch_command(cli_instance, "/branch")
original = session_db.get_session(original_id) assert original["end_reason"] == "branched"
def test_branch_with_custom_name(self, cli_instance, session_db): """Custom branch name should be used as the title.""" from cli import HermesCLI
HermesCLI._handle_branch_command(cli_instance, "/branch refactor approach")
title = session_db.get_session_title(cli_instance.session_id) assert title == "refactor approach"
def test_branch_auto_title_lineage(self, cli_instance, session_db): """Without a name, branch should auto-generate a title from the parent's title.""" from cli import HermesCLI
HermesCLI._handle_branch_command(cli_instance, "/branch")
title = session_db.get_session_title(cli_instance.session_id) assert title == "My Coding Session #2"
def test_branch_empty_conversation(self, cli_instance, session_db): """Branching with no history should show an error.""" from cli import HermesCLI cli_instance.conversation_history = []
HermesCLI._handle_branch_command(cli_instance, "/branch")
# session_id should not have changed assert cli_instance.session_id == "20260403_120000_abc123"
def test_branch_no_session_db(self, cli_instance): """Branching without a session DB should show an error.""" from cli import HermesCLI cli_instance._session_db = None
HermesCLI._handle_branch_command(cli_instance, "/branch")
# session_id should not have changed assert cli_instance.session_id == "20260403_120000_abc123"
def test_branch_syncs_agent(self, cli_instance, session_db): """If an agent is active, branch should sync it to the new session.""" from cli import HermesCLI
agent = MagicMock() agent._last_flushed_db_idx = 0 cli_instance.agent = agent
HermesCLI._handle_branch_command(cli_instance, "/branch")
# Agent should have been updated assert agent.session_id == cli_instance.session_id assert agent.reset_session_state.called assert agent._last_flushed_db_idx == 4 # len(conversation_history)
def test_branch_updates_agent_session_log_file(self, cli_instance, session_db, tmp_path): """Branching must redirect the agent's session_log_file to the new session's path.""" from cli import HermesCLI from pathlib import Path
logs_dir = tmp_path / "sessions" logs_dir.mkdir()
agent = MagicMock() agent._last_flushed_db_idx = 0 agent.logs_dir = logs_dir agent.session_log_file = logs_dir / f"session_{cli_instance.session_id}.json" cli_instance.agent = agent
old_log_file = agent.session_log_file HermesCLI._handle_branch_command(cli_instance, "/branch")
new_session_id = cli_instance.session_id expected_log = logs_dir / f"session_{new_session_id}.json" assert agent.session_log_file == expected_log, ( "session_log_file must point to the branch session, not the original" ) assert agent.session_log_file != old_log_file
def test_branch_sets_resumed_flag(self, cli_instance, session_db): """Branch should set _resumed=True to prevent auto-title generation.""" from cli import HermesCLI
HermesCLI._handle_branch_command(cli_instance, "/branch")
assert cli_instance._resumed is True
def test_branch_fires_on_session_switch_hook(self, cli_instance, session_db): """The /branch command must notify memory providers of the rotation.
Without this, providers that cache per-session state in initialize() keep writing under the old session_id. See #6672. """ from cli import HermesCLI
# Wire a real-ish agent object with a MagicMock memory_manager agent = MagicMock() mm = MagicMock() agent._memory_manager = mm cli_instance.agent = agent original_id = cli_instance.session_id
HermesCLI._handle_branch_command(cli_instance, "/branch")
# Hook must have been called exactly once with the new session_id, # parent pointing at the branched-from session, reset=False, and # reason="branch" for diagnostics. assert mm.on_session_switch.call_count == 1 _, kwargs = mm.on_session_switch.call_args assert mm.on_session_switch.call_args.args[0] == cli_instance.session_id assert kwargs["parent_session_id"] == original_id assert kwargs["reset"] is False assert kwargs["reason"] == "branch"
def test_fork_alias(self): """The /fork alias should resolve to 'branch'.""" from hermes_cli.commands import resolve_command result = resolve_command("fork") assert result is not None assert result.name == "branch"
class TestBranchCommandDef: """Test the CommandDef registration for /branch."""
def test_branch_in_registry(self): """The branch command should be in the command registry.""" from hermes_cli.commands import COMMAND_REGISTRY names = [c.name for c in COMMAND_REGISTRY] assert "branch" in names
def test_branch_has_fork_alias(self): """The branch command should have 'fork' as an alias.""" from hermes_cli.commands import COMMAND_REGISTRY branch = next(c for c in COMMAND_REGISTRY if c.name == "branch") assert "fork" in branch.aliases
def test_branch_in_session_category(self): """The branch command should be in the Session category.""" from hermes_cli.commands import COMMAND_REGISTRY branch = next(c for c in COMMAND_REGISTRY if c.name == "branch") assert branch.category == "Session"