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

PYTHON-5178 Make test_async_cancellation_closes_change_stream more robust (#2175)

Browse Source
This commit is contained in:
Steven Silvester 2025-03-03 15:09:29 -06:00 committed by GitHub
parent 5ac262783f
commit f1fe49784b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test/asynchronous/test_async_cancellation.py
View File

@ -102,6 +102,7 @@ class TestAsyncCancellation(AsyncIntegrationTest):
async def test_async_cancellation_closes_change_stream(self):
self.addAsyncCleanup(self.client.db.test.delete_many, {})
change_stream = await self.client.db.test.watch(batch_size=2)
event = asyncio.Event()
# Make sure getMore commands block
fail_command = {
@ -113,11 +114,12 @@ class TestAsyncCancellation(AsyncIntegrationTest):
async def task():
async with self.fail_point(fail_command):
await self.client.db.test.insert_many([{"x": 1}, {"x": 2}])
event.set()
await change_stream.next()
task = asyncio.create_task(task())
await asyncio.sleep(0.1)
await event.wait()
task.cancel()
with self.assertRaises(asyncio.CancelledError):