p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

UT: Use default config in InMemoryAgentConfigurationRepository

This commit is contained in:
Mike Salvatore 2022-07-22 11:49:04 -04:00
parent c86b637598
commit f321d72dc5
1 changed files with 2 additions and 4 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

@ -1,12 +1,10 @@
from tests.common.example_agent_configuration import AGENT_CONFIGURATION from common.configuration import DEFAULT_AGENT_CONFIGURATION
from common.configuration.agent_configuration import AgentConfiguration
from monkey_island.cc.repository import IAgentConfigurationRepository from monkey_island.cc.repository import IAgentConfigurationRepository
class InMemoryAgentConfigurationRepository(IAgentConfigurationRepository): class InMemoryAgentConfigurationRepository(IAgentConfigurationRepository):
def __init__(self): def __init__(self):
self._default_configuration = AgentConfiguration.from_mapping(AGENT_CONFIGURATION) self._default_configuration = DEFAULT_AGENT_CONFIGURATION
self._configuration = self._default_configuration self._configuration = self._default_configuration
def get_configuration(self): def get_configuration(self):