mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-05-22 12:39:22 +08: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.
49 lines
1.7 KiB
Python
49 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) == 2
|
|
assert "intent_3" in node.intents
|
|
assert "intent_4" in node.intents |