From e374341ce1804f1a3873674cc13fa675c4513879 Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Wed, 28 Sep 2022 20:01:13 +0530 Subject: [PATCH] UT: Fix tests in test_file_agent_log_repository.py --- .../cc/repository/test_file_agent_log_repository.py | 6 +++--- 1 file changed, 3 insertions(+), 3 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 035903e18..d49d0957d 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 uuid import UUID import pytest from tests.monkey_island import OpenErrorFileRepository, SingleFileRepository -from monkey_island.cc.repository import FileAgentLogRepository, RetrievalError, UnknownRecordError +from monkey_island.cc.repository import FileAgentLogRepository, RetrievalError LOG_CONTENTS = "lots of useful information" AGENT_ID = UUID("6bfd8b64-43d8-4449-8c70-d898aca74ad8") @@ -22,7 +22,7 @@ def test_store_agent_log(repository): def test_get_agent_log__unknown_record_error(repository): - with pytest.raises(UnknownRecordError): + with pytest.raises(RetrievalError): repository.get_agent_log(AGENT_ID) @@ -35,5 +35,5 @@ def test_get_agent_log__retrieval_error(): def test_reset_agent_logs(repository): repository.upsert_agent_log(AGENT_ID, LOG_CONTENTS) repository.reset() - with pytest.raises(UnknownRecordError): + with pytest.raises(RetrievalError): repository.get_agent_log(AGENT_ID)