diff --git a/test/asynchronous/test_cursor.py b/test/asynchronous/test_cursor.py index d80ab5c5a..5b3175a61 100644 --- a/test/asynchronous/test_cursor.py +++ b/test/asynchronous/test_cursor.py @@ -129,7 +129,7 @@ class TestCursor(AsyncIntegrationTest): # Exhaust - which mongos doesn't support if async_client_context.is_mongos: with self.assertRaises(InvalidOperation): - self.db.test.find(cursor_type=CursorType.EXHAUST) + await anext(self.db.test.find(cursor_type=CursorType.EXHAUST)) else: cursor = self.db.test.find(cursor_type=CursorType.EXHAUST) self.assertEqual(64, cursor._query_flags) diff --git a/test/asynchronous/test_transactions.py b/test/asynchronous/test_transactions.py index fe2c6b36b..373fda588 100644 --- a/test/asynchronous/test_transactions.py +++ b/test/asynchronous/test_transactions.py @@ -192,7 +192,7 @@ class TestTransactions(AsyncTransactionsBase): addresses = set() for _ in range(UNPIN_TEST_MAX_ATTEMPTS): async with await s.start_transaction(): - cursor = await coll.find({}, session=s) + cursor = coll.find({}, session=s) self.assertTrue(await anext(cursor)) addresses.add(cursor.address) # Break early if we can. @@ -219,7 +219,7 @@ class TestTransactions(AsyncTransactionsBase): addresses = set() for _ in range(UNPIN_TEST_MAX_ATTEMPTS): - cursor = await coll.find({}, session=s) + cursor = coll.find({}, session=s) self.assertTrue(await anext(cursor)) addresses.add(cursor.address) # Break early if we can. diff --git a/test/test_cursor.py b/test/test_cursor.py index 6fbdf38b2..03824a5e7 100644 --- a/test/test_cursor.py +++ b/test/test_cursor.py @@ -129,7 +129,7 @@ class TestCursor(IntegrationTest): # Exhaust - which mongos doesn't support if client_context.is_mongos: with self.assertRaises(InvalidOperation): - self.db.test.find(cursor_type=CursorType.EXHAUST) + next(self.db.test.find(cursor_type=CursorType.EXHAUST)) else: cursor = self.db.test.find(cursor_type=CursorType.EXHAUST) self.assertEqual(64, cursor._query_flags)