mirror of
https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura
synced 2025-04-20 20:59:48 +08:00
63 lines
2.4 KiB
Python
63 lines
2.4 KiB
Python
#Todo: Write tests
|
|
|
|
import pytest
|
|
|
|
from cura.MachineAction import MachineAction
|
|
from cura.MachineActionManager import MachineActionManager, NotUniqueMachineAction, UnknownMachineAction
|
|
|
|
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_action")
|
|
test_action_2 = MachineAction(key = "test_action_2")
|
|
test_machine = Machine("test_machine")
|
|
machine_manager.addMachineAction(test_action)
|
|
machine_manager.addMachineAction(test_action_2)
|
|
|
|
assert machine_manager.getMachineAction("test_action") == test_action
|
|
assert machine_manager.getMachineAction("key_that_doesnt_exist") is None
|
|
|
|
# Adding the same machine action is not allowed.
|
|
with pytest.raises(NotUniqueMachineAction):
|
|
machine_manager.addMachineAction(test_action)
|
|
|
|
# Check that the machine has no supported actions yet.
|
|
assert machine_manager.getSupportedActions(test_machine) == set()
|
|
|
|
# Check if adding a supported action works.
|
|
machine_manager.addSupportedAction(test_machine, "test_action")
|
|
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}
|
|
|
|
# Check if adding multiple supported actions works.
|
|
machine_manager.addSupportedAction(test_machine, "test_action_2")
|
|
assert machine_manager.getSupportedActions(test_machine) == {test_action, test_action_2}
|
|
|
|
# Check that the machine has no required actions yet.
|
|
assert machine_manager.getRequiredActions(test_machine) == set()
|
|
|
|
## Ensure that only known actions can be added.
|
|
with pytest.raises(UnknownMachineAction):
|
|
machine_manager.addRequiredAction(test_machine, "key_that_doesnt_exist")
|
|
|
|
## Check if adding single required action works
|
|
machine_manager.addRequiredAction(test_machine, "test_action")
|
|
assert machine_manager.getRequiredActions(test_machine) == {test_action}
|
|
|
|
# Check if adding multiple required actions works.
|
|
machine_manager.addRequiredAction(test_machine, "test_action_2")
|
|
assert machine_manager.getRequiredActions(test_machine) == {test_action, test_action_2}
|
|
|