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.