Move AsyncContextManagerMock to utils package for future re-use

Signed-off-by: dannosaur <461956+dannosaur@users.noreply.github.com>

test_opensearchpy/test_async_http_connection.py

@@ -28,23 +28,12 @@
 from unittest import IsolatedAsyncioTestCase
 
 import pytest
-from asynctest import CoroutineMock, MagicMock, patch
+from asynctest import patch
 
 from opensearchpy import AsyncHttpConnection
 from opensearchpy._async._extra_imports import aiohttp
 from opensearchpy._async.compat import get_running_loop
-
-
-class AsyncContextManagerMock(MagicMock):
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
-        type(self).__aenter__ = CoroutineMock(
-            return_value=MagicMock(
-                text=CoroutineMock(return_value='test'),
-                status=200,
-            )
-        )
-        type(self).__aexit__ = CoroutineMock(return_value=MagicMock())
+from test_opensearchpy.utils import AsyncContextManagerMock
 
 
 @pytest.mark.asyncio

test_opensearchpy/utils.py

@@ -27,6 +27,8 @@
 
 import time
 
+from asynctest import CoroutineMock, MagicMock
+
 from opensearchpy import OpenSearch
 
 
@@ -175,3 +177,15 @@ def wait_for_cluster_state_updates_to_finish(client, timeout=30):
     while time.time() < end_time:
         if not client.cluster.pending_tasks().get("tasks", ()):
             break
+
+
+class AsyncContextManagerMock(MagicMock):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        type(self).__aenter__ = CoroutineMock(
+            return_value=MagicMock(
+                text=CoroutineMock(return_value='test'),
+                status=200,
+            )
+        )
+        type(self).__aexit__ = CoroutineMock(return_value=MagicMock())