Function _make_stream_manager

tests/chat/test_testing.py:21–43

Build a mock stream_manager with configurable behaviour.

(
    tools=None, server_info=None, has_call_tool=True, has_internal_tools=True
)

def _make_stream_manager(
    tools=None, server_info=None, has_call_tool=True, has_internal_tools=True
):
    """Build a mock stream_manager with configurable behaviour."""
    mgr = MagicMock()
    tools = tools if tools is not None else [_make_tool()]
    server_info = server_info if server_info is not None else [{"name": "server1"}]

    if has_internal_tools:
        mgr.get_internal_tools.return_value = tools
    else:
        del mgr.get_internal_tools  # remove the attr so hasattr returns False
        mgr.get_all_tools.return_value = tools

    mgr.get_server_info.return_value = server_info
    mgr.get_server_for_tool.return_value = "test-server"

    if has_call_tool:
        mgr.call_tool = AsyncMock(return_value={"result": "ok"})
    else:
        del mgr.call_tool  # remove so hasattr returns False

    return mgr


def _make_model_manager():