0
0
mirror of https://github.com/bec-project/bec_widgets.git synced 2025-07-14 11:41:49 +02:00

test: validation errors tests

This commit is contained in:
wyzula-jan
2023-12-07 18:50:26 +01:00
parent 086804780d
commit 2a334156a8
3 changed files with 88 additions and 3 deletions

View File

@ -60,7 +60,7 @@ class Signal(BaseModel):
if entry not in device.signals: if entry not in device.signals:
raise PydanticCustomError( raise PydanticCustomError(
"no_entry_for_device", "no_entry_for_device",
"Entry '{wrong_value}' not found in device '{device_name}' signals", 'Entry "{wrong_value}" not found in device "{device_name}" signals',
{"wrong_value": entry, "device_name": name}, {"wrong_value": entry, "device_name": name},
) )
@ -100,8 +100,8 @@ class PlotConfig(BaseModel):
def validate_x_signals(cls, v): def validate_x_signals(cls, v):
if len(v.signals) != 1: if len(v.signals) != 1:
raise PydanticCustomError( raise PydanticCustomError(
"x_device_one_signal", "x_axis_multiple_signals",
"There must be exactly one signal for x axis. Number of x signals: '{wrong_value}'", 'There must be exactly one signal for x axis. Number of x signals: "{wrong_value}"',
{"wrong_value": v}, {"wrong_value": v},
) )

View File

@ -54,6 +54,11 @@ def mocked_client():
device_names = ["samx", "gauss_bpm", "gauss_adc1", "gauss_adc2", "gauss_adc3", "bpm4i"] device_names = ["samx", "gauss_bpm", "gauss_adc1", "gauss_adc2", "gauss_adc3", "bpm4i"]
mocked_devices = {name: get_mocked_device(name) for name in device_names} mocked_devices = {name: get_mocked_device(name) for name in device_names}
# Adding a device with empty signals for validation tests
no_signal_device = FakeDevice(name="no_signal_device")
del no_signal_device.signals # Simulate a device with no signals
mocked_devices["no_signal_device"] = no_signal_device
# Create a MagicMock object # Create a MagicMock object
client = MagicMock() client = MagicMock()

View File

@ -0,0 +1,80 @@
# pylint: disable=missing-function-docstring
import pytest
from pydantic import ValidationError
from bec_widgets.validation.monitor_config_validator import (
MonitorConfigValidator,
Signal,
PlotAxis,
PlotConfig,
)
from test_bec_monitor import mocked_client
@pytest.fixture(scope="function")
def setup_devices(mocked_client):
MonitorConfigValidator.devices = mocked_client.device_manager.devices
def test_signal_validation_name_missing(setup_devices):
with pytest.raises(ValidationError) as excinfo:
Signal(name=None)
errors = excinfo.value.errors()
assert len(errors) == 1
assert errors[0]["type"] == "no_device_name"
assert "Device name must be provided" in str(excinfo.value)
def test_signal_validation_name_not_in_bec(setup_devices):
with pytest.raises(ValidationError) as excinfo:
Signal(name="non_existent_device")
errors = excinfo.value.errors()
assert len(errors) == 1
assert errors[0]["type"] == "no_device_bec"
assert 'Device "non_existent_device" not found in current BEC session' in str(excinfo.value)
def test_signal_validation_device_has_no_signals(setup_devices):
with pytest.raises(ValidationError) as excinfo:
Signal(name="no_signal_device")
errors = excinfo.value.errors()
assert len(errors) == 1
assert errors[0]["type"] == "no_device_signals"
assert 'Device "no_signal_device" does not have "signals" defined' in errors[0]["msg"]
def test_signal_validation_entry_not_in_device(setup_devices):
with pytest.raises(ValidationError) as excinfo:
Signal(name="samx", entry="non_existent_entry")
errors = excinfo.value.errors()
assert len(errors) == 1
assert errors[0]["type"] == "no_entry_for_device"
assert 'Entry "non_existent_entry" not found in device "samx" signals' in errors[0]["msg"]
def test_signal_validation_success(setup_devices):
signal = Signal(name="samx")
assert signal.name == "samx"
def test_plot_config_x_axis_signal_validation(setup_devices):
# Setup a valid signal
valid_signal = Signal(name="samx")
# Case with more than one signal for x-axis
plot_axis_multiple_signals = PlotAxis(
signals=[valid_signal, valid_signal], label="X Axis Label"
)
with pytest.raises(ValidationError) as excinfo:
PlotConfig(
plot_name="Test Plot",
x=plot_axis_multiple_signals,
y=PlotAxis(signals=[valid_signal], label="Y Axis Label"),
)
errors = excinfo.value.errors()
assert len(errors) == 1
assert errors[0]["type"] == "x_axis_multiple_signals"
assert "There must be exactly one signal for x axis" in errors[0]["msg"]