mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-23 06:09:38 +08:00
32 lines
1.2 KiB
Python
32 lines
1.2 KiB
Python
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
|
|
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"]
|
|
|
|
|
|
def test_registerExtruder(extruder_manager):
|
|
extruder = createMockedExtruder("beep")
|
|
extruder.getMetaDataEntry = MagicMock(return_value = "0") # because the extruder position gets called
|
|
|
|
extruder_manager.extrudersChanged = MagicMock()
|
|
extruder_manager.registerExtruder(extruder, "zomg")
|
|
|
|
assert extruder_manager.extrudersChanged.emit.call_count == 1
|
|
|
|
# Doing it again should not trigger anything
|
|
extruder_manager.registerExtruder(extruder, "zomg")
|
|
assert extruder_manager.extrudersChanged.emit.call_count == 1
|