Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

gh-128550: TaskGroup: cancel tasks added to set after aborting #128551

Closed
wants to merge 13 commits into from
Closed
5 changes: 5 additions & 0 deletions Lib/asyncio/taskgroups.py
Original file line number Diff line number Diff line change
Expand Up @@ -205,6 +205,11 @@ def create_task(self, coro, *, name=None, context=None):
else:
self._tasks.add(task)
task.add_done_callback(self._on_task_done)
if self._aborting:
graingert marked this conversation as resolved.
Show resolved Hide resolved
# gh-128550: if this task is eager it might have started
# another eager task that aborts us, if so we must cancel
# this task.
task.cancel()
return task

# Since Python 3.8 Tasks propagate all exceptions correctly,
Expand Down
24 changes: 24 additions & 0 deletions Lib/test/test_asyncio/test_taskgroups.py
Original file line number Diff line number Diff line change
Expand Up @@ -997,6 +997,30 @@ class MyKeyboardInterrupt(KeyboardInterrupt):
self.assertIsNotNone(exc)
self.assertListEqual(gc.get_referrers(exc), no_other_refs())

async def test_cancels_task_if_created_during_creation(self):
graingert marked this conversation as resolved.
Show resolved Hide resolved
graingert marked this conversation as resolved.
Show resolved Hide resolved
ran = False
class MyError(Exception):
pass

try:
graingert marked this conversation as resolved.
Show resolved Hide resolved
async with asyncio.TaskGroup() as tg:
async def third_task():
raise MyError("third task failed")

async def second_task():
nonlocal ran
tg.create_task(third_task())
with self.assertRaises(asyncio.CancelledError):
await asyncio.sleep(0) # eager tasks cancel here
await asyncio.sleep(0) # lazy tasks cancel here
ran = True

tg.create_task(second_task())
except* MyError as excs:
exc = excs.exceptions[0]
graingert marked this conversation as resolved.
Show resolved Hide resolved

self.assertIsInstance(exc, MyError)
graingert marked this conversation as resolved.
Show resolved Hide resolved
self.assertTrue(ran)
graingert marked this conversation as resolved.
Show resolved Hide resolved
graingert marked this conversation as resolved.
Show resolved Hide resolved

if __name__ == "__main__":
unittest.main()
Loading