async def test_sched_error(caplog):
caplog.set_level(logging.WARNING, logger="psycopg")
s = AsyncScheduler()
t = create_task(s.run())
results = []
async def worker(i):
results.append((i, time()))
async def error():
1 / 0
t0 = time()
await s.enter(0.1, partial(worker, 1))
await s.enter(0.4, None)
await s.enter(0.3, partial(worker, 2))
await s.enter(0.2, error)
await asyncio.gather(t)
t1 = time()
assert t1 - t0 == pytest.approx(0.4, 0.1)
assert len(results) == 2
assert results[0][0] == 1
assert results[0][1] - t0 == pytest.approx(0.1, 0.1)
assert results[1][0] == 2
assert results[1][1] - t0 == pytest.approx(0.3, 0.1)
assert len(caplog.records) == 1
assert "ZeroDivisionError" in caplog.records[0].message
async def test_first_task_rescheduling():
s = AsyncScheduler()
t0 = time()
times = []
wait_orig = s._event.wait
async def wait_logging():
try:
rv = await wait_orig()
finally:
times.append(time() - t0)
return rv
s._event.wait = wait_logging
s.EMPTY_QUEUE_TIMEOUT = 0.1
async def noop():
pass
await s.enter(0.4, noop)
t = create_task(s.run())
await s.enter(0.6, None) # this task doesn't trigger a reschedule
await asyncio.sleep(0.1)
await s.enter(0.1, noop) # this triggers a reschedule
await asyncio.gather(t)
times.append(time() - t0)
for got, want in zip(times, [0.1, 0.2, 0.4, 0.6, 0.6]):
assert got == pytest.approx(want, 0.1), times
async def test_empty_queue_timeout():
s = AsyncScheduler()
t0 = time()
times = []
wait_orig = s._event.wait
async def wait_logging():
try:
rv = await wait_orig()
finally:
times.append(time() - t0)
return rv
s._event.wait = wait_logging
s.EMPTY_QUEUE_TIMEOUT = 0.2
t = create_task(s.run())
await asyncio.sleep(0.5)
await s.enter(0.5, None)
await asyncio.gather(t)
times.append(time() - t0)
for got, want in zip(times, [0.2, 0.4, 0.5, 1.0]):
assert got == pytest.approx(want, 0.1), times
async def test_sched_task():
s = AsyncScheduler()
t = create_task(s.run())
results = []
async def worker(i):
results.append((i, time()))
t0 = time()
await s.enter(0.1, partial(worker, 1))
await s.enter(0.4, partial(worker, 3))
await s.enter(0.3, None)
await s.enter(0.2, partial(worker, 2))
await asyncio.gather(t)
t1 = time()
assert t1 - t0 == pytest.approx(0.3, 0.1)
assert len(results) == 2
assert results[0][0] == 1
assert results[0][1] - t0 == pytest.approx(0.1, 0.1)
assert results[1][0] == 2
assert results[1][1] - t0 == pytest.approx(0.2, 0.1)