Create an AsyncContextManagerMock mock for testing asynchronous context managers

It can be used to test aiohttp request functions, since they are async context managers

1 files changed, 12 insertions, 0 deletions

diff --git a/tests/helpers.py b/tests/helpers.py
index 5df796c23..6aee8623f 100644
--- a/tests/helpers.py
+++ b/tests/helpers.py
@@ -127,6 +127,18 @@ class AsyncMock(CustomMockMixin, unittest.mock.MagicMock):
         return super().__call__(*args, **kwargs)
 
 
+class AsyncContextManagerMock(unittest.mock.MagicMock):
+    def __init__(self, return_value: Any):
+        super().__init__()
+        self._return_value = return_value
+
+    async def __aenter__(self):
+        return self._return_value
+
+    async def __aexit__(self, *args):
+        pass
+
+
 class AsyncIteratorMock:
     """
     A class to mock asynchronous iterators.