PYTHON-4673 - Add Async Encryption Tests (mongodb#1818)

Author: NoahStapp

pymongo/asynchronous/encryption.py

@@ -50,7 +50,7 @@
     _HAVE_PYMONGOCRYPT = True
 except ImportError:
     _HAVE_PYMONGOCRYPT = False
-    MongoCryptCallback = object
+    AsyncMongoCryptCallback = object
 
 from bson import _dict_to_bson, decode, encode
 from bson.binary import STANDARD, UUID_SUBTYPE, Binary
@@ -207,10 +207,10 @@ async def collection_info(
 
         :return: The first document from the listCollections command response as BSON.
         """
-        async with self.client_ref()[database].list_collections(
+        async with await self.client_ref()[database].list_collections(
             filter=RawBSONDocument(filter)
         ) as cursor:
-            for doc in cursor:
+            async for doc in cursor:
                 return _dict_to_bson(doc, False, _DATA_KEY_OPTS)
             return None
 

pymongo/asynchronous/network.py

@@ -297,7 +297,7 @@ async def command(
         )
 
     if client and client._encrypter and reply:
-        decrypted = client._encrypter.decrypt(reply.raw_command_response())
+        decrypted = await client._encrypter.decrypt(reply.raw_command_response())
         response_doc = cast(
             "_DocumentOut", _decode_all_selective(decrypted, codec_options, user_fields)[0]
         )

pymongo/asynchronous/server.py

@@ -309,7 +309,7 @@ async def run_operation(
         client = operation.client  # type: ignore[assignment]
         if client and client._encrypter:
             if use_cmd:
-                decrypted = client._encrypter.decrypt(reply.raw_command_response())
+                decrypted = await client._encrypter.decrypt(reply.raw_command_response())
                 docs = _decode_all_selective(decrypted, operation.codec_options, user_fields)
 
         response: Response

pymongo/synchronous/mongo_client.py

@@ -1538,7 +1538,7 @@ def close(self) -> None:
 
     if not _IS_SYNC:
         # Add support for contextlib.closing.
-        aclose = close
+        close = close
 
     def _get_topology(self) -> Topology:
         """Get the internal :class:`~pymongo.topology.Topology` object.