Function _make_agent_manager

tests/agents/test_group_store.py:24–47

Create a mock AgentManager with the given agent configs.

(*agent_configs)

def _make_agent_manager(*agent_configs):
    """Create a mock AgentManager with the given agent configs."""
    mgr = MagicMock()
    snapshots = {}
    statuses = []

    for cfg in agent_configs:
        ctx = MagicMock()
        ctx.agent_id = cfg.agent_id
        ctx.session_id = f"session-{cfg.agent_id}"
        ctx.conversation_history = []
        snapshots[cfg.agent_id] = {"config": cfg, "context": ctx}
        statuses.append(
            {
                "agent_id": cfg.agent_id,
                "name": cfg.name,
                "role": cfg.role,
                "status": "active",
            }
        )

    mgr.list_agents.return_value = statuses
    mgr.get_agent_snapshot.side_effect = lambda aid: snapshots.get(aid)
    return mgr


# ---------------------------------------------------------------------------