|
| 1 | +from pathlib import Path |
| 2 | +from unittest.mock import Mock, patch |
| 3 | + |
| 4 | +from lxml.etree import Element, SubElement, _ElementTree, tostring |
| 5 | +from lxml.objectify import fromstring |
| 6 | +from phoebusgen.widget import EmbeddedDisplay |
| 7 | + |
| 8 | + |
| 9 | +def test_validator_check_bobs(validator): |
| 10 | + validator._check_bob = Mock() |
| 11 | + |
| 12 | + validator.check_bobs() |
| 13 | + |
| 14 | + validator._check_bob.assert_called() |
| 15 | + |
| 16 | + |
| 17 | +def test_validator_check_bob(validator): |
| 18 | + validator._check_bob(validator.bobs[0]) |
| 19 | + |
| 20 | + assert len(validator.validate.keys()) > 0 |
| 21 | + assert list(validator.validate.keys())[0] == "motor-edited" |
| 22 | + |
| 23 | + |
| 24 | +def test_validator_read_bob(validator): |
| 25 | + with patch("techui_builder.validator.read_bob") as mock_read_bob: |
| 26 | + # We need to set the spec of the first Mock so it knows |
| 27 | + # it has a getroot() function |
| 28 | + mock_read_bob.return_value = (Mock(spec=_ElementTree), Mock()) |
| 29 | + |
| 30 | + validator._read_bob(validator.bobs[0]) |
| 31 | + |
| 32 | + |
| 33 | +# TODO: Clean up this test... (make fixture for mock xml?) |
| 34 | +def test_validator_validate_bob(validator): |
| 35 | + # You cannot set a text tag of an ObjectifiedElement, |
| 36 | + # so we need to make an etree.Element and convert it ... |
| 37 | + mock_root_element = Element("root") |
| 38 | + mock_widget_element = SubElement(mock_root_element, "widget") |
| 39 | + mock_name_element = SubElement(mock_widget_element, "name") |
| 40 | + mock_name_element.text = "motor" |
| 41 | + mock_width_element = SubElement(mock_widget_element, "width") |
| 42 | + mock_width_element.text = "205" |
| 43 | + mock_height_element = SubElement(mock_widget_element, "height") |
| 44 | + mock_height_element.text = "120" |
| 45 | + mock_file_element = SubElement(mock_widget_element, "file") |
| 46 | + mock_file_element.text = ( |
| 47 | + "example/t01-services/synoptic/techui_supportbob/pmac/motor_embed.bob" |
| 48 | + ) |
| 49 | + # ... which requires this horror |
| 50 | + mock_element = fromstring(tostring(mock_root_element)) |
| 51 | + # mock_element = ObjectifiedElement(mock_widget_element) |
| 52 | + # mock_name_element.text = "motor" |
| 53 | + validator._read_bob = Mock( |
| 54 | + return_value=( |
| 55 | + Mock(), |
| 56 | + {"motor": (mock_element)}, |
| 57 | + ) |
| 58 | + ) |
| 59 | + validator.validate = {"motor-edited": Path("tests/test_files/motor-edited.bob")} |
| 60 | + test_pwidget = EmbeddedDisplay( |
| 61 | + "motor", |
| 62 | + "example/t01-services/synoptic/techui_supportbob/pmac/motor_embed.bob", |
| 63 | + 0, |
| 64 | + 0, |
| 65 | + 205, |
| 66 | + 120, |
| 67 | + ) |
| 68 | + |
| 69 | + validator.validate_bob("motor-edited", "motor", [test_pwidget]) |
0 commit comments