Function mock_get_connection

benchmarks/dirty_streaming.py:259–260

(pid)

    mock_reader = MockStreamReader(messages)

    async def mock_get_connection(pid):
        return mock_reader, MockStreamWriter()

    arbiter._get_worker_connection = mock_get_connection