diff --git a/monkey/tests/monkey_island/__init__.py b/monkey/tests/monkey_island/__init__.py index 8f8ae7404..8f3654a6e 100644 --- a/monkey/tests/monkey_island/__init__.py +++ b/monkey/tests/monkey_island/__init__.py @@ -1,4 +1,4 @@ from .single_file_repository import SingleFileRepository from .mock_file_repository import MockFileRepository, FILE_CONTENTS, FILE_NAME from .open_error_file_repository import OpenErrorFileRepository -from .in_memory_file_agent_configuration_repository import InMemoryFileAgentConfigurationRepository +from .in_memory_agent_configuration_repository import InMemoryAgentConfigurationRepository diff --git a/monkey/tests/monkey_island/in_memory_file_agent_configuration_repository.py b/monkey/tests/monkey_island/in_memory_agent_configuration_repository.py similarity index 86% rename from monkey/tests/monkey_island/in_memory_file_agent_configuration_repository.py rename to monkey/tests/monkey_island/in_memory_agent_configuration_repository.py index 955e78878..e737d645c 100644 --- a/monkey/tests/monkey_island/in_memory_file_agent_configuration_repository.py +++ b/monkey/tests/monkey_island/in_memory_agent_configuration_repository.py @@ -4,7 +4,7 @@ from common.configuration.agent_configuration import AgentConfigurationSchema from monkey_island.cc.repository import IAgentConfigurationRepository -class InMemoryFileAgentConfigurationRepository(IAgentConfigurationRepository): +class InMemoryAgentConfigurationRepository(IAgentConfigurationRepository): def __init__(self): self._configuration = AgentConfigurationSchema().load(AGENT_CONFIGURATION) diff --git a/monkey/tests/unit_tests/monkey_island/cc/resources/test_agent_configuration.py b/monkey/tests/unit_tests/monkey_island/cc/resources/test_agent_configuration.py index 5cac629e0..874500b81 100644 --- a/monkey/tests/unit_tests/monkey_island/cc/resources/test_agent_configuration.py +++ b/monkey/tests/unit_tests/monkey_island/cc/resources/test_agent_configuration.py @@ -3,7 +3,7 @@ import json import pytest from tests.common import StubDIContainer from tests.common.example_agent_configuration import AGENT_CONFIGURATION -from tests.monkey_island import InMemoryFileAgentConfigurationRepository +from tests.monkey_island import InMemoryAgentConfigurationRepository from tests.unit_tests.monkey_island.conftest import get_url_for_resource from monkey_island.cc.repository import IAgentConfigurationRepository @@ -14,7 +14,7 @@ from monkey_island.cc.resources.agent_configuration import AgentConfiguration def flask_client(build_flask_client): container = StubDIContainer() - container.register(IAgentConfigurationRepository, InMemoryFileAgentConfigurationRepository) + container.register(IAgentConfigurationRepository, InMemoryAgentConfigurationRepository) with build_flask_client(container) as flask_client: yield flask_client