From 28e75319b02815d66eef3bee92eac40fb01ca67f Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Fri, 1 Jul 2022 09:35:38 -0400 Subject: [PATCH] Island: Make ISimulationRepository methods abstract --- .../monkey_island/cc/repository/i_simulation_repository.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/monkey/monkey_island/cc/repository/i_simulation_repository.py b/monkey/monkey_island/cc/repository/i_simulation_repository.py index 17c5ad9c9..cf2b06583 100644 --- a/monkey/monkey_island/cc/repository/i_simulation_repository.py +++ b/monkey/monkey_island/cc/repository/i_simulation_repository.py @@ -1,16 +1,19 @@ -from abc import ABC +from abc import ABC, abstractmethod from monkey_island.cc.models import Simulation from monkey_island.cc.services.mode.mode_enum import IslandModeEnum class ISimulationRepository(ABC): + @abstractmethod def save_simulation(self, simulation: Simulation): pass + @abstractmethod def get_simulation(self): pass + @abstractmethod def get_mode(self) -> IslandModeEnum: """ Get's the island's current mode @@ -20,6 +23,7 @@ class ISimulationRepository(ABC): """ pass + @abstractmethod def set_mode(self, mode: IslandModeEnum): """ Set the island's mode