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():
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 ):
You can’t perform that action at this time.
0 commit comments