Use channel.aclose() in tests and not channel.close()

Signed-off-by: Sahas Subramanian <[email protected]>

Author: shsms

tests/test_anycast.py

@@ -64,7 +64,7 @@ async def update_tracker_on_receive(receiver_id: int, recv: Receiver[int]) -> No
         senders.append(send_msg(acast.new_sender()))
 
     await asyncio.gather(*senders)
-    await acast.close()
+    await acast.aclose()
     await receivers_runs
 
     with pytest.raises(SenderError):
@@ -92,7 +92,7 @@ async def test_anycast_after_close() -> None:
 
     await sender.send(2)
 
-    await acast.close()
+    await acast.aclose()
 
     with pytest.raises(SenderError):
         await sender.send(5)
@@ -174,7 +174,7 @@ async def test_anycast_async_iterator() -> None:
     async def send_messages() -> None:
         for val in ["one", "two", "three", "four", "five"]:
             await sender.send(val)
-        await acast.close()
+        await acast.aclose()
 
     sender_task = asyncio.create_task(send_messages())
 

tests/test_broadcast.py

@@ -58,7 +58,7 @@ async def update_tracker_on_receive(receiver_id: int, recv: Receiver[int]) -> No
         senders.append(send_msg(bcast.new_sender()))
 
     await asyncio.gather(*senders)
-    await bcast.close()
+    await bcast.aclose()
     await receivers_runs
 
     actual_sum = 0
@@ -93,7 +93,7 @@ async def test_broadcast_after_close() -> None:
     receiver = bcast.new_receiver()
     sender = bcast.new_sender()
 
-    await bcast.close()
+    await bcast.aclose()
 
     with pytest.raises(SenderError):
         await sender.send(5)
@@ -204,7 +204,7 @@ async def test_broadcast_async_iterator() -> None:
     async def send_messages() -> None:
         for val in range(0, 10):
             await sender.send(val)
-        await bcast.close()
+        await bcast.aclose()
 
     sender_task = asyncio.create_task(send_messages())
 

tests/test_merge_integration.py

@@ -22,9 +22,9 @@ async def send(ch1: Sender[int], ch2: Sender[int]) -> None:
         for ctr in range(5):
             await ch1.send(ctr + 1)
             await ch2.send(ctr + 101)
-        await chan1.close()
+        await chan1.aclose()
         await ch2.send(1000)
-        await chan2.close()
+        await chan2.aclose()
 
     senders = asyncio.create_task(send(chan1.new_sender(), chan2.new_sender()))
 
@@ -52,8 +52,8 @@ async def send(ch1: Sender[int], ch2: Sender[int]) -> None:
         for ctr in range(5):
             await ch1.send(ctr + 1)
             await ch2.send(ctr + 101)
-        await chan1.close()
-        await chan2.close()
+        await chan1.aclose()
+        await chan2.aclose()
 
     rx1 = chan1.new_receiver()
     rx2 = chan2.new_receiver()