From 4a91777019c1ccb8481b6cb345713c991fa60334 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Fri, 1 Jul 2022 11:38:20 -0400 Subject: [PATCH] Island: Add tests for IslandModeService --- .../cc/services/test_island_mode_service.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 monkey/tests/unit_tests/monkey_island/cc/services/test_island_mode_service.py 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