Function _make_mock_tool_manager

tests/context/test_context_manager.py:60–65

Return a mock ToolManager with async helpers wired up.

(servers=None, tools=None)

def _make_mock_tool_manager(servers=None, tools=None) -> MagicMock:
    """Return a mock ToolManager with async helpers wired up."""
    tm = MagicMock()
    tm.get_server_info = AsyncMock(return_value=servers or [])
    tm.get_all_tools = AsyncMock(return_value=tools or [])
    return tm


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