Method on_wait

tests/unit/transport/test_consumer.py:460–462

(timeout)

        consumer._wait_next_records = AsyncMock()

        async def on_wait(timeout):
            consumer._stopped.set()
            return None, None

        consumer._wait_next_records.side_effect = on_wait
        assert [a async for a in consumer.getmany(1.0)] == []