From 9b7c5f6d69fda92f3d0531df0cf3bce1e8cc1bf1 Mon Sep 17 00:00:00 2001 From: Pete Jemian Date: Tue, 17 May 2022 13:39:11 -0500 Subject: [PATCH] TST #90 test top-level in customwidgets --- adl2pydm/tests/test_output_handler.py | 24 +++++++++++++++++++++++- 1 file changed, 23 insertions(+), 1 deletion(-) diff --git a/adl2pydm/tests/test_output_handler.py b/adl2pydm/tests/test_output_handler.py index 74baaa4..e243e11 100644 --- a/adl2pydm/tests/test_output_handler.py +++ b/adl2pydm/tests/test_output_handler.py @@ -20,6 +20,27 @@ def test_write_all_example_files_process(test_file, tempdir): assert os.path.exists(full_uiname), uiname +@pytest.mark.parametrize("test_file", _core.ALL_EXAMPLE_FILES) +def test_write_top_level_widget(test_file, tempdir): + uiname = _core.convertAdlFile("table_setup_SRI.adl", tempdir) + full_uiname = os.path.join(tempdir, uiname) + root = ElementTree.parse(full_uiname).getroot() + + widgets = _core.getSubElement(root, "widget") + assert len(widgets) > 0 + widget_classes = [ + w.attrib.get("class") + for w in widgets + if w.attrib.get("class") is not None + ] + assert output_handler.TOP_LEVEL_WIDGET_CLASS in widget_classes + + customwidgets = _core.getSubElement(root, "customwidgets") + widgets = customwidgets.findall("customwidget") + customs = [_core.getSubElement(w, "class").text for w in widgets] + assert output_handler.TOP_LEVEL_WIDGET_CLASS in customs + + def test_write_extends_customwidget(tempdir): uiname = _core.convertAdlFile("table_setup_SRI.adl", tempdir) full_uiname = os.path.join(tempdir, uiname) @@ -241,7 +262,8 @@ def test_write_widget_composite(tempdir): key = "composite" widget = _core.getNamedWidget(screen, key) - _core.assertEqualClassName(widget, "PyDMFrame", key) + # _core.assertEqualClassName(widget, "PyDMFrame", key) + _core.assertEqualClassName(widget, output_handler.TOP_LEVEL_WIDGET_CLASS, key) _core.assertEqual(len(widget), 6)