From fdacaa2b68e3bb06851bfb991ecd8ed063b49f3e Mon Sep 17 00:00:00 2001 From: Chris Withers Date: Mon, 3 Mar 2025 10:36:36 +0000 Subject: [PATCH] Add set_event_loop_policy to backports Since it's been deprecated in newer Python versions --- mock/backports.py | 6 ++++++ mock/tests/testasync.py | 4 ++-- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/mock/backports.py b/mock/backports.py index c268a3c..598a512 100644 --- 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 index 8a78efe..5bf0139 100644 --- 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: