diff --git a/tests/Machines/TestContainerTree.py b/tests/Machines/TestContainerTree.py index 5d269fc6a9..e82ce2384d 100644 --- a/tests/Machines/TestContainerTree.py +++ b/tests/Machines/TestContainerTree.py @@ -10,15 +10,30 @@ import cura.CuraApplication # DEBUG! 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_dict = {"1": extruder_right_mock, "0": extruder_left_mock} + result.extruders = extruder_dict return result @pytest.fixture def container_registry(): result = MagicMock() - result.findContainerStacks = MagicMock(return_value=[createMockedStack("machine_1"), createMockedStack("machine_2")]) + result.findContainerStacks = MagicMock(return_value = [createMockedStack("machine_1"), createMockedStack("machine_2")]) 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)): @@ -32,7 +47,7 @@ def test_newMachineAdded(container_registry): mocked_definition_container = MagicMock(spec=DefinitionContainer) mocked_definition_container.getId = MagicMock(return_value = "machine_3") - with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value=container_registry)): + with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value = container_registry)): container_tree = ContainerTree() # machine_3 shouldn't be there, as on init it wasn't in the registry assert "machine_3" not in container_tree.machines @@ -46,10 +61,10 @@ def test_newMachineAdded(container_registry): def test_alreadyKnownMachineAdded(container_registry): - mocked_definition_container = MagicMock(spec=DefinitionContainer) - mocked_definition_container.getId = MagicMock(return_value="machine_2") + mocked_definition_container = MagicMock(spec = DefinitionContainer) + mocked_definition_container.getId = MagicMock(return_value = "machine_2") - with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value=container_registry)): + with patch("UM.Settings.ContainerRegistry.ContainerRegistry.getInstance", MagicMock(return_value = container_registry)): container_tree = ContainerTree() assert len(container_tree.machines) == 2 @@ -63,4 +78,19 @@ def test_getCurrentQualityGroupsNoGlobalStack(container_registry): container_tree = ContainerTree() result = container_tree.getCurrentQualityGroups() - assert len(result) == 0 \ No newline at end of file + 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["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) \ No newline at end of file