From 412a58f1f2cb4b0bdb010f6b5a8ffebe903bf7a1 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Wed, 28 Sep 2022 13:30:01 -0400 Subject: [PATCH] UT: Use InMemoryFileRepository in test_file_agent_log_repository.py --- .../cc/repository/test_file_agent_log_repository.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/monkey/tests/unit_tests/monkey_island/cc/repository/test_file_agent_log_repository.py b/monkey/tests/unit_tests/monkey_island/cc/repository/test_file_agent_log_repository.py index c36b43545..6409e9eff 100644 --- a/monkey/tests/unit_tests/monkey_island/cc/repository/test_file_agent_log_repository.py +++ b/monkey/tests/unit_tests/monkey_island/cc/repository/test_file_agent_log_repository.py @@ -3,7 +3,7 @@ from unittest.mock import MagicMock from uuid import UUID import pytest -from tests.monkey_island import OpenErrorFileRepository, SingleFileRepository +from tests.monkey_island import InMemoryFileRepository, OpenErrorFileRepository from monkey_island.cc.repository import ( FileAgentLogRepository, @@ -19,7 +19,7 @@ AGENT_ID = UUID("6bfd8b64-43d8-4449-8c70-d898aca74ad8") @pytest.fixture def repository() -> IAgentLogRepository: - return FileAgentLogRepository(SingleFileRepository()) + return FileAgentLogRepository(InMemoryFileRepository()) def test_store_agent_log(repository: IAgentLogRepository):