Close a Client if it doesn't startup in time

dask · Jul 31, 2019 · 3e7aa67 · 3e7aa67
1 parent 2061fd4
commit 3e7aa67
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/distributed/client.py b/distributed/client.py
@@ -953,7 +953,11 @@ async def _start(self, timeout=no_default, **kwargs):
             self.scheduler = self.rpc(address)
         self.scheduler_comm = None
 
-        await self._ensure_connected(timeout=timeout)
+        try:
+            await self._ensure_connected(timeout=timeout)
+        except OSError:
+            await self._close()
+            raise
 
         for pc in self._periodic_callbacks.values():
             pc.start()

diff --git a/distributed/tests/test_client.py b/distributed/tests/test_client.py
@@ -5233,6 +5233,7 @@ def test_client_timeout_2():
             yield c
         stop = time()
 
+        assert c.status == "closed"
         yield c.close()
 
         assert stop - start < 1