Fix unit test

dask · Jun 23, 2021 · 66b9a16 · 66b9a16
1 parent b71340a
commit 66b9a16
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/distributed/tests/test_core.py b/distributed/tests/test_core.py
@@ -651,7 +651,9 @@ async def connect_to_server():
     assert pool._connecting
     close_fut = asyncio.create_task(pool.close())
 
-    with pytest.raises(CommClosedError, match="ConnectionPool closed already"):
+    with pytest.raises(
+        CommClosedError, match="ConnectionPool not running.  Status: Status.closed"
+    ):
         await asyncio.gather(*tasks)
 
     await close_fut