From e374341ce1804f1a3873674cc13fa675c4513879 Mon Sep 17 00:00:00 2001
From: Shreya Malviya <shreya.malviya@gmail.com>
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)