From 66b9a1697eed5a6fcd57d16897ece3995d4b7647 Mon Sep 17 00:00:00 2001 From: fjetter Date: Wed, 23 Jun 2021 18:17:21 +0200 Subject: [PATCH] Fix unit test --- distributed/tests/test_core.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/distributed/tests/test_core.py b/distributed/tests/test_core.py index 573e8a279fe..d3df2e61867 100644 --- 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