def wait_for(ms):
future = Future()
timer = QTimer()
timer.timeout.connect(lambda:
(future.set_result(None), timer.deleteLater()))
timer.setSingleShot(True)
timer.start(ms)
return future
def test_await_finished():
f = Future()
f.set_result(42)
async def entry():
return await f
sched = Scheduler()
assert sched.run_until_complete(entry()) == 42
def test_await_finished_exc():
f = Future()
f.set_exception(KeyError)
async def entry():
return await f
sched = Scheduler()
with pytest.raises(KeyError):
sched.run_until_complete(entry())
def test_cancel_recursively():
done_order = []
fut1 = Future()
fut2 = Future()
async def some_recursion(n=10):
if not n:
await fut1
else:
await some_recursion(n - 1)
async def entry():
await some_recursion()
await fut2
sched = Scheduler()
ef = sched.run(entry())
fut1.add_done_callback(done_order.append)
ef.add_done_callback(done_order.append)
sched.tick()
ef.cancel()
sched.tick()
assert fut1.cancelled()
assert ef.cancelled()
assert done_order == [fut1, ef]
assert not fut2.cancelled()
def signal(sig):
def emitted(*args):
sig.disconnect(emitted)
future.set_result(args)
future = Future()
sig.connect(emitted)
return future
def test_as_completed_cancel():
"""
Check: If a source future to as_completed() completes after as_completed() is cancelled,
no crash occurs and the source future completion is ignored.
"""
source_fut = Future()
sched = Scheduler()
ac = as_completed(source_fut, scheduler=sched)
as_completed_fut = sched.run(ac.__anext__())
sched.tick()
as_completed_fut.cancel()
sched.tick()
with pytest.raises(CancelledError):
sched.run_until_complete(as_completed_fut)
source_fut.set_exception(KeyError)
sched.tick()
def test_loop():
f = Future()
def get_response():
return f
async def entry():
return await get_response()
sched = Scheduler()
entry_future = sched.run(entry())
sched.tick()
assert not entry_future.done()
f.set_result(666)
assert not entry_future.done()
sched.tick()
assert entry_future.done()
assert entry_future.result() == 666
def test_gather_cancel():
"""
Check: If a gather() future is cancelled, all source futures are cancelled.
"""
fut1 = Future()
fut2 = Future()
fut3 = Future()
sched = Scheduler()
gf = gather(fut1, fut2, fut3, scheduler=sched)
assert not gf.done()
gf.cancel()
sched.tick()
assert fut1.cancelled()
assert fut2.cancelled()
assert fut3.cancelled()
assert gf.cancelled()
def test_loop_exc():
f = Future()
def get_response():
return f
async def entry():
return await get_response()
sched = Scheduler()
entry_future = sched.run(entry())
sched.tick()
assert not entry_future.done()
f.set_exception(KeyError)
assert not entry_future.done()
sched.tick()
assert entry_future.done()
with pytest.raises(KeyError):
entry_future.result()
def test_as_completed_exc():
"""
Check: If a source future to as_completed() throws an exception,
the iterator re-raises the exception and exits,
cancelling all remaining source futures.
"""
fut1 = Future()
fut2 = Future()
fut3 = Future()
sched = Scheduler()
ac = as_completed(fut1, fut2, fut3, scheduler=sched)
fut1.set_result(5)
assert sched.run_until_complete(ac.__anext__()) == 5
fut2.set_exception(KeyError)
with pytest.raises(KeyError):
sched.run_until_complete(ac.__anext__())
assert fut3.cancelled()
with pytest.raises(StopAsyncIteration):
sched.run_until_complete(ac.__anext__())
def test_as_completed():
fut1 = Future()
fut2 = Future()
fut3 = Future()
sched = Scheduler()
ac = as_completed(fut1, fut2, fut3, scheduler=sched)
fut1.set_result(5)
assert sched.run_until_complete(ac.__anext__()) == 5
fut2.set_result(1)
assert sched.run_until_complete(ac.__anext__()) == 1
fut3.set_result(3)
assert sched.run_until_complete(ac.__anext__()) == 3
def test_gather_exc():
fut1 = Future()
fut2 = Future()
fut3 = Future()
sched = Scheduler()
gf = gather(fut1, fut2, fut3, scheduler=sched)
assert not gf.done()
fut1.set_result(1)
assert not gf.done()
fut3.set_exception(KeyError)
assert not gf.done()
sched.tick()
assert gf.done()
# fut3 failed, so the other pending futures should enter cancelled state.
assert fut2.cancelled()
with pytest.raises(KeyError):
gf.result()
def test_gather():
fut1 = Future()
fut2 = Future()
fut3 = Future()
sched = Scheduler()
gf = gather(fut1, fut2, fut3, scheduler=sched)
assert not gf.done()
fut1.set_result(1)
assert not gf.done()
fut3.set_result(3)
fut2.set_result(2)
assert not gf.done()
sched.tick()
assert gf.done()
assert gf.result() == [1, 3, 2]