Skip to content

Commit 67f5616

Browse files
Added tests for changes in models
1 parent 907ba22 commit 67f5616

File tree

1 file changed

+21
-1
lines changed

1 file changed

+21
-1
lines changed

tests/test_models.py

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import pytest
22

3-
from techui_builder.models import Beamline, Component
3+
from techui_builder.models import Beamline, Component, GuiComponentEntry, GuiComponents
44

55

66
@pytest.fixture
@@ -13,6 +13,13 @@ def component() -> Component:
1313
return Component(prefix="BL01T-EA-TEST-02", desc="Test Device")
1414

1515

16+
@pytest.fixture
17+
def gui_components() -> GuiComponentEntry:
18+
return GuiComponentEntry(
19+
file="digitelMpc/digitelMpcIonp.bob", prefix="$(P)", type="embedded"
20+
)
21+
22+
1623
# @pytest.mark.parametrize("beamline,expected",[])
1724
def test_beamline_object(beamline: Beamline):
1825
assert beamline.short_dom == "t01"
@@ -39,3 +46,16 @@ def test_component_repr(component: Component):
3946
def test_component_bad_prefix():
4047
with pytest.raises(ValueError):
4148
Component(prefix="Test 2", desc="BAD_PREFIX")
49+
50+
51+
def test_gui_component_entry(gui_components: GuiComponentEntry):
52+
assert gui_components.file == "digitelMpc/digitelMpcIonp.bob"
53+
assert gui_components.prefix == "$(P)"
54+
assert gui_components.type == "embedded"
55+
56+
57+
def test_gui_components_object(gui_components: GuiComponentEntry):
58+
gc = GuiComponents({"digitelMpc.digitelMpcIonp": gui_components})
59+
assert gc.root["digitelMpc.digitelMpcIonp"].file == "digitelMpc/digitelMpcIonp.bob"
60+
assert gc.root["digitelMpc.digitelMpcIonp"].prefix == "$(P)"
61+
assert gc.root["digitelMpc.digitelMpcIonp"].type == "embedded"

0 commit comments

Comments
 (0)