Fix missing intent in mock and missing quality_type

These are used by the code. Otherwise there would be an AttributeError of 'intent' not existing, and the quality type in the quality node wouldn't match because it's a magic mock.
This commit is contained in:
Ghostkeeper 2019-09-27 15:13:23 +02:00
parent 7a1850a87b
commit 97eaeab3bd
No known key found for this signature in database
GPG key ID: 86BEF881AE2CF276

View file

@ -33,6 +33,7 @@ def mock_container_tree() -> MagicMock:
"generic_pla": MagicMock( "generic_pla": MagicMock(
qualities = { qualities = {
"um3_aa4_pla_normal": MagicMock( "um3_aa4_pla_normal": MagicMock(
quality_type = "normal",
intents = { intents = {
"smooth": MagicMock( "smooth": MagicMock(
intent_category = "smooth", intent_category = "smooth",
@ -41,6 +42,7 @@ def mock_container_tree() -> MagicMock:
} }
), ),
"um3_aa4_pla_abnorm": MagicMock( "um3_aa4_pla_abnorm": MagicMock(
quality_type = "abnorm",
intents = { intents = {
"strong": MagicMock( "strong": MagicMock(
intent_category = "strong", intent_category = "strong",
@ -104,10 +106,12 @@ def doSetup(application, extruder_manager, container_registry, global_stack) ->
extruder_stack_a.variant = MockContainer({"name": "AA 0.4"}) extruder_stack_a.variant = MockContainer({"name": "AA 0.4"})
extruder_stack_a.quality = MockContainer({"id": "um3_aa4_pla_normal"}) extruder_stack_a.quality = MockContainer({"id": "um3_aa4_pla_normal"})
extruder_stack_a.material = MockContainer({"base_file": "generic_pla"}) extruder_stack_a.material = MockContainer({"base_file": "generic_pla"})
extruder_stack_a.intent = MockContainer({"id": "empty_intent", "intent_category": "default"})
extruder_stack_b = MockContainer({"id": "Extruder II: Plastic Boogaloo"}) extruder_stack_b = MockContainer({"id": "Extruder II: Plastic Boogaloo"})
extruder_stack_b.variant = MockContainer({"name": "AA 0.4"}) extruder_stack_b.variant = MockContainer({"name": "AA 0.4"})
extruder_stack_b.quality = MockContainer({"id": "um3_aa4_pla_normal"}) extruder_stack_b.quality = MockContainer({"id": "um3_aa4_pla_normal"})
extruder_stack_b.material = MockContainer({"base_file": "generic_pla"}) extruder_stack_b.material = MockContainer({"base_file": "generic_pla"})
extruder_stack_b.intent = MockContainer({"id": "empty_intent", "intent_category": "default"})
global_stack.extruderList = [extruder_stack_a, extruder_stack_b] global_stack.extruderList = [extruder_stack_a, extruder_stack_b]
application.getGlobalContainerStack = MagicMock(return_value = global_stack) application.getGlobalContainerStack = MagicMock(return_value = global_stack)