Function wait_for_event

python/e2e/test_multi_client_e2e.py:190–206

(session, predicate, timeout: float = 30.0)

def wait_for_event(session, predicate, timeout: float = 30.0):
    loop = asyncio.get_running_loop()
    future = loop.create_future()

    def on_event(event):
        if not future.done() and predicate(event):
            future.set_result(event)

    unsubscribe = session.on(on_event)

    async def wait():
        try:
            return await asyncio.wait_for(future, timeout=timeout)
        finally:
            unsubscribe()

    return loop.create_task(wait())


class TestMultiClientBroadcast: