mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-06-04 11:14:21 +08:00
8179dfc412
We don't want to use it outside of the mapping. This mapping should be transparent. We are still using it from our tests though but that's fine. Tests are allowed to touch private fields. Contributes to issue CURA-6793.
93 lines
5.3 KiB
Python
93 lines
5.3 KiB
Python
# 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 UM.Settings.DefinitionContainer import DefinitionContainer
|
|
from cura.Machines.ContainerTree import ContainerTree
|
|
from cura.Settings.GlobalStack import GlobalStack
|
|
|
|
|
|
def createMockedStack(definition_id: str):
|
|
result = MagicMock(spec = GlobalStack)
|
|
result.definition.getId = MagicMock(return_value = definition_id)
|
|
|
|
extruder_left_mock = MagicMock()
|
|
extruder_left_mock.variant.getName = MagicMock(return_value = definition_id + "_left_variant_name")
|
|
extruder_left_mock.material.getMetaDataEntry = MagicMock(return_value = definition_id + "_left_material_base_file")
|
|
extruder_left_mock.isEnabled = True
|
|
extruder_right_mock = MagicMock()
|
|
extruder_right_mock.variant.getName = MagicMock(return_value = definition_id + "_right_variant_name")
|
|
extruder_right_mock.material.getMetaDataEntry = MagicMock(return_value = definition_id + "_right_material_base_file")
|
|
extruder_right_mock.isEnabled = True
|
|
extruder_list = [extruder_left_mock, extruder_right_mock]
|
|
result.extruderList = extruder_list
|
|
return result
|
|
|
|
|
|
@pytest.fixture
|
|
def container_registry():
|
|
result = MagicMock()
|
|
result.findContainerStacks = MagicMock(return_value = [createMockedStack("machine_1"), createMockedStack("machine_2")])
|
|
result.findContainersMetadata = lambda id: [{"id": id}] if id in {"machine_1", "machine_2"} else []
|
|
return result
|
|
|
|
@pytest.fixture
|
|
def application():
|
|
return MagicMock(getGlobalContainerStack = MagicMock(return_value = createMockedStack("current_global_stack")))
|
|
|
|
|
|
def test_containerTreeInit(container_registry):
|
|
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value = container_registry)):
|
|
container_tree = ContainerTree()
|
|
|
|
assert "machine_1" in container_tree.machines
|
|
assert "machine_2" in container_tree.machines
|
|
|
|
def test_getCurrentQualityGroupsNoGlobalStack(container_registry):
|
|
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value = container_registry)):
|
|
with patch("cura.CuraApplication.CuraApplication.getInstance", MagicMock(return_value = MagicMock(getGlobalContainerStack = MagicMock(return_value = None)))):
|
|
container_tree = ContainerTree()
|
|
result = container_tree.getCurrentQualityGroups()
|
|
|
|
assert len(result) == 0
|
|
|
|
def test_getCurrentQualityGroups(container_registry, application):
|
|
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value = container_registry)):
|
|
container_tree = ContainerTree()
|
|
container_tree.machines._machines["current_global_stack"] = MagicMock() # Mock so that we can track whether the getQualityGroups function gets called with correct parameters.
|
|
|
|
with patch("cura.CuraApplication.CuraApplication.getInstance", MagicMock(return_value = application)):
|
|
result = container_tree.getCurrentQualityGroups()
|
|
|
|
# As defined in the fixture for application.
|
|
expected_variant_names = ["current_global_stack_left_variant_name", "current_global_stack_right_variant_name"]
|
|
expected_material_base_files = ["current_global_stack_left_material_base_file", "current_global_stack_right_material_base_file"]
|
|
expected_is_enabled = [True, True]
|
|
|
|
container_tree.machines["current_global_stack"].getQualityGroups.assert_called_with(expected_variant_names, expected_material_base_files, expected_is_enabled)
|
|
assert result == container_tree.machines["current_global_stack"].getQualityGroups.return_value
|
|
|
|
def test_getCurrentQualityChangesGroupsNoGlobalStack(container_registry):
|
|
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value = container_registry)):
|
|
with patch("cura.CuraApplication.CuraApplication.getInstance", MagicMock(return_value = MagicMock(getGlobalContainerStack = MagicMock(return_value = None)))):
|
|
container_tree = ContainerTree()
|
|
result = container_tree.getCurrentQualityChangesGroups()
|
|
|
|
assert len(result) == 0
|
|
|
|
def test_getCurrentQualityChangesGroups(container_registry, application):
|
|
with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value = container_registry)):
|
|
container_tree = ContainerTree()
|
|
container_tree.machines._machines["current_global_stack"] = MagicMock() # Mock so that we can track whether the getQualityGroups function gets called with correct parameters.
|
|
|
|
with patch("cura.CuraApplication.CuraApplication.getInstance", MagicMock(return_value = application)):
|
|
result = container_tree.getCurrentQualityChangesGroups()
|
|
|
|
# As defined in the fixture for application.
|
|
expected_variant_names = ["current_global_stack_left_variant_name", "current_global_stack_right_variant_name"]
|
|
expected_material_base_files = ["current_global_stack_left_material_base_file", "current_global_stack_right_material_base_file"]
|
|
expected_is_enabled = [True, True]
|
|
|
|
container_tree.machines["current_global_stack"].getQualityChangesGroups.assert_called_with(expected_variant_names, expected_material_base_files, expected_is_enabled)
|
|
assert result == container_tree.machines["current_global_stack"].getQualityChangesGroups.return_value |