mirror of
https://github.com/ivan-usov-org/bec.git
synced 2025-04-22 02:20:02 +02:00
67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
from unittest import mock
|
|
|
|
import pytest
|
|
from bec_utils import BECMessage
|
|
from scan_server.scan_guard import ScanGuard, ScanRejection
|
|
|
|
from utils import load_ScanServerMock
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"scan_queue_msg",
|
|
[
|
|
(
|
|
BECMessage.ScanQueueMessage(
|
|
scan_type="fermat_scan",
|
|
parameter={
|
|
"args": {"samx": (-5, 5), "samy": (-5, 5)},
|
|
"kwargs": {"step": 3},
|
|
},
|
|
queue="primary",
|
|
)
|
|
),
|
|
(
|
|
BECMessage.ScanQueueMessage(
|
|
scan_type="device_rpc",
|
|
parameter={
|
|
"device": "samy",
|
|
"args": {},
|
|
"kwargs": {},
|
|
},
|
|
queue="primary",
|
|
)
|
|
),
|
|
(
|
|
BECMessage.ScanQueueMessage(
|
|
scan_type="device_rpc",
|
|
parameter={
|
|
"device": ["samy"],
|
|
"args": {},
|
|
"kwargs": {},
|
|
},
|
|
queue="primary",
|
|
)
|
|
),
|
|
],
|
|
)
|
|
def test_check_motors_movable_enabled(scan_queue_msg):
|
|
k = load_ScanServerMock()
|
|
|
|
sg = ScanGuard(parent=k)
|
|
sg._check_motors_movable(scan_queue_msg)
|
|
config_reply = BECMessage.RequestResponseMessage(accepted=True, message="")
|
|
with mock.patch.object(k.device_manager, "wait_for_config_reply", return_value=config_reply):
|
|
k.device_manager.devices["samx"].enabled = True
|
|
k.device_manager.devices["samy"].enabled = False
|
|
with pytest.raises(ScanRejection) as scan_rejection:
|
|
sg._check_motors_movable(scan_queue_msg)
|
|
assert "Device samy is not enabled." in scan_rejection.value.args
|
|
|
|
|
|
@pytest.mark.parametrize("device,func,is_valid", [("samx", "read", True)])
|
|
def test_device_rpc_is_valid(device, func, is_valid):
|
|
k = load_ScanServerMock()
|
|
|
|
sg = ScanGuard(parent=k)
|
|
assert sg._device_rpc_is_valid(device, func) == is_valid
|