@@ -139,6 +139,7 @@ async def test___call___generator_retry(self, sleep):
139139 unpacked = [await generator .__anext__ () for i in range (10 )]
140140 assert unpacked == [0 , 1 , 2 , 0 , 1 , 2 , 0 , 1 , 2 , 0 ]
141141 assert on_error .call_count == 3
142+ await generator .aclose ()
142143
143144 @mock .patch ("random.uniform" , autospec = True , side_effect = lambda m , n : n )
144145 @mock .patch ("asyncio.sleep" , autospec = True )
@@ -246,6 +247,7 @@ async def _mock_send_gen():
246247 recv = await generator .asend (msg )
247248 out_messages .append (recv )
248249 assert in_messages == out_messages
250+ await generator .aclose ()
249251
250252 @mock .patch ("asyncio.sleep" , autospec = True )
251253 @pytest .mark .asyncio
@@ -263,6 +265,7 @@ async def test___call___generator_send_retry(self, sleep):
263265 with pytest .raises (TypeError ) as exc_info :
264266 await generator .asend ("cannot send to fresh generator" )
265267 assert exc_info .match ("can't send non-None value" )
268+ await generator .aclose ()
266269
267270 # error thrown on 3
268271 # generator should contain 0, 1, 2 looping
@@ -271,6 +274,7 @@ async def test___call___generator_send_retry(self, sleep):
271274 unpacked = [await generator .asend (i ) for i in range (10 )]
272275 assert unpacked == [1 , 2 , 0 , 1 , 2 , 0 , 1 , 2 , 0 , 1 ]
273276 assert on_error .call_count == 3
277+ await generator .aclose ()
274278
275279 @mock .patch ("asyncio.sleep" , autospec = True )
276280 @pytest .mark .asyncio
@@ -382,6 +386,7 @@ async def wrapper():
382386 assert await retryable .asend ("test" ) == 1
383387 assert await retryable .asend ("test2" ) == 2
384388 assert await retryable .asend ("test3" ) == 3
389+ await retryable .aclose ()
385390
386391 @pytest .mark .parametrize ("awaitable_wrapped" , [True , False ])
387392 @mock .patch ("asyncio.sleep" , autospec = True )
0 commit comments