p59342861
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

UT: Add reset_to_default() to InMemoryAgentConfigurationRepository

This commit is contained in:
Mike Salvatore 2022-06-30 11:35:29 -04:00
parent e293639a22
commit 41b1c56fd0
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
monkey/tests/monkey_island/in_memory_agent_configuration_repository.py
View File

@ -6,10 +6,14 @@ from monkey_island.cc.repository import IAgentConfigurationRepository
class InMemoryAgentConfigurationRepository(IAgentConfigurationRepository): class InMemoryAgentConfigurationRepository(IAgentConfigurationRepository):
def __init__(self): def __init__(self):
self._configuration = AgentConfiguration.from_mapping(AGENT_CONFIGURATION) self._default_configuration = AgentConfiguration.from_mapping(AGENT_CONFIGURATION)
self._configuration = self._default_configuration
def get_configuration(self): def get_configuration(self):
return self._configuration return self._configuration
def store_configuration(self, agent_configuration): def store_configuration(self, agent_configuration):
self._configuration = agent_configuration self._configuration = agent_configuration
def reset_to_default(self):
self._configuration = self._default_configuration