diff options
author | Miss Islington (bot) <31488909+miss-islington@users.noreply.github.com> | 2020-12-16 17:57:23 (GMT) |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-12-16 17:57:23 (GMT) |
commit | d549d0b5575b390431b7b26999151f79f74d4516 (patch) | |
tree | 784ea43d0f92ea7a5a9557ecba62a9347ad8fee2 | |
parent | 0a24a57888798905e3b8891c59e61ed4f1bfc5a8 (diff) | |
download | cpython-d549d0b5575b390431b7b26999151f79f74d4516.zip cpython-d549d0b5575b390431b7b26999151f79f74d4516.tar.gz cpython-d549d0b5575b390431b7b26999151f79f74d4516.tar.bz2 |
bpo-39101: Fixes BaseException hang in IsolatedAsyncioTestCase. (GH-22654)
(cherry picked from commit 8374d2ee1589791be8892b00f4bbf8121dde24bd)
Co-authored-by: Lisa Roach <lisaroach14@gmail.com>
-rw-r--r-- | Lib/unittest/async_case.py | 4 | ||||
-rw-r--r-- | Lib/unittest/test/test_async_case.py | 27 | ||||
-rw-r--r-- | Misc/NEWS.d/next/Library/2020-10-11-21-43-03.bpo-39101.-I49Pm.rst | 1 |
3 files changed, 30 insertions, 2 deletions
diff --git a/Lib/unittest/async_case.py b/Lib/unittest/async_case.py index 1bc1312..520213c 100644 --- a/Lib/unittest/async_case.py +++ b/Lib/unittest/async_case.py @@ -102,9 +102,9 @@ class IsolatedAsyncioTestCase(TestCase): ret = await awaitable if not fut.cancelled(): fut.set_result(ret) - except asyncio.CancelledError: + except (SystemExit, KeyboardInterrupt): raise - except Exception as ex: + except (BaseException, asyncio.CancelledError) as ex: if not fut.cancelled(): fut.set_exception(ex) diff --git a/Lib/unittest/test/test_async_case.py b/Lib/unittest/test/test_async_case.py index 2db441d..d01864b 100644 --- a/Lib/unittest/test/test_async_case.py +++ b/Lib/unittest/test/test_async_case.py @@ -190,6 +190,33 @@ class TestAsyncCase(unittest.TestCase): 'async_cleanup 2', 'sync_cleanup 1']) + def test_base_exception_from_async_method(self): + events = [] + class Test(unittest.IsolatedAsyncioTestCase): + async def test_base(self): + events.append("test_base") + raise BaseException() + events.append("not it") + + async def test_no_err(self): + events.append("test_no_err") + + async def test_cancel(self): + raise asyncio.CancelledError() + + test = Test("test_base") + output = test.run() + self.assertFalse(output.wasSuccessful()) + + test = Test("test_no_err") + test.run() + self.assertEqual(events, ['test_base', 'test_no_err']) + + test = Test("test_cancel") + output = test.run() + self.assertFalse(output.wasSuccessful()) + + if __name__ == "__main__": unittest.main() diff --git a/Misc/NEWS.d/next/Library/2020-10-11-21-43-03.bpo-39101.-I49Pm.rst b/Misc/NEWS.d/next/Library/2020-10-11-21-43-03.bpo-39101.-I49Pm.rst new file mode 100644 index 0000000..a571e83 --- /dev/null +++ b/Misc/NEWS.d/next/Library/2020-10-11-21-43-03.bpo-39101.-I49Pm.rst @@ -0,0 +1 @@ +Fixed tests using IsolatedAsyncioTestCase from hanging on BaseExceptions.
\ No newline at end of file |