mirror of
https://github.com/Ultimaker/Cura.git
synced 2025-11-02 20:52:20 -07:00
Add tests for ExtruderManager
This commit is contained in:
parent
1c93dffc22
commit
9ad8d91ff9
3 changed files with 51 additions and 18 deletions
31
tests/TestExtruderManager.py
Normal file
31
tests/TestExtruderManager.py
Normal file
|
|
@ -0,0 +1,31 @@
|
|||
|
||||
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
|
||||
Loading…
Add table
Add a link
Reference in a new issue