diff options
author | Dong-hee Na <donghee.na@python.org> | 2021-05-06 23:10:52 +0900 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-05-06 23:10:52 +0900 |
commit | 698e9a8211c46ed5dc93e5cd7026ea05dec2f373 (patch) | |
tree | e29b8550b52183a513e3ce536cd1ee13481f3a80 | |
parent | 985ac016373403e8ad41f8d563c4355ffa8d49ff (diff) | |
download | cpython-git-698e9a8211c46ed5dc93e5cd7026ea05dec2f373.tar.gz |
bpo-44017: Update test_contextlib_async not to emit DeprecationWarn (GH-25918)
-rw-r--r-- | Lib/test/test_contextlib_async.py | 12 | ||||
-rw-r--r-- | Lib/unittest/test/testmock/testasync.py | 3 |
2 files changed, 6 insertions, 9 deletions
diff --git a/Lib/test/test_contextlib_async.py b/Lib/test/test_contextlib_async.py index 290ef05b82..cbc82dfd8f 100644 --- a/Lib/test/test_contextlib_async.py +++ b/Lib/test/test_contextlib_async.py @@ -369,16 +369,14 @@ class TestAsyncExitStack(TestBaseExitStack, unittest.TestCase): class SyncAsyncExitStack(AsyncExitStack): @staticmethod def run_coroutine(coro): - loop = asyncio.get_event_loop() - - f = asyncio.ensure_future(coro) - f.add_done_callback(lambda f: loop.stop()) + loop = asyncio.get_event_loop_policy().get_event_loop() + t = loop.create_task(coro) + t.add_done_callback(lambda f: loop.stop()) loop.run_forever() - exc = f.exception() - + exc = t.exception() if not exc: - return f.result() + return t.result() else: context = exc.__context__ diff --git a/Lib/unittest/test/testmock/testasync.py b/Lib/unittest/test/testmock/testasync.py index e1866a3492..122e695663 100644 --- a/Lib/unittest/test/testmock/testasync.py +++ b/Lib/unittest/test/testmock/testasync.py @@ -173,8 +173,7 @@ class AsyncMockTest(unittest.TestCase): def test_future_isfuture(self): loop = asyncio.new_event_loop() - asyncio.set_event_loop(loop) - fut = asyncio.Future() + fut = loop.create_future() loop.stop() loop.close() mock = AsyncMock(fut) |