diff options
author | Yury Selivanov <yury@magic.io> | 2016-10-09 16:20:44 (GMT) |
---|---|---|
committer | Yury Selivanov <yury@magic.io> | 2016-10-09 16:20:44 (GMT) |
commit | 917c1c3ee1b4cbe7a7a78ffde053fb5628eee486 (patch) | |
tree | 6ef83f2beb72efd8e68ab67a996eb11b54492aa0 | |
parent | 88e8aca78d71fa917a057e8d60b24f4a364d3248 (diff) | |
parent | 4145c8380699f7af14ae15179e01652e25f0d102 (diff) | |
download | cpython-917c1c3ee1b4cbe7a7a78ffde053fb5628eee486.zip cpython-917c1c3ee1b4cbe7a7a78ffde053fb5628eee486.tar.gz cpython-917c1c3ee1b4cbe7a7a78ffde053fb5628eee486.tar.bz2 |
Merge 3.5 (issue #27972)
-rw-r--r-- | Lib/asyncio/tasks.py | 21 | ||||
-rw-r--r-- | Lib/test/test_asyncio/test_tasks.py | 11 | ||||
-rw-r--r-- | Misc/NEWS | 2 |
3 files changed, 27 insertions, 7 deletions
diff --git a/Lib/asyncio/tasks.py b/Lib/asyncio/tasks.py index f735b44..14949d1 100644 --- a/Lib/asyncio/tasks.py +++ b/Lib/asyncio/tasks.py @@ -241,7 +241,7 @@ class Task(futures.Future): result = coro.throw(exc) except StopIteration as exc: self.set_result(exc.value) - except futures.CancelledError as exc: + except futures.CancelledError: super().cancel() # I.e., Future.cancel(self). except Exception as exc: self.set_exception(exc) @@ -259,12 +259,19 @@ class Task(futures.Future): 'Task {!r} got Future {!r} attached to a ' 'different loop'.format(self, result))) elif blocking: - result._asyncio_future_blocking = False - result.add_done_callback(self._wakeup) - self._fut_waiter = result - if self._must_cancel: - if self._fut_waiter.cancel(): - self._must_cancel = False + if result is self: + self._loop.call_soon( + self._step, + RuntimeError( + 'Task cannot await on itself: {!r}'.format( + self))) + else: + result._asyncio_future_blocking = False + result.add_done_callback(self._wakeup) + self._fut_waiter = result + if self._must_cancel: + if self._fut_waiter.cancel(): + self._must_cancel = False else: self._loop.call_soon( self._step, diff --git a/Lib/test/test_asyncio/test_tasks.py b/Lib/test/test_asyncio/test_tasks.py index 2863c42..a5af7d1 100644 --- a/Lib/test/test_asyncio/test_tasks.py +++ b/Lib/test/test_asyncio/test_tasks.py @@ -92,6 +92,17 @@ class TaskTests(test_utils.TestCase): finally: other_loop.close() + def test_task_awaits_on_itself(self): + @asyncio.coroutine + def test(): + yield from task + + task = asyncio.ensure_future(test(), loop=self.loop) + + with self.assertRaisesRegex(RuntimeError, + 'Task cannot await on itself'): + self.loop.run_until_complete(task) + def test_task_class(self): @asyncio.coroutine def notmuch(): @@ -219,6 +219,8 @@ Library - Issue #28399: Remove UNIX socket from FS before binding. Patch by Коренберг Марк. +- Issue #27972: Prohibit Tasks to await on themselves. + Windows ------- |