1 files changed, 12 insertions, 0 deletions
diff --git a/tests/helpers.py b/tests/helpers.py
index 9d9dd5da6..6f50f6ae3 100644
--- a/tests/helpers.py
+++ b/tests/helpers.py
@@ -128,6 +128,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.