PYTHON-4533 - Convert test/test_transactions.py to async (mongodb#1732)

Author: NoahStapp

gridfs/asynchronous/grid_file.py

@@ -154,7 +154,7 @@ async def put(self, data: Any, **kwargs: Any) -> Any:
         """
         async with AsyncGridIn(self._collection, **kwargs) as grid_file:
             await grid_file.write(data)
-            return await grid_file._id
+            return grid_file._id
 
     async def get(self, file_id: Any, session: Optional[AsyncClientSession] = None) -> AsyncGridOut:
         """Get a file from GridFS by ``"_id"``.

pymongo/asynchronous/client_session.py

@@ -144,6 +144,7 @@
     Any,
     AsyncContextManager,
     Callable,
+    Coroutine,
     Mapping,
     MutableMapping,
     NoReturn,
@@ -598,7 +599,7 @@ def _inherit_option(self, name: str, val: _T) -> _T:
 
     async def with_transaction(
         self,
-        callback: Callable[[AsyncClientSession], _T],
+        callback: Callable[[AsyncClientSession], Coroutine[Any, Any, _T]],
         read_concern: Optional[ReadConcern] = None,
         write_concern: Optional[WriteConcern] = None,
         read_preference: Optional[_ServerMode] = None,
@@ -693,7 +694,7 @@ async def callback(session, custom_arg, custom_kwarg=None):
                 read_concern, write_concern, read_preference, max_commit_time_ms
             )
             try:
-                ret = callback(self)
+                ret = await callback(self)
             except Exception as exc:
                 if self.in_transaction:
                     await self.abort_transaction()