diff --git a/monkey/monkey_island/cc/repository/i_simulation_repository.py b/monkey/monkey_island/cc/repository/i_simulation_repository.py index 2b41181c8..17c5ad9c9 100644 --- a/monkey/monkey_island/cc/repository/i_simulation_repository.py +++ b/monkey/monkey_island/cc/repository/i_simulation_repository.py @@ -1,6 +1,7 @@ from abc import ABC from monkey_island.cc.models import Simulation +from monkey_island.cc.services.mode.mode_enum import IslandModeEnum class ISimulationRepository(ABC): @@ -9,3 +10,22 @@ class ISimulationRepository(ABC): def get_simulation(self): pass + + def get_mode(self) -> IslandModeEnum: + """ + Get's the island's current mode + + :return The island's current mode + :raises RetrievalError: If the mode could not be retrieved + """ + pass + + def set_mode(self, mode: IslandModeEnum): + """ + Set the island's mode + + :param mode: The island's new mode + :raises StorageError: If the mode could not be saved + """ + + pass