Added test for activeMaterialId's

This commit is contained in:
Jaime van Kessel 2019-07-19 13:04:54 +02:00
parent 9940023816
commit 2ee6f4af15

View file

@ -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"}