"""Tests for /resume gateway slash command. Tests the _handle_resume_command handler (switch to a previously-named session) across gateway messenger platforms. """ from unittest.mock import MagicMock, AsyncMock import pytest from gateway.config import Platform from gateway.platforms.base import MessageEvent from gateway.session import SessionSource, build_session_key def _make_event(text="/resume", platform=Platform.TELEGRAM, user_id="12345", chat_id="67890"): """Build a MessageEvent for testing.""" source = SessionSource( platform=platform, user_id=user_id, chat_id=chat_id, user_name="testuser", ) return MessageEvent(text=text, source=source) def _session_key_for_event(event): """Get the session key that build_session_key produces for an event.""" return build_session_key(event.source) def _make_runner(session_db=None, current_session_id="current_session_001", event=None): """Create a bare GatewayRunner with a mock session_store and optional session_db.""" from gateway.run import GatewayRunner runner = object.__new__(GatewayRunner) runner.adapters = {} runner._session_db = session_db runner._running_agents = {} # Compute the real session key if an event is provided session_key = build_session_key(event.source) if event else "agent:main:telegram:dm" # Mock session_store that returns a session entry with a known session_id mock_session_entry = MagicMock() mock_session_entry.session_id = current_session_id mock_session_entry.session_key = session_key mock_store = MagicMock() mock_store.get_or_create_session.return_value = mock_session_entry mock_store.load_transcript.return_value = [] mock_store.switch_session.return_value = mock_session_entry runner.session_store = mock_store # Stub out memory flushing runner._async_flush_memories = AsyncMock() return runner # --------------------------------------------------------------------------- # _handle_resume_command # --------------------------------------------------------------------------- class TestHandleResumeCommand: """Tests for GatewayRunner._handle_resume_command.""" @pytest.mark.asyncio async def test_no_session_db(self): """Returns error when session database is unavailable.""" runner = _make_runner(session_db=None) event = _make_event(text="/resume My Project") result = await runner._handle_resume_command(event) assert "not available" in result.lower() @pytest.mark.asyncio async def test_list_named_sessions_when_no_arg(self, tmp_path): """With no argument, lists recently titled sessions.""" from hermes_state import SessionDB db = SessionDB(db_path=tmp_path / "state.db") db.create_session("sess_001", "telegram") db.create_session("sess_002", "telegram") db.set_session_title("sess_001", "Research") db.set_session_title("sess_002", "Coding") event = _make_event(text="/resume") runner = _make_runner(session_db=db, event=event) result = await runner._handle_resume_command(event) assert "Research" in result assert "Coding" in result assert "Named Sessions" in result db.close() @pytest.mark.asyncio async def test_list_shows_usage_when_no_titled(self, tmp_path): """With no arg and no titled sessions, shows instructions.""" from hermes_state import SessionDB db = SessionDB(db_path=tmp_path / "state.db") db.create_session("sess_001", "telegram") # No title event = _make_event(text="/resume") runner = _make_runner(session_db=db, event=event) result = await runner._handle_resume_command(event) assert "No named sessions" in result assert "/title" in result db.close() @pytest.mark.asyncio async def test_resume_by_name(self, tmp_path): """Resolves a title and switches to that session.""" from hermes_state import SessionDB db = SessionDB(db_path=tmp_path / "state.db") db.create_session("old_session_abc", "telegram") db.set_session_title("old_session_abc", "My Project") db.create_session("current_session_001", "telegram") event = _make_event(text="/resume My Project") runner = _make_runner(session_db=db, current_session_id="current_session_001", event=event) result = await runner._handle_resume_command(event) assert "Resumed" in result assert "My Project" in result # Verify switch_session was called with the old session ID runner.session_store.switch_session.assert_called_once() call_args = runner.session_store.switch_session.call_args assert call_args[0][1] == "old_session_abc" db.close() @pytest.mark.asyncio async def test_resume_nonexistent_name(self, tmp_path): """Returns error for unknown session name.""" from hermes_state import SessionDB db = SessionDB(db_path=tmp_path / "state.db") db.create_session("current_session_001", "telegram") event = _make_event(text="/resume Nonexistent Session") runner = _make_runner(session_db=db, event=event) result = await runner._handle_resume_command(event) assert "No session found" in result db.close() @pytest.mark.asyncio async def test_resume_already_on_session(self, tmp_path): """Returns friendly message when already on the requested session.""" from hermes_state import SessionDB db = SessionDB(db_path=tmp_path / "state.db") db.create_session("current_session_001", "telegram") db.set_session_title("current_session_001", "Active Project") event = _make_event(text="/resume Active Project") runner = _make_runner(session_db=db, current_session_id="current_session_001", event=event) result = await runner._handle_resume_command(event) assert "Already on session" in result db.close() @pytest.mark.asyncio async def test_resume_auto_lineage(self, tmp_path): """Asking for 'My Project' when 'My Project #2' exists gets the latest.""" from hermes_state import SessionDB db = SessionDB(db_path=tmp_path / "state.db") db.create_session("sess_v1", "telegram") db.set_session_title("sess_v1", "My Project") db.create_session("sess_v2", "telegram") db.set_session_title("sess_v2", "My Project #2") db.create_session("current_session_001", "telegram") event = _make_event(text="/resume My Project") runner = _make_runner(session_db=db, current_session_id="current_session_001", event=event) result = await runner._handle_resume_command(event) assert "Resumed" in result # Should resolve to #2 (latest in lineage) call_args = runner.session_store.switch_session.call_args assert call_args[0][1] == "sess_v2" db.close() @pytest.mark.asyncio async def test_resume_clears_running_agent(self, tmp_path): """Switching sessions clears any cached running agent.""" from hermes_state import SessionDB db = SessionDB(db_path=tmp_path / "state.db") db.create_session("old_session", "telegram") db.set_session_title("old_session", "Old Work") db.create_session("current_session_001", "telegram") event = _make_event(text="/resume Old Work") runner = _make_runner(session_db=db, current_session_id="current_session_001", event=event) # Simulate a running agent using the real session key real_key = _session_key_for_event(event) runner._running_agents[real_key] = MagicMock() await runner._handle_resume_command(event) assert real_key not in runner._running_agents db.close()