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

@ -19,7 +19,7 @@ mocked_intent_metadata = [
mocked_qualitygroup_metadata = {
"normal": QualityGroup("um3_aa4_pla_normal", "normal"),
"abnorm": QualityGroup("um3_aa4_pla_abnorm", "abnorm")} # type:Dict[str, QualityGroup]
"abnorm": QualityGroup("um3_aa4_pla_abnorm", "abnorm")} # type: Dict[str, QualityGroup]
@pytest.fixture
def mock_container_tree() -> MagicMock:
@ -33,6 +33,7 @@ def mock_container_tree() -> MagicMock:
"generic_pla": MagicMock(
qualities = {
"um3_aa4_pla_normal": MagicMock(
quality_type = "normal",
intents = {
"smooth": MagicMock(
intent_category = "smooth",
@ -41,6 +42,7 @@ def mock_container_tree() -> MagicMock:
}
),
"um3_aa4_pla_abnorm": MagicMock(
quality_type = "abnorm",
intents = {
"strong": MagicMock(
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.quality = MockContainer({"id": "um3_aa4_pla_normal"})
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.variant = MockContainer({"name": "AA 0.4"})
extruder_stack_b.quality = MockContainer({"id": "um3_aa4_pla_normal"})
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]
application.getGlobalContainerStack = MagicMock(return_value = global_stack)