from unittest.mock import MagicMock def createMockedExtruder(extruder_id): extruder = MagicMock() extruder.getId = MagicMock(return_value = extruder_id) return extruder def test_getAllExtruderSettings(extruder_manager): extruder_1 = createMockedExtruder("extruder_1") extruder_1.getProperty = MagicMock(return_value ="beep") extruder_2 = createMockedExtruder("extruder_2") extruder_2.getProperty = MagicMock(return_value="zomg") extruder_manager.getActiveExtruderStacks = MagicMock(return_value = [extruder_1, extruder_2]) assert extruder_manager.getAllExtruderSettings("whatever", "value") == ["beep", "zomg"]