mongodb/mongo-python-driver
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

PYTHON-4610 More robust to_list tests (#1773)

Browse Source
This commit is contained in:
Shane Harvey 2024-08-06 16:29:24 -07:00 committed by GitHub
parent d08fec6342
commit da59318327
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 24 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
test/asynchronous/test_cursor.py
View File

@ -1380,41 +1380,39 @@ class TestCursor(AsyncIntegrationTest):
self.assertEqual("getMore", started[1].command_name)
self.assertNotIn("$readPreference", started[1].command)
@async_client_context.require_version_min(4, 0)
@async_client_context.require_replica_set
async def test_to_list_tailable(self):
oplog = self.client.local.oplog.rs
last = await oplog.find().sort("$natural", pymongo.DESCENDING).limit(-1).next()
ts = last["ts"]
# Set maxAwaitTimeMS=1 to speed up the test and avoid blocking on the noop writer.
c = oplog.find(
{"ts": {"$gte": ts}}, cursor_type=pymongo.CursorType.TAILABLE_AWAIT, oplog_replay=True
)
).max_await_time_ms(1)
self.addAsyncCleanup(c.close)
docs = await c.to_list()
self.assertGreaterEqual(len(docs), 1)
async def test_to_list_empty(self):
c = self.db.does_not_exist.find()
docs = await c.to_list()
self.assertEqual([], docs)
@async_client_context.require_replica_set
@async_client_context.require_change_streams
async def test_command_cursor_to_list(self):
c = await self.db.test.aggregate([{"$changeStream": {}}])
# Set maxAwaitTimeMS=1 to speed up the test.
c = await self.db.test.aggregate([{"$changeStream": {}}], maxAwaitTimeMS=1)
self.addAsyncCleanup(c.close)
docs = await c.to_list()
self.assertGreaterEqual(len(docs), 0)
@async_client_context.require_replica_set
@async_client_context.require_change_streams
async def test_command_cursor_to_list_empty(self):
c = await self.db.does_not_exist.aggregate([{"$changeStream": {}}])
# Set maxAwaitTimeMS=1 to speed up the test.
c = await self.db.does_not_exist.aggregate([{"$changeStream": {}}], maxAwaitTimeMS=1)
self.addAsyncCleanup(c.close)
docs = await c.to_list()
self.assertEqual([], docs)

26
test/test_cursor.py
View File

@ -1371,41 +1371,39 @@ class TestCursor(IntegrationTest):
self.assertEqual("getMore", started[1].command_name)
self.assertNotIn("$readPreference", started[1].command)
@client_context.require_version_min(4, 0)
@client_context.require_replica_set
def test_to_list_tailable(self):
oplog = self.client.local.oplog.rs
last = oplog.find().sort("$natural", pymongo.DESCENDING).limit(-1).next()
ts = last["ts"]
# Set maxAwaitTimeMS=1 to speed up the test and avoid blocking on the noop writer.
c = oplog.find(
{"ts": {"$gte": ts}}, cursor_type=pymongo.CursorType.TAILABLE_AWAIT, oplog_replay=True
)
).max_await_time_ms(1)
self.addCleanup(c.close)
docs = c.to_list()
self.assertGreaterEqual(len(docs), 1)
def test_to_list_empty(self):
c = self.db.does_not_exist.find()
docs = c.to_list()
self.assertEqual([], docs)
@client_context.require_replica_set
@client_context.require_change_streams
def test_command_cursor_to_list(self):
c = self.db.test.aggregate([{"$changeStream": {}}])
# Set maxAwaitTimeMS=1 to speed up the test.
c = self.db.test.aggregate([{"$changeStream": {}}], maxAwaitTimeMS=1)
self.addCleanup(c.close)
docs = c.to_list()
self.assertGreaterEqual(len(docs), 0)
@client_context.require_replica_set
@client_context.require_change_streams
def test_command_cursor_to_list_empty(self):
c = self.db.does_not_exist.aggregate([{"$changeStream": {}}])
# Set maxAwaitTimeMS=1 to speed up the test.
c = self.db.does_not_exist.aggregate([{"$changeStream": {}}], maxAwaitTimeMS=1)
self.addCleanup(c.close)
docs = c.to_list()
self.assertEqual([], docs)