close connections in test_retry_connect

dask · Apr 5, 2022 · 61554fe · 61554fe
1 parent 2cf6c27
commit 61554fe
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/distributed/comm/tests/test_comms.py b/distributed/comm/tests/test_comms.py
@@ -896,6 +896,7 @@ async def test_retry_connect(tcp, monkeypatch):
     async def echo(comm):
         message = await comm.read()
         await comm.write(message)
+        await comm.close()
 
     class UnreliableConnector(tcp.TCPConnector):
         def __init__(self):
@@ -922,6 +923,7 @@ class UnreliableBackend(tcp.TCPBackend):
         await comm.write(b"test")
         msg = await comm.read()
         assert msg == b"test"
+        await comm.close()
     finally:
         listener.stop()