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

Merge branch 'master' of github.com:mongodb/mongo-python-driver

Browse Source
This commit is contained in:
Steven Silvester 2025-03-03 19:18:33 -06:00
commit e3b7a2e5cd
No known key found for this signature in database
GPG Key ID: B1BF5EC3A8B32F91
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):