From 2ee6f4af157a6d4a070f18316ea3f326515f5667 Mon Sep 17 00:00:00 2001 From: Jaime van Kessel Date: Fri, 19 Jul 2019 13:04:54 +0200 Subject: [PATCH] Added test for activeMaterialId's --- tests/TestMachineManager.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/tests/TestMachineManager.py b/tests/TestMachineManager.py index ab89fdcc29..25e8035487 100644 --- a/tests/TestMachineManager.py +++ b/tests/TestMachineManager.py @@ -94,3 +94,24 @@ def test_totalNumberOfSettings(machine_manager): registry.findDefinitionContainers = MagicMock(return_value = [mocked_definition]) with patch("cura.Settings.CuraContainerRegistry.CuraContainerRegistry.getInstance", MagicMock(return_value=registry)): assert machine_manager.totalNumberOfSettings == 3 + + +def createMockedExtruder(extruder_id): + extruder = MagicMock() + extruder.getId = MagicMock(return_value = extruder_id) + return extruder + + +def createMockedMaterial(material_id): + material = MagicMock() + material.getId = MagicMock(return_value=material_id) + return material + + +def test_allActiveMaterialIds(machine_manager, extruder_manager): + extruder_1 = createMockedExtruder("extruder_1") + extruder_2 = createMockedExtruder("extruder_2") + extruder_1.material = createMockedMaterial("material_1") + extruder_2.material = createMockedMaterial("material_2") + extruder_manager.getActiveExtruderStacks = MagicMock(return_value = [extruder_1, extruder_2]) + assert machine_manager.allActiveMaterialIds == {"extruder_1": "material_1", "extruder_2": "material_2"}