From a24bf921c1302e9fccc2b9cafb318930099705d7 Mon Sep 17 00:00:00 2001
From: Mike Salvatore <mike.s.salvatore@gmail.com>
Date: Fri, 1 Jul 2022 09:19:51 -0400
Subject: [PATCH] Island: Add {get,set}_mode() to ISimulationRepository

---
 .../cc/repository/i_simulation_repository.py  | 20 +++++++++++++++++++
 1 file changed, 20 insertions(+)

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