fix(BA-4974): replace xfail with proper agent_registry mocking for transit status tests

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Author: jopemachine

tests/component/session/test_session_lifecycle.py

@@ -6,6 +6,7 @@
 from dataclasses import dataclass
 from datetime import datetime
 from typing import Any
+from unittest.mock import AsyncMock, MagicMock
 
 import pytest
 import sqlalchemy as sa
@@ -176,29 +177,49 @@ async def test_restart_nonexistent_session_returns_not_found(
 class TestSessionStatusTransition:
     """Test session status transit endpoint (POST /_/transit-status)."""
 
-    @pytest.mark.xfail(
-        reason="Requires live agent — agent_registry mock does not return proper transit result"
-    )
     async def test_transit_single_session_status(
         self,
         admin_registry: BackendAIClientRegistry,
         session_seed: SessionSeedData,
+        agent_registry: AsyncMock,
     ) -> None:
+        mock_lifecycle_mgr = AsyncMock()
+        mock_row = MagicMock()
+        mock_row.id = session_seed.session_id
+        mock_row.status = SessionStatus.RUNNING
+        mock_lifecycle_mgr.transit_session_status.return_value = [(mock_row, False)]
+        mock_lifecycle_mgr.deregister_status_updatable_session.return_value = 0
+        agent_registry.session_lifecycle_mgr = mock_lifecycle_mgr
+
         result = await admin_registry.session.transit_session_status(
             TransitSessionStatusRequest(ids=[session_seed.session_id]),
         )
         assert isinstance(result, TransitSessionStatusResponse)
         assert isinstance(result.session_status_map, dict)
 
-    @pytest.mark.xfail(
-        reason="Requires live agent — agent_registry mock does not return proper transit result"
-    )
     async def test_transit_multiple_sessions(
         self,
         admin_registry: BackendAIClientRegistry,
         session_seed: SessionSeedData,
         terminated_session_seed: SessionSeedData,
+        agent_registry: AsyncMock,
     ) -> None:
+        mock_lifecycle_mgr = AsyncMock()
+        mock_row_running = MagicMock()
+        mock_row_running.id = session_seed.session_id
+        mock_row_running.status = SessionStatus.RUNNING
+
+        mock_row_terminated = MagicMock()
+        mock_row_terminated.id = terminated_session_seed.session_id
+        mock_row_terminated.status = SessionStatus.TERMINATED
+
+        mock_lifecycle_mgr.transit_session_status.side_effect = [
+            [(mock_row_running, False)],
+            [(mock_row_terminated, False)],
+        ]
+        mock_lifecycle_mgr.deregister_status_updatable_session.return_value = 0
+        agent_registry.session_lifecycle_mgr = mock_lifecycle_mgr
+
         result = await admin_registry.session.transit_session_status(
             TransitSessionStatusRequest(
                 ids=[session_seed.session_id, terminated_session_seed.session_id],