From 22e29142567fc37c2661e95dd4c7d54ccb0b11fb Mon Sep 17 00:00:00 2001
From: Callum Forrester <callum.forrester@diamond.ac.uk>
Date: Wed, 25 Oct 2023 20:43:59 +0100
Subject: [PATCH] Make devices connect with a timeout

---
 src/blueapi/core/context.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/blueapi/core/context.py b/src/blueapi/core/context.py
index ffad9b98a..6118a420a 100644
--- a/src/blueapi/core/context.py
+++ b/src/blueapi/core/context.py
@@ -1,3 +1,4 @@
+import asyncio
 import functools
 import logging
 from dataclasses import dataclass, field
@@ -114,7 +115,7 @@ async def connect_devices(self, sim: bool = False) -> None:
                 coros[device_name] = device.connect(sim)
 
         if len(coros) > 0:
-            await wait_for_connection(**coros)
+            await asyncio.wait(wait_for_connection(**coros), timeout=30.0)
 
     def with_plan_module(self, module: ModuleType) -> None:
         """