[Python-checkins] bpo-31250, test_asyncio: fix dangling threads (#3252)
Victor Stinner
webhook-mailer at python.org
Fri Sep 1 08:46:09 EDT 2017
https://github.com/python/cpython/commit/16432beadb8eba079c9786cc0c0eaacfd9fd2f7b
commit: 16432beadb8eba079c9786cc0c0eaacfd9fd2f7b
branch: master
author: Victor Stinner <victor.stinner at gmail.com>
committer: GitHub <noreply at github.com>
date: 2017-09-01T14:46:06+02:00
summary:
bpo-31250, test_asyncio: fix dangling threads (#3252)
* Explicitly call shutdown(wait=True) on executors to wait until all
threads complete to prevent side effects between tests.
* Fix test_loop_self_reading_exception(): don't mock loop.close().
Previously, the original close() method was called rather than the
mock, because how set_event_loop() registered loop.close().
files:
M Lib/asyncio/test_utils.py
M Lib/test/test_asyncio/test_futures.py
M Lib/test/test_asyncio/test_proactor_events.py
diff --git a/Lib/asyncio/test_utils.py b/Lib/asyncio/test_utils.py
index 15059501070..ecb3fca097b 100644
--- a/Lib/asyncio/test_utils.py
+++ b/Lib/asyncio/test_utils.py
@@ -437,12 +437,19 @@ def get_function_source(func):
class TestCase(unittest.TestCase):
+ @staticmethod
+ def close_loop(loop):
+ executor = loop._default_executor
+ if executor is not None:
+ executor.shutdown(wait=True)
+ loop.close()
+
def set_event_loop(self, loop, *, cleanup=True):
assert loop is not None
# ensure that the event loop is passed explicitly in asyncio
events.set_event_loop(None)
if cleanup:
- self.addCleanup(loop.close)
+ self.addCleanup(self.close_loop, loop)
def new_test_loop(self, gen=None):
loop = TestLoop(gen)
diff --git a/Lib/test/test_asyncio/test_futures.py b/Lib/test/test_asyncio/test_futures.py
index ebedfec7fa3..f8f614f1c30 100644
--- a/Lib/test/test_asyncio/test_futures.py
+++ b/Lib/test/test_asyncio/test_futures.py
@@ -380,6 +380,7 @@ def run(arg):
self.assertTrue(asyncio.isfuture(f2))
self.assertEqual(res, 'oi')
self.assertNotEqual(ident, threading.get_ident())
+ ex.shutdown(wait=True)
def test_wrap_future_future(self):
f1 = self._new_future(loop=self.loop)
@@ -395,6 +396,7 @@ def run(arg):
f1 = ex.submit(run, 'oi')
f2 = asyncio.wrap_future(f1)
self.assertIs(self.loop, f2._loop)
+ ex.shutdown(wait=True)
def test_wrap_future_cancel(self):
f1 = concurrent.futures.Future()
diff --git a/Lib/test/test_asyncio/test_proactor_events.py b/Lib/test/test_asyncio/test_proactor_events.py
index 4dfc61259f8..d76da661ab9 100644
--- a/Lib/test/test_asyncio/test_proactor_events.py
+++ b/Lib/test/test_asyncio/test_proactor_events.py
@@ -529,7 +529,6 @@ def test_loop_self_reading_fut(self):
self.loop._loop_self_reading)
def test_loop_self_reading_exception(self):
- self.loop.close = mock.Mock()
self.loop.call_exception_handler = mock.Mock()
self.proactor.recv.side_effect = OSError()
self.loop._loop_self_reading()
More information about the Python-checkins
mailing list