Fix tests

This commit is contained in:
Jaime van Kessel 2019-08-29 17:07:36 +02:00
parent 784ab6e903
commit d9e94f5019
5 changed files with 31 additions and 14 deletions

View file

@ -11,6 +11,7 @@ metadata_dict = {}
def container_registry():
result = MagicMock()
result.findInstanceContainersMetadata = MagicMock(return_value = [{"id": "variant_1", "name": "Variant One", "quality_type": "normal"}, {"id": "variant_2", "name": "Variant Two", "quality_type": "great"}])
result.findContainersMetadata = MagicMock(return_value = [{"has_variants": True}])
return result
@ -26,8 +27,9 @@ def createMockedInstanceContainer():
def createMachineNode(container_id, container_registry):
with patch("cura.Machines.MachineNode.VariantNode"): # We're not testing the variant node here, so patch it out.
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value=container_registry)):
return MachineNode(container_id)
with patch("cura.Machines.MachineNode.QualityNode"):
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value=container_registry)):
return MachineNode(container_id)
def test_machineNodeInit(container_registry):