#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}