mirror of
https://github.com/Ultimaker/Cura.git
synced 2025-07-06 22:47:29 -06:00
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
#Todo: Write tests
|
|
|
|
import pytest
|
|
|
|
from cura.MachineAction import MachineAction
|
|
from cura.MachineActionManager import MachineActionManager, NotUniqueMachineAction
|
|
|
|
class Machine:
|
|
def __init__(self, key = ""):
|
|
self._key = key
|
|
|
|
def getKey(self):
|
|
return self._key
|
|
|
|
|
|
def test_addMachineAction():
|
|
|
|
machine_manager = MachineActionManager()
|
|
|
|
test_action = MachineAction(key = "test")
|
|
test_machine = Machine("test_machine")
|
|
machine_manager.addMachineAction(test_action)
|
|
|
|
assert machine_manager.getMachineAction("test") == test_action
|
|
|
|
# Adding the same machine action is not allowed.
|
|
with pytest.raises(NotUniqueMachineAction):
|
|
machine_manager.addMachineAction(test_action)
|
|
|
|
# Check if adding a supported action works.
|
|
machine_manager.addSupportedAction(test_machine, "test")
|
|
assert machine_manager.getSupportedActions(test_machine) == {test_action}
|
|
|
|
# Check that adding a unknown action doesn't change anything.
|
|
machine_manager.addSupportedAction(test_machine, "key_that_doesnt_exist")
|
|
assert machine_manager.getSupportedActions(test_machine) == {test_action}
|
|
|