Add set_event_loop_policy to backports

Since it's been deprecated in newer Python versions
testing-cabal · Mar 3, 2025 · fdacaa2 · fdacaa2
1 parent 4221c21
commit fdacaa2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/mock/backports.py b/mock/backports.py
@@ -90,3 +90,9 @@ def run(self, result=None):
                 return super().run(result)
             finally:
                 self._tearDownAsyncioLoop()
+
+
+try:
+    from asyncio import _set_event_loop_policy as set_event_loop_policy
+except ImportError:
+    from asyncio import set_event_loop_policy
diff --git a/mock/tests/testasync.py b/mock/tests/testasync.py
@@ -8,7 +8,7 @@
 
 from mock import (ANY, call, AsyncMock, patch, MagicMock, Mock,
                   create_autospec, sentinel, seal)
-from mock.backports import IsolatedAsyncioTestCase, iscoroutinefunction
+from mock.backports import IsolatedAsyncioTestCase, iscoroutinefunction, set_event_loop_policy
 from mock.mock import _CallList
 
 
@@ -25,7 +25,7 @@ def run(main):
 
 
 def tearDownModule():
-    asyncio._set_event_loop_policy(None)
+    set_event_loop_policy(None)
 
 
 class AsyncClass: