mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-23 06:09:38 +08:00
50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
from unittest.mock import patch, MagicMock
|
|
import pytest
|
|
|
|
from cura.Machines.QualityNode import QualityNode
|
|
|
|
|
|
instance_container_metadata_dict = {"fdmprinter": {"variant_1": {"material_1": [{"id": "intent_1"}, {"id": "intent_2"}]}},
|
|
"machine_1": {"variant_2": {"material_2": [{"id": "intent_3"}, {"id": "intent_4"}]}}}
|
|
|
|
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):
|
|
|
|
definition_dict = instance_container_metadata_dict.get(kwargs["definition"])
|
|
variant_dict = definition_dict.get(kwargs["variant"])
|
|
return variant_dict.get(kwargs["material"])
|
|
|
|
@pytest.fixture
|
|
def container_registry():
|
|
result = MagicMock()
|
|
result.findInstanceContainersMetadata = MagicMock(side_effect=getInstanceContainerSideEffect)
|
|
return result
|
|
|
|
|
|
def test_qualityNode_machine_1(container_registry):
|
|
material_node = MagicMock()
|
|
material_node.variant.machine.quality_definition = "machine_1"
|
|
material_node.variant.variant_name = "variant_2"
|
|
material_node.base_file = "material_2"
|
|
|
|
with patch("cura.Machines.QualityNode.IntentNode"):
|
|
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value=container_registry)):
|
|
node = QualityNode("quality_1", material_node)
|
|
|
|
assert len(node.intents) == 3
|
|
assert "intent_3" in node.intents
|
|
assert "intent_4" in node.intents
|
|
assert "empty_intent" in node.intents |