mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-05-07 02:09:00 +08:00
0862fd493e
The PrintInformation test wasn't cleaning up after itself correclty. This left some stuff behind that the other tests were using. Since this is bad (as at that point tests can influence other tests), i've fixed that
96 lines
5.4 KiB
Python
96 lines
5.4 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)):
|
|
with patch("UM.Application.Application.getInstance"):
|
|
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)):
|
|
with patch("cura.CuraApplication.CuraApplication.getInstance", MagicMock(return_value=application)):
|
|
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.
|
|
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)):
|
|
with patch("cura.CuraApplication.CuraApplication.getInstance", MagicMock(return_value=application)):
|
|
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.
|
|
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 |