mirror of
https://github.com/bec-project/ophyd_devices.git
synced 2026-02-08 08:08:41 +01:00
fix(mock-pv): add get_ctrlvars to better test alarm state callbacks
This commit is contained in:
@@ -9,6 +9,7 @@ from bec_lib.devicemanager import ScanInfo
|
|||||||
from bec_lib.logger import bec_logger
|
from bec_lib.logger import bec_logger
|
||||||
from bec_lib.utils.import_utils import lazy_import_from
|
from bec_lib.utils.import_utils import lazy_import_from
|
||||||
from ophyd import Device
|
from ophyd import Device
|
||||||
|
from ophyd.utils.epics_pvs import AlarmSeverity, AlarmStatus
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from bec_lib.messages import ScanStatusMessage
|
from bec_lib.messages import ScanStatusMessage
|
||||||
@@ -240,6 +241,27 @@ class MockPV:
|
|||||||
|
|
||||||
get_metadata_thread(pvname=self.pvname)
|
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
|
# pylint disable: unused-argument
|
||||||
def put(
|
def put(
|
||||||
self, value, wait=False, timeout=None, use_complete=False, callback=None, callback_data=None
|
self, value, wait=False, timeout=None, use_complete=False, callback=None, callback_data=None
|
||||||
|
|||||||
Reference in New Issue
Block a user