Browse Source

Add tests for setQualityChangesById

One for when the ID exists, and one for when it doesn't.

Contributes to issue CURA-3497.
Ghostkeeper 8 years ago
parent
commit
a5d91139ef
1 changed files with 18 additions and 1 deletions
  1. 18 1
      tests/Settings/TestExtruderStack.py

+ 18 - 1
tests/Settings/TestExtruderStack.py

@@ -321,4 +321,21 @@ def test_setQualityByIdExists(extruder_stack, container_registry):
 ##  Tests setting qualities by specifying an ID of a quality that doesn't exist.
 def test_setQualityByIdDoesntExist(extruder_stack):
     with pytest.raises(KeyError):
-        extruder_stack.setQualityById("some_quality") #Container registry is empty now.
+        extruder_stack.setQualityById("some_quality") #Container registry is empty now.
+
+##  Tests setting quality changes by specifying an ID of a quality change that
+#   exists.
+def test_setQualityChangesByIdExists(extruder_stack, container_registry):
+    original_container_registry = UM.Settings.ContainerStack._containerRegistry
+    UM.Settings.ContainerStack._containerRegistry = container_registry #Always has all the profiles you ask of.
+
+    extruder_stack.setQualityChangesById("some_quality_changes") #The container registry always has a container with the ID.
+
+    #Restore.
+    UM.Settings.ContainerStack._containerRegistry = original_container_registry
+
+##  Tests setting quality changes by specifying an ID of a quality change that
+#   doesn't exist.
+def test_setQualityChangesByIdDoesntExist(extruder_stack):
+    with pytest.raises(KeyError):
+        extruder_stack.setQualityChangesById("some_quality_changes") #Container registry is empty now.