summaryrefslogtreecommitdiffstats
path: root/Lib
diff options
context:
space:
mode:
authorVictor Stinner <victor.stinner@gmail.com>2014-10-11 12:35:33 (GMT)
committerVictor Stinner <victor.stinner@gmail.com>2014-10-11 12:35:33 (GMT)
commit2de8a6980ac544ed9c10d609d7fc4d93eca2595f (patch)
tree53d912b279e687681e0dc21794115d5e3046fe7f /Lib
parent86a89f02e20ce141df9b3d473c704e31f608e802 (diff)
parentc8bd53f81512acf805708abca31f5491c13ae185 (diff)
downloadcpython-2de8a6980ac544ed9c10d609d7fc4d93eca2595f.zip
cpython-2de8a6980ac544ed9c10d609d7fc4d93eca2595f.tar.gz
cpython-2de8a6980ac544ed9c10d609d7fc4d93eca2595f.tar.bz2
(Merge 3.4) Issue #22601: run_forever() now consumes BaseException of the
temporary task If the coroutine raised a BaseException, consume the exception to not log a warning. The caller doesn't have access to the local task.
Diffstat (limited to 'Lib')
-rw-r--r--Lib/asyncio/base_events.py10
-rw-r--r--Lib/test/test_asyncio/test_base_events.py22
2 files changed, 30 insertions, 2 deletions
diff --git a/Lib/asyncio/base_events.py b/Lib/asyncio/base_events.py
index 3cff72a..b6b7123 100644
--- a/Lib/asyncio/base_events.py
+++ b/Lib/asyncio/base_events.py
@@ -268,7 +268,15 @@ class BaseEventLoop(events.AbstractEventLoop):
future._log_destroy_pending = False
future.add_done_callback(_raise_stop_error)
- self.run_forever()
+ try:
+ self.run_forever()
+ except:
+ if new_task and future.done() and not future.cancelled():
+ # The coroutine raised a BaseException. Consume the exception
+ # to not log a warning, the caller doesn't have access to the
+ # local task.
+ future.exception()
+ raise
future.remove_done_callback(_raise_stop_error)
if not future.done():
raise RuntimeError('Event loop stopped before Future completed.')
diff --git a/Lib/test/test_asyncio/test_base_events.py b/Lib/test/test_asyncio/test_base_events.py
index 294872a..afc448c 100644
--- a/Lib/test/test_asyncio/test_base_events.py
+++ b/Lib/test/test_asyncio/test_base_events.py
@@ -9,7 +9,7 @@ import time
import unittest
from unittest import mock
from test.script_helper import assert_python_ok
-from test.support import IPV6_ENABLED
+from test.support import IPV6_ENABLED, gc_collect
import asyncio
from asyncio import base_events
@@ -618,6 +618,26 @@ class BaseEventLoopTests(test_utils.TestCase):
task._log_destroy_pending = False
coro.close()
+ def test_run_forever_keyboard_interrupt(self):
+ # Python issue #22601: ensure that the temporary task created by
+ # run_forever() consumes the KeyboardInterrupt and so don't log
+ # a warning
+ @asyncio.coroutine
+ def raise_keyboard_interrupt():
+ raise KeyboardInterrupt
+
+ self.loop._process_events = mock.Mock()
+ self.loop.call_exception_handler = mock.Mock()
+
+ try:
+ self.loop.run_until_complete(raise_keyboard_interrupt())
+ except KeyboardInterrupt:
+ pass
+ self.loop.close()
+ gc_collect()
+
+ self.assertFalse(self.loop.call_exception_handler.called)
+
class MyProto(asyncio.Protocol):
done = None