mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-22 13:49:39 +08:00
74 lines
3.0 KiB
Python
74 lines
3.0 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", "material": "test_material_type", "GUID": "omg zomg"}])
|
|
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 |