From 50fffd3bd4d5936dad1d637fc736c8bfd0566785 Mon Sep 17 00:00:00 2001
From: appleb_m <martin.appleby@psi.ch>
Date: Fri, 22 May 2026 15:25:34 +0200
Subject: [PATCH] bec_worker.py: added check_is_state and check_current_state

---
 src/aare/devices/bec_worker.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/aare/devices/bec_worker.py b/src/aare/devices/bec_worker.py
index e42f2fa8..b44fcf2f 100644
--- a/src/aare/devices/bec_worker.py
+++ b/src/aare/devices/bec_worker.py
@@ -118,6 +118,18 @@ class BECClientWorker:
         except Exception as e:
             self._raise_bec_error(e, operation=f"planner.move_to:{state.value}")
 
+    def check_is_state(self, state: BeamlineState):
+        if self.simulated:
+            logger.debug(f"Simulating check_beamline_state: {state.value}")
+            return True
+        return self.planner.is_state(state)
+
+    def check_current_state(self):
+        if self.simulated:
+            logger.debug(f"Simulating check_beamline_state")
+            return BeamlineState.MAINTENANCE
+        return self.planner.current_state()
+
     def show_all_devices(self):
         return self.dev.show_all