diff --git a/tests/Settings/TestExtruderStack.py b/tests/Settings/TestExtruderStack.py index 2327e2d5bb..3a74f36efa 100644 --- a/tests/Settings/TestExtruderStack.py +++ b/tests/Settings/TestExtruderStack.py @@ -306,4 +306,19 @@ def test_setMaterialByIdExists(extruder_stack, container_registry): # exist. def test_setMaterialByIdDoesntExist(extruder_stack): with pytest.raises(KeyError): - extruder_stack.setMaterialById("some_material") #Container registry is empty now. \ No newline at end of file + extruder_stack.setMaterialById("some_material") #Container registry is empty now. + +## Tests setting qualities by specifying an ID of a quality that exists. +def test_setQualityByIdExists(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.setQualityById("some_quality") #The container registry always has a container with the ID. + + #Restore. + UM.Settings.ContainerStack._containerRegistry = original_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. \ No newline at end of file