diff --git a/monkey/tests/unit_tests/monkey_island/cc/services/test_island_mode_service.py b/monkey/tests/unit_tests/monkey_island/cc/services/test_island_mode_service.py new file mode 100644 index 000000000..79074b213 --- /dev/null +++ b/monkey/tests/unit_tests/monkey_island/cc/services/test_island_mode_service.py @@ -0,0 +1,16 @@ +import pytest +from tests.monkey_island import InMemorySimulationRepository + +from monkey_island.cc.services import IslandModeService +from monkey_island.cc.services.mode.mode_enum import IslandModeEnum + + +@pytest.fixture +def island_mode_service(): + return IslandModeService(InMemorySimulationRepository()) + + +@pytest.mark.parametrize("mode", list(IslandModeEnum)) +def test_set_mode(island_mode_service, mode): + island_mode_service.set_mode(mode) + assert island_mode_service.get_mode() == mode