From 2bb37ba3954da9199f81253a791fa5ac66a3509c Mon Sep 17 00:00:00 2001 From: appel_c Date: Mon, 11 Aug 2025 15:29:46 +0200 Subject: [PATCH] fix(mock-pv): add get_ctrlvars to better test alarm state callbacks --- ophyd_devices/tests/utils.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/ophyd_devices/tests/utils.py b/ophyd_devices/tests/utils.py index 8ebe89d..23afc80 100644 --- a/ophyd_devices/tests/utils.py +++ b/ophyd_devices/tests/utils.py @@ -9,6 +9,7 @@ from bec_lib.devicemanager import ScanInfo from bec_lib.logger import bec_logger from bec_lib.utils.import_utils import lazy_import_from from ophyd import Device +from ophyd.utils.epics_pvs import AlarmSeverity, AlarmStatus if TYPE_CHECKING: from bec_lib.messages import ScanStatusMessage @@ -240,6 +241,27 @@ class MockPV: get_metadata_thread(pvname=self.pvname) + def set_severity(self, severity: AlarmSeverity): + """Set severity for the mock PV""" + self._args["severity"] = severity + + def set_alarm_status(self, status: AlarmStatus): + """Set alarm status for the mock PV""" + self._args["status"] = status + + def get_ctrlvars(self, timeout=5, warn=True): + "get control values for variable" + return self._args + + def get_timevars(self, timeout=5, warn=True): + "get time values for variable" + kwds = { + "status": self._args["status"], + "severity": self._args["severity"], + "timestamp": self._args["timestamp"], + } + return kwds + # pylint disable: unused-argument def put( self, value, wait=False, timeout=None, use_complete=False, callback=None, callback_data=None