File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -102,6 +102,7 @@ async def task():
102
102
async def test_async_cancellation_closes_change_stream (self ):
103
103
self .addAsyncCleanup (self .client .db .test .delete_many , {})
104
104
change_stream = await self .client .db .test .watch (batch_size = 2 )
105
+ event = asyncio .Event ()
105
106
106
107
# Make sure getMore commands block
107
108
fail_command = {
@@ -113,11 +114,12 @@ async def test_async_cancellation_closes_change_stream(self):
113
114
async def task ():
114
115
async with self .fail_point (fail_command ):
115
116
await self .client .db .test .insert_many ([{"x" : 1 }, {"x" : 2 }])
117
+ event .set ()
116
118
await change_stream .next ()
117
119
118
120
task = asyncio .create_task (task ())
119
121
120
- await asyncio . sleep ( 0.1 )
122
+ await event . wait ( )
121
123
122
124
task .cancel ()
123
125
with self .assertRaises (asyncio .CancelledError ):
You can’t perform that action at this time.
0 commit comments