Add tests for quality node

tests/Machines/TestMaterialNode.py

@@ -27,6 +27,7 @@ metadata_dict = {}
 def getMetadataEntrySideEffect(*args, **kwargs):
     return metadata_dict.get(args[0])
 
+
 def createMockedInstanceContainer(container_id):
     result = MagicMock()
     result.getId = MagicMock(return_value=container_id)
@@ -100,7 +101,6 @@ def test_qualityAdded_hasMachineQuality(container_registry, metadata, qualities_
         assert name in node.qualities
 
 
-
 @pytest.mark.parametrize("metadata,qualities_result_list", quality_metadata_no_machine_quality)
 def test_qualityAdded_noMachineQuality(container_registry, metadata, qualities_result_list):
     variant_node = MagicMock()

tests/Machines/TestQualityNode.py

@@ -0,0 +1,78 @@
+from unittest.mock import patch, MagicMock
+import pytest
+
+from cura.Machines.MaterialNode import MaterialNode
+from cura.Machines.QualityNode import QualityNode
+
+
+instance_container_metadata_dict = {"fdmprinter": {"variant_1": {"material_1": [{"id": "intent_1"}, {"id": "intent_2"}]}},
+                                    "machine_1": {"variant_2": {"material_2": [{"id": "intent_3"}, {"id": "intent_4"}]}}}
+
+
+intent_metadata_intent_added_data = [({"type": "Not an intent"}, ["intent_3", "intent_4"]),  # Wrong type
+                                    ({"type": "intent", "definition": "machine_9000"}, ["intent_3", "intent_4"]),  # wrong definition
+                                    ({"type": "intent", "definition": "machine_1", "variant": "variant_299101"}, ["intent_3", "intent_4"]),  # wrong variant
+                                    ({"type": "intent", "definition": "machine_1", "variant": "variant_2", "material": "super cool material!"}, ["intent_3", "intent_4"]),  # Wrong material
+                                    ({"type": "intent", "definition": "machine_1", "variant": "variant_2", "material": "material_2"}, ["intent_3", "intent_4", "intent_9001"]),  # Yay, all good.
+]
+
+metadata_dict = {}
+
+
+def getMetadataEntrySideEffect(*args, **kwargs):
+    return metadata_dict.get(args[0])
+
+
+def createMockedInstanceContainer(container_id):
+    result = MagicMock()
+    result.getId = MagicMock(return_value=container_id)
+    result.getMetaDataEntry = MagicMock(side_effect=getMetadataEntrySideEffect)
+    return result
+
+
+def getInstanceContainerSideEffect(*args, **kwargs):
+
+    definition_dict = instance_container_metadata_dict.get(kwargs["definition"])
+    variant_dict = definition_dict.get(kwargs["variant"])
+    return variant_dict.get(kwargs["material"])
+
+@pytest.fixture
+def container_registry():
+    result = MagicMock()
+    result.findInstanceContainersMetadata = MagicMock(side_effect=getInstanceContainerSideEffect)
+    return result
+
+
+def test_qualityNode_machine_1(container_registry):
+    material_node = MagicMock()
+    material_node.variant.machine.quality_definition = "machine_1"
+    material_node.variant.variant_name = "variant_2"
+    material_node.base_file = "material_2"
+
+    with patch("cura.Machines.QualityNode.IntentNode"):
+        with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value=container_registry)):
+            node = QualityNode("quality_1", material_node)
+
+    assert len(node.intents) == 2
+    assert "intent_3" in node.intents
+    assert "intent_4" in node.intents
+
+@pytest.mark.parametrize("metadata,intent_result_list", intent_metadata_intent_added_data)
+def test_intentNodeAdded(container_registry, metadata, intent_result_list):
+    material_node = MagicMock()
+    material_node.variant.machine.quality_definition = "machine_1"
+    material_node.variant.variant_name = "variant_2"
+    material_node.base_file = "material_2"
+
+    intent_container = createMockedInstanceContainer("intent_9001")
+
+    with patch("cura.Machines.QualityNode.IntentNode"):
+        with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value=container_registry)):
+            node = QualityNode("quality_1", material_node)
+            with patch.dict(metadata_dict, metadata):
+                node._intentAdded(intent_container)
+
+    assert len(intent_result_list) == len(node.intents)
+    for identifier in intent_result_list:
+        assert identifier in node.intents
+