mirror of
https://github.com/bec-project/bec_widgets.git
synced 2025-07-14 03:31:50 +02:00
67 lines
2.4 KiB
Python
67 lines
2.4 KiB
Python
import pytest
|
|
|
|
from bec_widgets.widgets.device_inputs.device_input_base import DeviceInputBase
|
|
|
|
from .client_mocks import mocked_client
|
|
|
|
|
|
@pytest.fixture
|
|
def device_input_base(mocked_client):
|
|
widget = DeviceInputBase(client=mocked_client)
|
|
yield widget
|
|
|
|
|
|
def test_device_input_base_init(device_input_base):
|
|
assert device_input_base is not None
|
|
assert device_input_base.client is not None
|
|
assert isinstance(device_input_base, DeviceInputBase)
|
|
assert device_input_base.config.widget_class == "DeviceInputBase"
|
|
assert device_input_base.config.device_filter is None
|
|
assert device_input_base.config.default is None
|
|
assert device_input_base.devices == []
|
|
|
|
|
|
def test_device_input_base_init_with_config(mocked_client):
|
|
config = {
|
|
"widget_class": "DeviceInputBase",
|
|
"gui_id": "test_gui_id",
|
|
"device_filter": "FakePositioner",
|
|
"default": "samx",
|
|
}
|
|
widget = DeviceInputBase(client=mocked_client, config=config)
|
|
assert widget.config.gui_id == "test_gui_id"
|
|
assert widget.config.device_filter == "FakePositioner"
|
|
assert widget.config.default == "samx"
|
|
|
|
|
|
def test_device_input_base_set_device_filter(device_input_base):
|
|
device_input_base.set_device_filter("FakePositioner")
|
|
assert device_input_base.config.device_filter == "FakePositioner"
|
|
|
|
|
|
def test_device_input_base_set_device_filter_error(device_input_base):
|
|
with pytest.raises(ValueError) as excinfo:
|
|
device_input_base.set_device_filter("NonExistingClass")
|
|
assert "Device filter NonExistingClass is not in the device list." in str(excinfo.value)
|
|
|
|
|
|
def test_device_input_base_set_default_device(device_input_base):
|
|
device_input_base.set_default_device("samx")
|
|
assert device_input_base.config.default == "samx"
|
|
|
|
|
|
def test_device_input_base_set_default_device_error(device_input_base):
|
|
with pytest.raises(ValueError) as excinfo:
|
|
device_input_base.set_default_device("NonExistingDevice")
|
|
assert "Default device NonExistingDevice is not in the device list." in str(excinfo.value)
|
|
|
|
|
|
def test_device_input_base_get_device_list(device_input_base):
|
|
devices = device_input_base.get_device_list("FakePositioner")
|
|
assert devices == ["samx", "samy", "aptrx", "aptry"]
|
|
|
|
|
|
def test_device_input_base_get_filters(device_input_base):
|
|
filters = device_input_base.get_available_filters()
|
|
assert filters == {"FakePositioner", "FakeDevice"}
|