summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorDong-hee Na <donghee.na@python.org>2021-05-06 23:10:52 +0900
committerGitHub <noreply@github.com>2021-05-06 23:10:52 +0900
commit698e9a8211c46ed5dc93e5cd7026ea05dec2f373 (patch)
treee29b8550b52183a513e3ce536cd1ee13481f3a80
parent985ac016373403e8ad41f8d563c4355ffa8d49ff (diff)
downloadcpython-git-698e9a8211c46ed5dc93e5cd7026ea05dec2f373.tar.gz
bpo-44017: Update test_contextlib_async not to emit DeprecationWarn (GH-25918)
-rw-r--r--Lib/test/test_contextlib_async.py12
-rw-r--r--Lib/unittest/test/testmock/testasync.py3
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)