mirror of
https://github.com/Ultimaker/Cura.git
synced 2025-07-08 07:27:29 -06:00
80baeb9873
Among the machines, variants, materials, qualities and intents, only machines and materials can ever be added during runtime. For the rest, we don't need to listen to these signals. Contributes to issue CURA-6600.
74 lines
No EOL
3 KiB
Python
74 lines
No EOL
3 KiB
Python
from unittest.mock import patch, MagicMock
|
|
import pytest
|
|
|
|
from cura.Machines.MaterialNode import MaterialNode
|
|
|
|
instance_container_metadata_dict = {"fdmprinter": {"no_variant": [{"id": "quality_1", "material": "material_1"}]},
|
|
"machine_1": {"variant_1": {"material_1": [{"id": "quality_2", "material": "material_1"}, {"id": "quality_3","material": "material_1"}]}}}
|
|
|
|
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):
|
|
variant = kwargs.get("variant")
|
|
definition = kwargs.get("definition")
|
|
type = kwargs.get("type")
|
|
material = kwargs.get("material")
|
|
if material is not None and variant is not None:
|
|
definition_dict = instance_container_metadata_dict.get(definition)
|
|
variant_dict = definition_dict.get(variant)
|
|
material_dict = variant_dict.get(material)
|
|
return material_dict
|
|
if type == "quality":
|
|
if variant is None:
|
|
return instance_container_metadata_dict.get(definition).get("no_variant")
|
|
else:
|
|
return instance_container_metadata_dict.get(definition).get(variant).get("material_1")
|
|
if definition is None:
|
|
return [{"id": "material_1", "material": "material_1"}]
|
|
return instance_container_metadata_dict.get(definition).get("no_variant")
|
|
|
|
|
|
@pytest.fixture
|
|
def container_registry():
|
|
result = MagicMock()
|
|
result.findInstanceContainersMetadata = MagicMock(side_effect=getInstanceContainerSideEffect)
|
|
result.findContainersMetadata = MagicMock(return_value = [{"base_file": "material_1"}])
|
|
return result
|
|
|
|
|
|
def test_materialNodeInit_noMachineQuality(container_registry):
|
|
variant_node = MagicMock()
|
|
variant_node.variant_name = "variant_1"
|
|
variant_node.machine.has_machine_quality = False
|
|
with patch("cura.Machines.MaterialNode.QualityNode"):
|
|
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value=container_registry)):
|
|
node = MaterialNode("material_1", variant_node)
|
|
|
|
assert len(node.qualities) == 1
|
|
assert "quality_1" in node.qualities
|
|
|
|
|
|
def test_materialNodeInit_MachineQuality(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)
|
|
|
|
assert len(node.qualities) == 2
|
|
assert "quality_2" in node.qualities
|
|
assert "quality_3" in node.qualities |