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