mirror of
https://github.com/python/cpython.git
synced 2025-10-31 05:31:20 +00:00
bpo-47038: Rewrite asyncio.wait_for test to use IsolatedAsyncioTestCase (GH-31942)
This commit is contained in:
Andrew Svetlov
GitHub
parent
dbbe4d2d00
commit
dd0082c627
3 changed files with 270 additions and 313 deletions
|
|
@ -1,61 +0,0 @@
|
|||
import asyncio
|
||||
import unittest
|
||||
import time
|
||||
|
||||
def tearDownModule():
|
||||
asyncio.set_event_loop_policy(None)
|
||||
|
||||
|
||||
class SlowTask:
|
||||
""" Task will run for this defined time, ignoring cancel requests """
|
||||
TASK_TIMEOUT = 0.2
|
||||
|
||||
def __init__(self):
|
||||
self.exited = False
|
||||
|
||||
async def run(self):
|
||||
exitat = time.monotonic() + self.TASK_TIMEOUT
|
||||
|
||||
while True:
|
||||
tosleep = exitat - time.monotonic()
|
||||
if tosleep <= 0:
|
||||
break
|
||||
|
||||
try:
|
||||
await asyncio.sleep(tosleep)
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
self.exited = True
|
||||
|
||||
class AsyncioWaitForTest(unittest.TestCase):
|
||||
|
||||
async def atest_asyncio_wait_for_cancelled(self):
|
||||
t = SlowTask()
|
||||
|
||||
waitfortask = asyncio.create_task(asyncio.wait_for(t.run(), t.TASK_TIMEOUT * 2))
|
||||
await asyncio.sleep(0)
|
||||
waitfortask.cancel()
|
||||
await asyncio.wait({waitfortask})
|
||||
|
||||
self.assertTrue(t.exited)
|
||||
|
||||
def test_asyncio_wait_for_cancelled(self):
|
||||
asyncio.run(self.atest_asyncio_wait_for_cancelled())
|
||||
|
||||
async def atest_asyncio_wait_for_timeout(self):
|
||||
t = SlowTask()
|
||||
|
||||
try:
|
||||
await asyncio.wait_for(t.run(), t.TASK_TIMEOUT / 2)
|
||||
except asyncio.TimeoutError:
|
||||
pass
|
||||
|
||||
self.assertTrue(t.exited)
|
||||
|
||||
def test_asyncio_wait_for_timeout(self):
|
||||
asyncio.run(self.atest_asyncio_wait_for_timeout())
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
|
|
@ -84,12 +84,6 @@ def __await__(self):
|
|||
return self
|
||||
|
||||
|
||||
# The following value can be used as a very small timeout:
|
||||
# it passes check "timeout > 0", but has almost
|
||||
# no effect on the test performance
|
||||
_EPSILON = 0.0001
|
||||
|
||||
|
||||
class BaseTaskTests:
|
||||
|
||||
Task = None
|
||||
|
|
@ -107,7 +101,6 @@ def setUp(self):
|
|||
self.loop.set_task_factory(self.new_task)
|
||||
self.loop.create_future = lambda: self.new_future(self.loop)
|
||||
|
||||
|
||||
def test_generic_alias(self):
|
||||
task = self.__class__.Task[str]
|
||||
self.assertEqual(task.__args__, (str,))
|
||||
|
|
@ -971,251 +964,6 @@ async def coro():
|
|||
task._log_traceback = True
|
||||
self.loop.run_until_complete(task)
|
||||
|
||||
def test_wait_for_timeout_less_then_0_or_0_future_done(self):
|
||||
def gen():
|
||||
when = yield
|
||||
self.assertAlmostEqual(0, when)
|
||||
|
||||
loop = self.new_test_loop(gen)
|
||||
|
||||
fut = self.new_future(loop)
|
||||
fut.set_result('done')
|
||||
|
||||
ret = loop.run_until_complete(asyncio.wait_for(fut, 0))
|
||||
|
||||
self.assertEqual(ret, 'done')
|
||||
self.assertTrue(fut.done())
|
||||
self.assertAlmostEqual(0, loop.time())
|
||||
|
||||
def test_wait_for_timeout_less_then_0_or_0_coroutine_do_not_started(self):
|
||||
def gen():
|
||||
when = yield
|
||||
self.assertAlmostEqual(0, when)
|
||||
|
||||
loop = self.new_test_loop(gen)
|
||||
|
||||
foo_started = False
|
||||
|
||||
async def foo():
|
||||
nonlocal foo_started
|
||||
foo_started = True
|
||||
|
||||
with self.assertRaises(asyncio.TimeoutError):
|
||||
loop.run_until_complete(asyncio.wait_for(foo(), 0))
|
||||
|
||||
self.assertAlmostEqual(0, loop.time())
|
||||
self.assertEqual(foo_started, False)
|
||||
|
||||
def test_wait_for_timeout_less_then_0_or_0(self):
|
||||
def gen():
|
||||
when = yield
|
||||
self.assertAlmostEqual(0.2, when)
|
||||
when = yield 0
|
||||
self.assertAlmostEqual(0, when)
|
||||
|
||||
for timeout in [0, -1]:
|
||||
with self.subTest(timeout=timeout):
|
||||
loop = self.new_test_loop(gen)
|
||||
|
||||
foo_running = None
|
||||
|
||||
async def foo():
|
||||
nonlocal foo_running
|
||||
foo_running = True
|
||||
try:
|
||||
await asyncio.sleep(0.2)
|
||||
finally:
|
||||
foo_running = False
|
||||
return 'done'
|
||||
|
||||
fut = self.new_task(loop, foo())
|
||||
|
||||
with self.assertRaises(asyncio.TimeoutError):
|
||||
loop.run_until_complete(asyncio.wait_for(fut, timeout))
|
||||
self.assertTrue(fut.done())
|
||||
# it should have been cancelled due to the timeout
|
||||
self.assertTrue(fut.cancelled())
|
||||
self.assertAlmostEqual(0, loop.time())
|
||||
self.assertEqual(foo_running, False)
|
||||
|
||||
def test_wait_for(self):
|
||||
|
||||
def gen():
|
||||
when = yield
|
||||
self.assertAlmostEqual(0.2, when)
|
||||
when = yield 0
|
||||
self.assertAlmostEqual(0.1, when)
|
||||
when = yield 0.1
|
||||
|
||||
loop = self.new_test_loop(gen)
|
||||
|
||||
foo_running = None
|
||||
|
||||
async def foo():
|
||||
nonlocal foo_running
|
||||
foo_running = True
|
||||
try:
|
||||
await asyncio.sleep(0.2)
|
||||
finally:
|
||||
foo_running = False
|
||||
return 'done'
|
||||
|
||||
fut = self.new_task(loop, foo())
|
||||
|
||||
with self.assertRaises(asyncio.TimeoutError):
|
||||
loop.run_until_complete(asyncio.wait_for(fut, 0.1))
|
||||
self.assertTrue(fut.done())
|
||||
# it should have been cancelled due to the timeout
|
||||
self.assertTrue(fut.cancelled())
|
||||
self.assertAlmostEqual(0.1, loop.time())
|
||||
self.assertEqual(foo_running, False)
|
||||
|
||||
def test_wait_for_blocking(self):
|
||||
loop = self.new_test_loop()
|
||||
|
||||
async def coro():
|
||||
return 'done'
|
||||
|
||||
res = loop.run_until_complete(asyncio.wait_for(coro(), timeout=None))
|
||||
self.assertEqual(res, 'done')
|
||||
|
||||
def test_wait_for_race_condition(self):
|
||||
|
||||
def gen():
|
||||
yield 0.1
|
||||
yield 0.1
|
||||
yield 0.1
|
||||
|
||||
loop = self.new_test_loop(gen)
|
||||
|
||||
fut = self.new_future(loop)
|
||||
task = asyncio.wait_for(fut, timeout=0.2)
|
||||
loop.call_later(0.1, fut.set_result, "ok")
|
||||
res = loop.run_until_complete(task)
|
||||
self.assertEqual(res, "ok")
|
||||
|
||||
def test_wait_for_cancellation_race_condition(self):
|
||||
async def inner():
|
||||
with contextlib.suppress(asyncio.CancelledError):
|
||||
await asyncio.sleep(1)
|
||||
return 1
|
||||
|
||||
async def main():
|
||||
result = await asyncio.wait_for(inner(), timeout=.01)
|
||||
self.assertEqual(result, 1)
|
||||
|
||||
asyncio.run(main())
|
||||
|
||||
def test_wait_for_waits_for_task_cancellation(self):
|
||||
loop = asyncio.new_event_loop()
|
||||
self.addCleanup(loop.close)
|
||||
|
||||
task_done = False
|
||||
|
||||
async def foo():
|
||||
async def inner():
|
||||
nonlocal task_done
|
||||
try:
|
||||
await asyncio.sleep(0.2)
|
||||
except asyncio.CancelledError:
|
||||
await asyncio.sleep(_EPSILON)
|
||||
raise
|
||||
finally:
|
||||
task_done = True
|
||||
|
||||
inner_task = self.new_task(loop, inner())
|
||||
|
||||
await asyncio.wait_for(inner_task, timeout=_EPSILON)
|
||||
|
||||
with self.assertRaises(asyncio.TimeoutError) as cm:
|
||||
loop.run_until_complete(foo())
|
||||
|
||||
self.assertTrue(task_done)
|
||||
chained = cm.exception.__context__
|
||||
self.assertEqual(type(chained), asyncio.CancelledError)
|
||||
|
||||
def test_wait_for_waits_for_task_cancellation_w_timeout_0(self):
|
||||
loop = asyncio.new_event_loop()
|
||||
self.addCleanup(loop.close)
|
||||
|
||||
task_done = False
|
||||
|
||||
async def foo():
|
||||
async def inner():
|
||||
nonlocal task_done
|
||||
try:
|
||||
await asyncio.sleep(10)
|
||||
except asyncio.CancelledError:
|
||||
await asyncio.sleep(_EPSILON)
|
||||
raise
|
||||
finally:
|
||||
task_done = True
|
||||
|
||||
inner_task = self.new_task(loop, inner())
|
||||
await asyncio.sleep(_EPSILON)
|
||||
await asyncio.wait_for(inner_task, timeout=0)
|
||||
|
||||
with self.assertRaises(asyncio.TimeoutError) as cm:
|
||||
loop.run_until_complete(foo())
|
||||
|
||||
self.assertTrue(task_done)
|
||||
chained = cm.exception.__context__
|
||||
self.assertEqual(type(chained), asyncio.CancelledError)
|
||||
|
||||
def test_wait_for_reraises_exception_during_cancellation(self):
|
||||
loop = asyncio.new_event_loop()
|
||||
self.addCleanup(loop.close)
|
||||
|
||||
class FooException(Exception):
|
||||
pass
|
||||
|
||||
async def foo():
|
||||
async def inner():
|
||||
try:
|
||||
await asyncio.sleep(0.2)
|
||||
finally:
|
||||
raise FooException
|
||||
|
||||
inner_task = self.new_task(loop, inner())
|
||||
|
||||
await asyncio.wait_for(inner_task, timeout=_EPSILON)
|
||||
|
||||
with self.assertRaises(FooException):
|
||||
loop.run_until_complete(foo())
|
||||
|
||||
def test_wait_for_self_cancellation(self):
|
||||
loop = asyncio.new_event_loop()
|
||||
self.addCleanup(loop.close)
|
||||
|
||||
async def foo():
|
||||
async def inner():
|
||||
try:
|
||||
await asyncio.sleep(0.3)
|
||||
except asyncio.CancelledError:
|
||||
try:
|
||||
await asyncio.sleep(0.3)
|
||||
except asyncio.CancelledError:
|
||||
await asyncio.sleep(0.3)
|
||||
|
||||
return 42
|
||||
|
||||
inner_task = self.new_task(loop, inner())
|
||||
|
||||
wait = asyncio.wait_for(inner_task, timeout=0.1)
|
||||
|
||||
# Test that wait_for itself is properly cancellable
|
||||
# even when the initial task holds up the initial cancellation.
|
||||
task = self.new_task(loop, wait)
|
||||
await asyncio.sleep(0.2)
|
||||
task.cancel()
|
||||
|
||||
with self.assertRaises(asyncio.CancelledError):
|
||||
await task
|
||||
|
||||
self.assertEqual(await inner_task, 42)
|
||||
|
||||
loop.run_until_complete(foo())
|
||||
|
||||
def test_wait(self):
|
||||
|
||||
def gen():
|
||||
|
|
|
|||
270
Lib/test/test_asyncio/test_waitfor.py
Normal file
270
Lib/test/test_asyncio/test_waitfor.py
Normal file
|
|
@ -0,0 +1,270 @@
|
|||
import asyncio
|
||||
import unittest
|
||||
import time
|
||||
|
||||
|
||||
def tearDownModule():
|
||||
asyncio.set_event_loop_policy(None)
|
||||
|
||||
|
||||
# The following value can be used as a very small timeout:
|
||||
# it passes check "timeout > 0", but has almost
|
||||
# no effect on the test performance
|
||||
_EPSILON = 0.0001
|
||||
|
||||
|
||||
class SlowTask:
|
||||
""" Task will run for this defined time, ignoring cancel requests """
|
||||
TASK_TIMEOUT = 0.2
|
||||
|
||||
def __init__(self):
|
||||
self.exited = False
|
||||
|
||||
async def run(self):
|
||||
exitat = time.monotonic() + self.TASK_TIMEOUT
|
||||
|
||||
while True:
|
||||
tosleep = exitat - time.monotonic()
|
||||
if tosleep <= 0:
|
||||
break
|
||||
|
||||
try:
|
||||
await asyncio.sleep(tosleep)
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
self.exited = True
|
||||
|
||||
|
||||
class AsyncioWaitForTest(unittest.IsolatedAsyncioTestCase):
|
||||
|
||||
async def test_asyncio_wait_for_cancelled(self):
|
||||
t = SlowTask()
|
||||
|
||||
waitfortask = asyncio.create_task(
|
||||
asyncio.wait_for(t.run(), t.TASK_TIMEOUT * 2))
|
||||
await asyncio.sleep(0)
|
||||
waitfortask.cancel()
|
||||
await asyncio.wait({waitfortask})
|
||||
|
||||
self.assertTrue(t.exited)
|
||||
|
||||
async def test_asyncio_wait_for_timeout(self):
|
||||
t = SlowTask()
|
||||
|
||||
try:
|
||||
await asyncio.wait_for(t.run(), t.TASK_TIMEOUT / 2)
|
||||
except asyncio.TimeoutError:
|
||||
pass
|
||||
|
||||
self.assertTrue(t.exited)
|
||||
|
||||
async def test_wait_for_timeout_less_then_0_or_0_future_done(self):
|
||||
loop = asyncio.get_running_loop()
|
||||
|
||||
fut = loop.create_future()
|
||||
fut.set_result('done')
|
||||
|
||||
t0 = loop.time()
|
||||
ret = await asyncio.wait_for(fut, 0)
|
||||
t1 = loop.time()
|
||||
|
||||
self.assertEqual(ret, 'done')
|
||||
self.assertTrue(fut.done())
|
||||
self.assertLess(t1 - t0, 0.1)
|
||||
|
||||
async def test_wait_for_timeout_less_then_0_or_0_coroutine_do_not_started(self):
|
||||
loop = asyncio.get_running_loop()
|
||||
|
||||
foo_started = False
|
||||
|
||||
async def foo():
|
||||
nonlocal foo_started
|
||||
foo_started = True
|
||||
|
||||
with self.assertRaises(asyncio.TimeoutError):
|
||||
t0 = loop.time()
|
||||
await asyncio.wait_for(foo(), 0)
|
||||
t1 = loop.time()
|
||||
|
||||
self.assertEqual(foo_started, False)
|
||||
self.assertLess(t1 - t0, 0.1)
|
||||
|
||||
async def test_wait_for_timeout_less_then_0_or_0(self):
|
||||
loop = asyncio.get_running_loop()
|
||||
|
||||
for timeout in [0, -1]:
|
||||
with self.subTest(timeout=timeout):
|
||||
foo_running = None
|
||||
started = loop.create_future()
|
||||
|
||||
async def foo():
|
||||
nonlocal foo_running
|
||||
foo_running = True
|
||||
started.set_result(None)
|
||||
try:
|
||||
await asyncio.sleep(10)
|
||||
finally:
|
||||
foo_running = False
|
||||
return 'done'
|
||||
|
||||
fut = asyncio.create_task(foo())
|
||||
await started
|
||||
|
||||
with self.assertRaises(asyncio.TimeoutError):
|
||||
t0 = loop.time()
|
||||
await asyncio.wait_for(fut, timeout)
|
||||
t1 = loop.time()
|
||||
|
||||
self.assertTrue(fut.done())
|
||||
# it should have been cancelled due to the timeout
|
||||
self.assertTrue(fut.cancelled())
|
||||
self.assertEqual(foo_running, False)
|
||||
self.assertLess(t1 - t0, 0.1)
|
||||
|
||||
async def test_wait_for(self):
|
||||
loop = asyncio.get_running_loop()
|
||||
foo_running = None
|
||||
|
||||
async def foo():
|
||||
nonlocal foo_running
|
||||
foo_running = True
|
||||
try:
|
||||
await asyncio.sleep(10)
|
||||
finally:
|
||||
foo_running = False
|
||||
return 'done'
|
||||
|
||||
fut = asyncio.create_task(foo())
|
||||
|
||||
with self.assertRaises(asyncio.TimeoutError):
|
||||
t0 = loop.time()
|
||||
await asyncio.wait_for(fut, 0.1)
|
||||
t1 = loop.time()
|
||||
self.assertTrue(fut.done())
|
||||
# it should have been cancelled due to the timeout
|
||||
self.assertTrue(fut.cancelled())
|
||||
self.assertLess(t1 - t0, 0.2)
|
||||
self.assertEqual(foo_running, False)
|
||||
|
||||
async def test_wait_for_blocking(self):
|
||||
async def coro():
|
||||
return 'done'
|
||||
|
||||
res = await asyncio.wait_for(coro(), timeout=None)
|
||||
self.assertEqual(res, 'done')
|
||||
|
||||
async def test_wait_for_race_condition(self):
|
||||
loop = asyncio.get_running_loop()
|
||||
|
||||
fut = loop.create_future()
|
||||
task = asyncio.wait_for(fut, timeout=0.2)
|
||||
loop.call_later(0.1, fut.set_result, "ok")
|
||||
res = await task
|
||||
self.assertEqual(res, "ok")
|
||||
|
||||
async def test_wait_for_cancellation_race_condition(self):
|
||||
async def inner():
|
||||
with self.assertRaises(asyncio.CancelledError):
|
||||
await asyncio.sleep(1)
|
||||
return 1
|
||||
|
||||
result = await asyncio.wait_for(inner(), timeout=.01)
|
||||
self.assertEqual(result, 1)
|
||||
|
||||
async def test_wait_for_waits_for_task_cancellation(self):
|
||||
task_done = False
|
||||
|
||||
async def inner():
|
||||
nonlocal task_done
|
||||
try:
|
||||
await asyncio.sleep(10)
|
||||
except asyncio.CancelledError:
|
||||
await asyncio.sleep(_EPSILON)
|
||||
raise
|
||||
finally:
|
||||
task_done = True
|
||||
|
||||
inner_task = asyncio.create_task(inner())
|
||||
|
||||
with self.assertRaises(asyncio.TimeoutError) as cm:
|
||||
await asyncio.wait_for(inner_task, timeout=_EPSILON)
|
||||
|
||||
self.assertTrue(task_done)
|
||||
chained = cm.exception.__context__
|
||||
self.assertEqual(type(chained), asyncio.CancelledError)
|
||||
|
||||
async def test_wait_for_waits_for_task_cancellation_w_timeout_0(self):
|
||||
task_done = False
|
||||
|
||||
async def foo():
|
||||
async def inner():
|
||||
nonlocal task_done
|
||||
try:
|
||||
await asyncio.sleep(10)
|
||||
except asyncio.CancelledError:
|
||||
await asyncio.sleep(_EPSILON)
|
||||
raise
|
||||
finally:
|
||||
task_done = True
|
||||
|
||||
inner_task = asyncio.create_task(inner())
|
||||
await asyncio.sleep(_EPSILON)
|
||||
await asyncio.wait_for(inner_task, timeout=0)
|
||||
|
||||
with self.assertRaises(asyncio.TimeoutError) as cm:
|
||||
await foo()
|
||||
|
||||
self.assertTrue(task_done)
|
||||
chained = cm.exception.__context__
|
||||
self.assertEqual(type(chained), asyncio.CancelledError)
|
||||
|
||||
async def test_wait_for_reraises_exception_during_cancellation(self):
|
||||
class FooException(Exception):
|
||||
pass
|
||||
|
||||
async def foo():
|
||||
async def inner():
|
||||
try:
|
||||
await asyncio.sleep(0.2)
|
||||
finally:
|
||||
raise FooException
|
||||
|
||||
inner_task = asyncio.create_task(inner())
|
||||
|
||||
await asyncio.wait_for(inner_task, timeout=_EPSILON)
|
||||
|
||||
with self.assertRaises(FooException):
|
||||
await foo()
|
||||
|
||||
async def test_wait_for_self_cancellation(self):
|
||||
async def inner():
|
||||
try:
|
||||
await asyncio.sleep(0.3)
|
||||
except asyncio.CancelledError:
|
||||
try:
|
||||
await asyncio.sleep(0.3)
|
||||
except asyncio.CancelledError:
|
||||
await asyncio.sleep(0.3)
|
||||
|
||||
return 42
|
||||
|
||||
inner_task = asyncio.create_task(inner())
|
||||
|
||||
wait = asyncio.wait_for(inner_task, timeout=0.1)
|
||||
|
||||
# Test that wait_for itself is properly cancellable
|
||||
# even when the initial task holds up the initial cancellation.
|
||||
task = asyncio.create_task(wait)
|
||||
await asyncio.sleep(0.2)
|
||||
task.cancel()
|
||||
|
||||
with self.assertRaises(asyncio.CancelledError):
|
||||
await task
|
||||
|
||||
self.assertEqual(await inner_task, 42)
|
||||
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
Loading…
Add table
Add a link
Reference in a new issue