From f321d72dc55e24acc351c1dfb7f0378d3434db0c Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Fri, 22 Jul 2022 11:49:04 -0400 Subject: [PATCH] UT: Use default config in InMemoryAgentConfigurationRepository --- .../in_memory_agent_configuration_repository.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/monkey/tests/monkey_island/in_memory_agent_configuration_repository.py b/monkey/tests/monkey_island/in_memory_agent_configuration_repository.py index cf97ac08a..94bba3ddc 100644 --- a/monkey/tests/monkey_island/in_memory_agent_configuration_repository.py +++ b/monkey/tests/monkey_island/in_memory_agent_configuration_repository.py @@ -1,12 +1,10 @@ -from tests.common.example_agent_configuration import AGENT_CONFIGURATION - -from common.configuration.agent_configuration import AgentConfiguration +from common.configuration import DEFAULT_AGENT_CONFIGURATION from monkey_island.cc.repository import IAgentConfigurationRepository class InMemoryAgentConfigurationRepository(IAgentConfigurationRepository): def __init__(self): - self._default_configuration = AgentConfiguration.from_mapping(AGENT_CONFIGURATION) + self._default_configuration = DEFAULT_AGENT_CONFIGURATION self._configuration = self._default_configuration def get_configuration(self):