1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
|
# Copyright (c) 2019 Ultimaker B.V.
# Cura is released under the terms of the LGPLv3 or higher.
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_1")
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
def test_onRemoved_wrongContainer(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"
variant_node.materials = {"material_1": MagicMock()}
with patch("cura.Machines.MaterialNode.QualityNode"):
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance",MagicMock(return_value=container_registry)):
node = MaterialNode("material_1", variant_node)
container = createMockedInstanceContainer("material_2")
node._onRemoved(container)
assert "material_1" in variant_node.materials
def test_onRemoved_rightContainer(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)
container = createMockedInstanceContainer("material_1")
variant_node.materials = {"material_1": MagicMock()}
node._onRemoved(container)
assert "material_1" not in variant_node.materials
def test_onMetadataChanged(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)
# We only do this now since we do want it to be constructed but not actually re-evaluated.
node._loadAll = MagicMock()
container = createMockedInstanceContainer("material_1")
container.getMetaData = MagicMock(return_value = {"base_file": "new_base_file", "material": "new_material_type", "GUID": "new_guid"})
node._onMetadataChanged(container)
assert node.material_type == "new_material_type"
assert node.guid == "new_guid"
assert node.base_file == "new_base_file"
def test_onMetadataChanged_wrongContainer(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)
# We only do this now since we do want it to be constructed but not actually re-evaluated.
node._loadAll = MagicMock()
container = createMockedInstanceContainer("material_2")
container.getMetaData = MagicMock(
return_value={"base_file": "new_base_file", "material": "new_material_type", "GUID": "new_guid"})
node._onMetadataChanged(container)
assert node.material_type == "test_material_type"
assert node.guid == "omg zomg"
assert node.base_file == "material_1"
|
