diff --git a/distributed/tests/test_client.py b/distributed/tests/test_client.py index 3a2548cf4c..0cbe97a50a 100644 --- a/distributed/tests/test_client.py +++ b/distributed/tests/test_client.py @@ -7525,7 +7525,7 @@ async def test_client_story(c, s, *workers): class WorkerBrokenStory(Worker): async def get_story(self, *args, **kw): - await asyncio.Future() + raise CommClosedError @gen_cluster(client=True, Worker=WorkerBrokenStory)