From 6ea2b8fc8be8725e1553fbbb159ea07806e9f41a Mon Sep 17 00:00:00 2001 From: Yury Selivanov Date: Mon, 7 Nov 2016 19:00:46 -0500 Subject: [PATCH] Issue #28613: Expose asyncio._get_running_loop() and _set_running_loop() --- Lib/asyncio/events.py | 1 + Lib/test/test_asyncio/test_events.py | 4 ++++ 2 files changed, 5 insertions(+) diff --git a/Lib/asyncio/events.py b/Lib/asyncio/events.py index 8575e2c1c40..28a45fc3cc5 100644 --- a/Lib/asyncio/events.py +++ b/Lib/asyncio/events.py @@ -6,6 +6,7 @@ __all__ = ['AbstractEventLoopPolicy', 'get_event_loop_policy', 'set_event_loop_policy', 'get_event_loop', 'set_event_loop', 'new_event_loop', 'get_child_watcher', 'set_child_watcher', + '_set_running_loop', '_get_running_loop', ] import functools diff --git a/Lib/test/test_asyncio/test_events.py b/Lib/test/test_asyncio/test_events.py index 4c18300bab2..d6489d5f8d1 100644 --- a/Lib/test/test_asyncio/test_events.py +++ b/Lib/test/test_asyncio/test_events.py @@ -2732,9 +2732,11 @@ class PolicyTests(unittest.TestCase): try: asyncio.set_event_loop_policy(Policy()) loop = asyncio.new_event_loop() + self.assertIs(asyncio._get_running_loop(), None) async def func(): self.assertIs(asyncio.get_event_loop(), loop) + self.assertIs(asyncio._get_running_loop(), loop) loop.run_until_complete(func()) finally: @@ -2742,6 +2744,8 @@ class PolicyTests(unittest.TestCase): if loop is not None: loop.close() + self.assertIs(asyncio._get_running_loop(), None) + if __name__ == '__main__': unittest.main()