add support.set_event_loop_policy for setting the event loop policy in tests

Author: graingert

Lib/test/libregrtest/save_env.py

@@ -97,7 +97,7 @@ def get_asyncio_events__event_loop_policy(self):
         return support.maybe_get_event_loop_policy()
     def restore_asyncio_events__event_loop_policy(self, policy):
         asyncio = self.get_module('asyncio')
-        asyncio.set_event_loop_policy(policy)
+        support.set_event_loop_policy(policy)
 
     def get_sys_argv(self):
         return id(sys.argv), sys.argv, sys.argv[:]

Lib/test/support/__init__.py

@@ -2089,6 +2089,13 @@ def maybe_get_event_loop_policy():
     import asyncio.events
     return asyncio.events._event_loop_policy
 
+def set_event_loop_policy(policy):
+    """Set the global event loop policy ignoring deprecation warnings"""
+    import asyncio
+    with warnings.catch_warnings():
+        warnings.simplefilter("ignore", DeprecationWarning)
+        asyncio.set_event_loop_policy(policy)
+
 # Helpers for testing hashing.
 NHASHBITS = sys.hash_info.width # number of bits in hash() result
 assert NHASHBITS in (32, 64)

Lib/test/test_asyncgen.py

@@ -5,6 +5,7 @@
 
 from test.support.import_helper import import_module
 from test.support import gc_collect, requires_working_socket
+from test import support
 asyncio = import_module("asyncio")
 
 
@@ -429,7 +430,7 @@ def setUp(self):
     def tearDown(self):
         self.loop.close()
         self.loop = None
-        asyncio.set_event_loop_policy(None)
+        support.set_event_loop_policy(None)
 
     def check_async_iterator_anext(self, ait_class):
         with self.subTest(anext="pure-Python"):

Lib/test/test_asyncio/test_base_events.py

@@ -24,7 +24,7 @@
 
 
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 def mock_socket_module():

Lib/test/test_asyncio/test_buffered_proto.py

@@ -2,10 +2,11 @@
 import unittest
 
 from test.test_asyncio import functional as func_tests
+from test import support
 
 
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 class ReceiveStuffProto(asyncio.BufferedProtocol):

Lib/test/test_asyncio/test_context.py

@@ -1,10 +1,11 @@
 import asyncio
 import decimal
 import unittest
+from test import support
 
 
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 @unittest.skipUnless(decimal.HAVE_CONTEXTVAR, "decimal is built with a thread-local context")

Lib/test/test_asyncio/test_eager_task_factory.py

@@ -8,12 +8,13 @@
 from asyncio import tasks
 from test.test_asyncio import utils as test_utils
 from test.support.script_helper import assert_python_ok
+from test import support
 
 MOCK_ANY = mock.ANY
 
 
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 class EagerTaskFactoryLoopTests:

Lib/test/test_asyncio/test_events.py

@@ -39,7 +39,7 @@
 
 
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 def broken_unix_getsockname():
@@ -2684,8 +2684,9 @@ def test_get_event_loop_policy(self):
         self.assertIs(policy, asyncio.get_event_loop_policy())
 
     def test_set_event_loop_policy(self):
-        self.assertRaises(
-            TypeError, asyncio.set_event_loop_policy, object())
+        with self.assertWarns(DeprecationWarning):
+            self.assertRaises(
+                TypeError, asyncio.set_event_loop_policy, object())
 
         old_policy = asyncio.get_event_loop_policy()
 
@@ -2790,7 +2791,8 @@ def get_event_loop(self):
 
         old_policy = asyncio.get_event_loop_policy()
         try:
-            asyncio.set_event_loop_policy(Policy())
+            with self.assertWarns(DeprecationWarning):
+                asyncio.set_event_loop_policy(Policy())
             loop = asyncio.new_event_loop()
 
             with self.assertRaises(TestError):
@@ -2818,7 +2820,7 @@ async def func():
                 asyncio.get_event_loop()
 
         finally:
-            asyncio.set_event_loop_policy(old_policy)
+            support.set_event_loop_policy(old_policy)
             if loop is not None:
                 loop.close()
 
@@ -2830,7 +2832,8 @@ async def func():
     def test_get_event_loop_returns_running_loop2(self):
         old_policy = asyncio.get_event_loop_policy()
         try:
-            asyncio.set_event_loop_policy(asyncio.DefaultEventLoopPolicy())
+            with self.assertWarns(DeprecationWarning):
+                asyncio.set_event_loop_policy(asyncio.DefaultEventLoopPolicy())
             loop = asyncio.new_event_loop()
             self.addCleanup(loop.close)
 
@@ -2861,7 +2864,7 @@ async def func():
                 asyncio.get_event_loop()
 
         finally:
-            asyncio.set_event_loop_policy(old_policy)
+            support.set_event_loop_policy(old_policy)
             if loop is not None:
                 loop.close()
 

Lib/test/test_asyncio/test_futures.py

@@ -16,7 +16,7 @@
 
 
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 def _fakefunc(f):

Lib/test/test_asyncio/test_futures2.py

@@ -4,10 +4,11 @@
 import traceback
 import unittest
 from asyncio import tasks
+from test import support
 
 
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 class FutureTests:

Lib/test/test_asyncio/test_locks.py

@@ -6,6 +6,7 @@
 
 import asyncio
 import collections
+from test import support
 
 STR_RGX_REPR = (
     r'^<(?P<class>.*?) object at (?P<address>.*?)'
@@ -20,7 +21,7 @@
 
 
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 class LockTests(unittest.IsolatedAsyncioTestCase):

Lib/test/test_asyncio/test_pep492.py

@@ -8,10 +8,11 @@
 
 import asyncio
 from test.test_asyncio import utils as test_utils
+from test import support
 
 
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 # Test that asyncio.iscoroutine() uses collections.abc.Coroutine

Lib/test/test_asyncio/test_proactor_events.py

@@ -15,10 +15,11 @@
 from test.support import os_helper
 from test.support import socket_helper
 from test.test_asyncio import utils as test_utils
+from test import support
 
 
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 def close_transport(transport):

Lib/test/test_asyncio/test_protocols.py

@@ -2,12 +2,13 @@
 from unittest import mock
 
 import asyncio
+from test import support
 
 
 def tearDownModule():
     # not needed for the test file but added for uniformness with all other
     # asyncio test files for the sake of unified cleanup
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 class ProtocolsAbsTests(unittest.TestCase):

Lib/test/test_asyncio/test_queues.py

@@ -3,10 +3,11 @@
 import asyncio
 import unittest
 from types import GenericAlias
+from test import support
 
 
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 class QueueBasicTests(unittest.IsolatedAsyncioTestCase):

Lib/test/test_asyncio/test_runners.py

@@ -6,12 +6,13 @@
 import threading
 import unittest
 from test.test_asyncio import utils as test_utils
+from test import support
 from unittest import mock
 from unittest.mock import patch
 
 
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 def interrupt_self():
@@ -60,15 +61,15 @@ def setUp(self):
         super().setUp()
 
         policy = TestPolicy(self.new_loop)
-        asyncio.set_event_loop_policy(policy)
+        support.set_event_loop_policy(policy)
 
     def tearDown(self):
         policy = asyncio.get_event_loop_policy()
         if policy.loop is not None:
             self.assertTrue(policy.loop.is_closed())
             self.assertTrue(policy.loop.shutdown_ag_run)
 
-        asyncio.set_event_loop_policy(None)
+        support.set_event_loop_policy(None)
         super().tearDown()
 
 
@@ -258,7 +259,7 @@ def new_event_loop():
             loop.set_task_factory(Task)
             return loop
 
-        asyncio.set_event_loop_policy(TestPolicy(new_event_loop))
+        support.set_event_loop_policy(TestPolicy(new_event_loop))
         with self.assertRaises(asyncio.CancelledError):
             asyncio.run(main())
 

Lib/test/test_asyncio/test_selector_events.py

@@ -19,12 +19,13 @@
                                      _SelectorSocketTransport,
                                      _SelectorTransport)
 from test.test_asyncio import utils as test_utils
+from test import support
 
 MOCK_ANY = mock.ANY
 
 
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 class TestBaseSelectorEventLoop(BaseSelectorEventLoop):

Lib/test/test_asyncio/test_sendfile.py

@@ -22,7 +22,7 @@
 
 
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 class MySendfileProto(asyncio.Protocol):

Lib/test/test_asyncio/test_server.py

@@ -6,10 +6,11 @@
 from test.support import socket_helper
 from test.test_asyncio import utils as test_utils
 from test.test_asyncio import functional as func_tests
+from test import support
 
 
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 class BaseStartServer(func_tests.FunctionalTestCaseMixin):

Lib/test/test_asyncio/test_sock_lowlevel.py

@@ -15,7 +15,7 @@
 
 
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 class MyProto(asyncio.Protocol):

Lib/test/test_asyncio/test_ssl.py

@@ -25,7 +25,7 @@
 
 
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 class MyBaseProto(asyncio.Protocol):

Lib/test/test_asyncio/test_sslproto.py

@@ -21,7 +21,7 @@
 
 
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 @unittest.skipIf(ssl is None, 'No ssl module')

Lib/test/test_asyncio/test_streams.py

@@ -18,10 +18,11 @@
 
 import asyncio
 from test.test_asyncio import utils as test_utils
+from test import support
 
 
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 class StreamTests(test_utils.TestCase):

Lib/test/test_asyncio/test_subprocess.py

@@ -35,7 +35,7 @@
 
 
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 class TestSubprocessTransport(base_subprocess.BaseSubprocessTransport):

Lib/test/test_asyncio/test_taskgroups.py

@@ -7,11 +7,12 @@
 import contextlib
 from asyncio import taskgroups
 import unittest
+from test import support
 
 
 # To prevent a warning "test altered the execution environment"
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 class MyExc(Exception):

Lib/test/test_asyncio/test_tasks.py

@@ -22,7 +22,7 @@
 
 
 def tearDownModule():
-    asyncio.set_event_loop_policy(None)
+    support.set_event_loop_policy(None)
 
 
 async def coroutine_function():