move all tests to pytest

.travis.yml

@@ -8,10 +8,11 @@ before_install:
 install:
   - pip install python-dateutil
   - pip install cryptography
+  - pip install pytest
   - pip install pyqt5
   - pip install pytz
   - pip install lxml
   - git clone https://github.com/FreeOpcUa/python-opcua.git 
   - git clone https://github.com/FreeOpcUa/opcua-widgets.git 
 # command to run tests
-script: python tests.py
+script: pytest -v

test_uamodeler.py

@@ -0,0 +1,266 @@
+
+import sys
+import pytest
+
+from opcua import ua
+
+from PyQt5.QtCore import QTimer, QSettings, QModelIndex, Qt, QCoreApplication
+from PyQt5.QtWidgets import QApplication, QAbstractItemDelegate
+from PyQt5.QtTest import QTest
+
+from uamodeler.uamodeler import UaModeler
+from uawidgets.new_node_dialogs import NewNodeBaseDialog, NewUaObjectDialog, NewUaVariableDialog, NewUaMethodDialog
+
+
+@pytest.fixture(scope="module")
+def modeler():
+    app = QApplication(sys.argv)
+    modeler = UaModeler()
+    yield modeler
+    #sys.exit(0)
+
+
+@pytest.fixture
+def mgr(modeler):
+    mgr = modeler.model_mgr._model_mgr
+    yield mgr
+    mgr.close_model(force=True)  # make sure we close model
+
+
+@pytest.fixture
+def model(mgr):
+    mgr.new_model()
+    yield model
+    mgr.close_model(True)
+
+
+def test_new_close(modeler, mgr):
+    mgr.new_model()
+    assert not mgr.modified
+    modeler.tree_ui.expand_to_node("Objects")
+    mgr.add_folder(1, "myfolder")
+    assert mgr.modified
+    with pytest.raises(RuntimeError):
+        mgr.close_model()
+    mgr.close_model(True)
+
+
+def test_save_open_xml(modeler, mgr):
+    path = "test_save_open.xml"
+    val = 0.99
+    mgr.new_model()
+    modeler.tree_ui.expand_to_node("Objects")
+    node = mgr.add_variable(1, "myvar", val)
+    mgr.save_xml(path)
+    mgr.close_model()
+    with pytest.raises(Exception):
+        node = mgr.server_mgr.get_node(node.nodeid)
+        node.get_value()
+    mgr.open(path)
+    node = mgr.server_mgr.get_node(node.nodeid)
+    assert node.get_value() == val
+    mgr.close_model(True)
+
+
+def test_save_open_ua_model(modeler, mgr):
+    path = "test_save_open.uamodel"
+    val = 0.99
+    mgr.new_model()
+    modeler.tree_ui.expand_to_node("Objects")
+    node = mgr.add_variable(1, "myvar", val)
+    mgr.save_ua_model(path)
+    mgr.save_xml(path)
+    mgr.close_model()
+    with pytest.raises(Exception):
+        node = mgr.server_mgr.get_node(node.nodeid)
+        node.get_value()
+    mgr.open(path)
+    node = mgr.server_mgr.get_node(node.nodeid)
+    node.get_value() == val
+    mgr.close_model()
+
+
+@pytest.mark.skip("Something wrong with expand_to_node")
+def test_delete_save(modeler, mgr):
+    path = "test_delete_save.uamodel"
+    val = 0.99
+    mgr.new_model()
+    modeler.tree_ui.expand_to_node("Objects")
+    obj_node = mgr.add_folder(1, "myobj")
+    modeler.tree_ui.expand_to_node("Objects")
+    modeler.tree_ui.expand_to_node("myobj")
+    obj2_node = mgr.add_folder(1, "myobj2")
+    modeler.tree_ui.reload_current()
+
+    modeler.tree_ui.expand_to_node("myobj2")
+    #var_node = mgr.add_variable(1, "myvar", val)
+    mgr.save_ua_model(path)
+    mgr.save_xml(path)
+
+    modeler.tree_ui.expand_to_node("myobj2")
+    mgr.delete_node(obj2_node)
+    mgr.save_ua_model(path)
+    mgr.save_xml(path)
+    assert obj2_node not in mgr.new_nodes
+    assert var_node not in mgr.new_nodes
+
+    mgr.close_model(force=True)
+
+
+def test_structs(modeler, mgr):
+    mgr.new_model()
+
+    urns = modeler.get_current_server().get_namespace_array()
+    ns_node = mgr.server_mgr.get_node(ua.ObjectIds.Server_NamespaceArray)
+    urns = ns_node.get_value()
+    urns.append("urn://modeller/testing")
+    ns_node.set_value(urns)
+
+    path = "test_save_structs.xml"
+
+    struct_node = mgr.server_mgr.get_node(ua.ObjectIds.Structure)
+    modeler.tree_ui.expand_to_node(struct_node)
+    mystruct = mgr.add_data_type(1, "MyStruct")
+    var1 = mystruct.add_variable(1, "MyFloat", 0.1, varianttype=ua.VariantType.Float)
+    var2 = mystruct.add_variable(1, "MyBytes", b'lkjlk', varianttype=ua.VariantType.ByteString)
+    mgr.save_xml(path)
+    assert len(mgr.new_nodes) == 4  # one struct + TypeDictionary node + namespace and struct node under typedict
+
+    # FIXME: test for presence of nodes under typedict for every new struct
+    opc_binary = mgr.server_mgr.get_node(ua.ObjectIds.OPCBinarySchema_TypeSystem)
+    typedict = opc_binary.get_child("1:TypeDictionary")
+    xml = typedict.get_value()
+    assert b"MyFloat" in xml
+    assert b"MyStruct" in xml
+
+    mgr.save_xml(path)
+    mgr.close_model()
+    mgr.open_xml(path)
+
+    opc_binary = mgr.server_mgr.get_node(ua.ObjectIds.OPCBinarySchema_TypeSystem)
+    typedict = opc_binary.get_child("1:TypeDictionary")
+    xml = typedict.get_value()
+
+    assert b"MyFloat" in xml
+    assert b"MyStruct" in xml
+
+    struct_node = mgr.server_mgr.get_node(ua.ObjectIds.Structure)
+    struct_node.get_child("1:MyStruct")
+
+    mgr.server_mgr.load_type_definitions()
+
+    st = ua.MyStruct()
+    assert hasattr(st, "MyFloat")
+    st.MyBytes = b"klk"
+
+
+"""
+
+class TestModeler(unittest.TestCase):
+
+    @classmethod
+    def setUpClass(cls):
+        cls.modeler = UaModeler()
+        cls.modeler.ui.actionNew.activate(0)
+        cls.mgr = cls.modeler.model_mgr._model_mgr
+        #modeler.show()
+        #sys.exit(app.exec_())
+
+    @classmethod
+    def tearDownClass(cls):
+        cls.modeler.get_current_server().stop_server()
+
+"""
+
+def test_set_current_node(modeler, mgr, model):
+    objects = modeler.get_current_server().nodes.objects
+    modeler.tree_ui.expand_to_node("Objects")
+    assert objects == modeler.tree_ui.get_current_node()
+
+
+def test_set_current_node_nodeid(modeler, mgr, model):
+    struct_node = mgr.server_mgr.get_node(ua.ObjectIds.Structure)
+    modeler.tree_ui.expand_to_node(struct_node)
+    assert struct_node == modeler.tree_ui.get_current_node()
+
+
+def test_add_folder(modeler, mgr, model):
+    modeler.tree_ui.expand_to_node("Objects")
+    dia = NewNodeBaseDialog(modeler, "Add Folder", modeler.get_current_server())
+    args = dia.get_args()
+    new_node = mgr.add_folder(*args)
+    assert new_node in modeler.get_current_server().nodes.objects.get_children()
+
+
+def test_add_variable_double(modeler, mgr, model):
+    modeler.tree_ui.expand_to_node("Objects")
+    dia = NewUaVariableDialog(modeler, "Add Variable", modeler.get_current_server(), dtype=ua.ObjectIds.Double)
+    args = dia.get_args()
+    new_node = mgr.add_variable(*args)
+    assert new_node in modeler.get_current_server().nodes.objects.get_children()
+
+
+def test_add_variable_double_list(modeler, mgr, model):
+    modeler.tree_ui.expand_to_node("Objects")
+    dia = NewUaVariableDialog(modeler, "Add Variable", modeler.get_current_server(), dtype=ua.ObjectIds.Double)
+    args = dia.get_args()
+    new_node = mgr.add_variable(*args)
+    assert new_node in modeler.get_current_server().nodes.objects.get_children()
+    assert isinstance(new_node.get_value(), float)
+
+
+def test_add_variable_string(modeler, mgr, model):
+    modeler.tree_ui.expand_to_node("Objects")
+    dia = NewUaVariableDialog(modeler, "Add Variable", modeler.get_current_server(), dtype=ua.ObjectIds.String)
+    args = dia.get_args()
+    new_node = mgr.add_variable(*args)
+    assert new_node in modeler.get_current_server().nodes.objects.get_children()
+
+
+def test_add_variable_extobj(modeler, mgr, model):
+    modeler.tree_ui.expand_to_node("Objects")
+    dia = NewUaVariableDialog(modeler, "Add Variable", modeler.get_current_server(), dtype=ua.ObjectIds.Structure)
+    args = dia.get_args()
+    new_node = mgr.add_variable(*args)
+    assert new_node in modeler.get_current_server().nodes.objects.get_children()
+
+
+def test_add_variable_bytes(modeler, mgr, model):
+    modeler.tree_ui.expand_to_node("Objects")
+    dia = NewUaVariableDialog(modeler, "Add Variable", modeler.get_current_server(), dtype=ua.ObjectIds.ByteString)
+    val = b"lkjkjl"
+    dia.valLineEdit.setText(val.decode())
+    args = dia.get_args()
+    new_node = mgr.add_variable(*args)
+    assert new_node in modeler.get_current_server().nodes.objects.get_children()
+    assert val == new_node.get_value()
+
+
+def test_add_variable_float_fail(modeler, mgr, model):
+    modeler.tree_ui.expand_to_node("Objects")
+    dia = NewUaVariableDialog(modeler, "Add Variable", modeler.get_current_server(), dtype=ua.ObjectIds.Float)
+    dia.valLineEdit.setText("oiuiu")
+    with pytest.raises(ValueError):
+        args = dia.get_args()
+        new_node = mgr.add_variable(*args)
+
+
+def test_add_namespace(modeler, mgr, model):
+    view = modeler.idx_ui.view
+    modeler.idx_ui.addNamespaceAction.activate(0)
+    editor = view.focusWidget()
+    urn = "urn:new_namespace"
+    editor.setText(urn)
+    view.commitData(editor)
+    view.closeEditor(editor, QAbstractItemDelegate.NoHint)
+    urns = modeler.get_current_server().get_namespace_array()
+    assert urn in urns
+
+    root = view.model().index(0, 0)
+    idx = root.child(len(urns)-1, 0)
+    view.setCurrentIndex(idx)
+    modeler.idx_ui.removeNamespaceAction.activate(0)
+    urns = modeler.get_current_server().get_namespace_array()
+    assert urn not in urns
+
+