Skip to content

Commit f1fe497

Browse files
authored
PYTHON-5178 Make test_async_cancellation_closes_change_stream more robust (#2175)
1 parent 5ac2627 commit f1fe497

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

test/asynchronous/test_async_cancellation.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,6 +102,7 @@ async def task():
102102
async def test_async_cancellation_closes_change_stream(self):
103103
self.addAsyncCleanup(self.client.db.test.delete_many, {})
104104
change_stream = await self.client.db.test.watch(batch_size=2)
105+
event = asyncio.Event()
105106

106107
# Make sure getMore commands block
107108
fail_command = {
@@ -113,11 +114,12 @@ async def test_async_cancellation_closes_change_stream(self):
113114
async def task():
114115
async with self.fail_point(fail_command):
115116
await self.client.db.test.insert_many([{"x": 1}, {"x": 2}])
117+
event.set()
116118
await change_stream.next()
117119

118120
task = asyncio.create_task(task())
119121

120-
await asyncio.sleep(0.1)
122+
await event.wait()
121123

122124
task.cancel()
123125
with self.assertRaises(asyncio.CancelledError):

0 commit comments

Comments
 (0)