3D-printing/Cura
1
0
Fork 0
mirror of https://github.com/Ultimaker/Cura.git synced 2025-08-07 14:04:03 -06:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

Add tests for the _onRemoved handler

Browse source 
CURA-6600
This commit is contained in:
Jaime van Kessel 2019-08-21 14:45:59 +02:00
parent a6be5ac52d
commit 25ea1dd66a
1 changed files with 32 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

33
tests/Machines/TestMaterialNode.py
View file

@ -71,4 +71,35 @@ def test_materialNodeInit_MachineQuality(container_registry):
assert len(node.qualities) == 2
assert "quality_2" in node.qualities
assert "quality_3" in node.qualities
assert "quality_3" in node.qualities
def test_onRemoved_wrongContainer(container_registry):
variant_node = MagicMock()
variant_node.variant_name = "variant_1"
variant_node.machine.has_machine_quality = True
variant_node.machine.quality_definition = "machine_1"
variant_node.materials = {"material_1": MagicMock()}
with patch("cura.Machines.MaterialNode.QualityNode"):
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance",MagicMock(return_value=container_registry)):
node = MaterialNode("material_1", variant_node)
container = createMockedInstanceContainer("material_2")
node._onRemoved(container)
assert "material_1" in variant_node.materials
def test_onRemoved_rightContainer(container_registry):
variant_node = MagicMock()
variant_node.variant_name = "variant_1"
variant_node.machine.has_machine_quality = True
variant_node.machine.quality_definition = "machine_1"
with patch("cura.Machines.MaterialNode.QualityNode"):
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance",MagicMock(return_value=container_registry)):
node = MaterialNode("material_1", variant_node)
container = createMockedInstanceContainer("material_1")
variant_node.materials = {"material_1": MagicMock()}
node._onRemoved(container)
assert "material_1" not in variant_node.materials