diff --git a/tests/unit_tests/core/fake_device_module_failing.py b/tests/unit_tests/core/fake_device_module_failing.py index 1fb64c945..0fe3df71a 100644 --- a/tests/unit_tests/core/fake_device_module_failing.py +++ b/tests/unit_tests/core/fake_device_module_failing.py @@ -1,5 +1,20 @@ +from ophyd_async.core import LazyMock, StandardReadable from ophyd_async.epics.motor import Motor +class TimingOutDevice(StandardReadable): + async def connect( + self, + mock: bool | LazyMock = False, + timeout: float = 1.0, + force_reconnect: bool = False, + ) -> None: + raise TimeoutError("Connection timed out") + + def failing_device() -> Motor: raise TimeoutError("FooBar") + + +def timing_out_device() -> TimingOutDevice: + return TimingOutDevice()