fix timed out tests pt4

mongodb · Sep 13, 2024 · 7d04121 · 7d04121
1 parent fa3207c
commit 7d04121
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/test/asynchronous/test_change_stream.py b/test/asynchronous/test_change_stream.py
@@ -103,11 +103,11 @@ async def get_resume_token(self, invalidate=False):
                 if isinstance(cs._target, AsyncMongoClient):
                     self.skipTest("cluster-level change streams cannot be invalidated")
                 self.generate_invalidate_event(cs)
-                return cs.next()["_id"]
+                return (await cs.next())["_id"]
         else:
             async with await self.change_stream() as cs:
                 await coll.insert_one({"data": 1})
-                return cs.next()["_id"]
+                return (await cs.next())["_id"]
 
     async def get_start_at_operation_time(self):
         """Get an operationTime. Advances the operation clock beyond the most

diff --git a/test/test_change_stream.py b/test/test_change_stream.py
@@ -101,11 +101,11 @@ def get_resume_token(self, invalidate=False):
                 if isinstance(cs._target, MongoClient):
                     self.skipTest("cluster-level change streams cannot be invalidated")
                 self.generate_invalidate_event(cs)
-                return cs.next()["_id"]
+                return (cs.next())["_id"]
         else:
             with self.change_stream() as cs:
                 coll.insert_one({"data": 1})
-                return cs.next()["_id"]
+                return (cs.next())["_id"]
 
     def get_start_at_operation_time(self):
         """Get an operationTime. Advances the operation clock beyond the most