diff --git a/monkey/tests/common/SingleFileRepository.py b/monkey/tests/common/SingleFileRepository.py deleted file mode 100644 index d00ee9ae9..000000000 --- a/monkey/tests/common/SingleFileRepository.py +++ /dev/null @@ -1,23 +0,0 @@ -import io -from typing import BinaryIO - -from monkey_island.cc.repository import FileRetrievalError, IFileRepository - - -class SingleFileRepository(IFileRepository): - def __init__(self): - self._file = None - - def save_file(self, unsafe_file_name: str, file_contents: BinaryIO): - self._file = io.BytesIO(file_contents.read()) - - def open_file(self, unsafe_file_name: str) -> BinaryIO: - if self._file is None: - raise FileRetrievalError() - return self._file - - def delete_file(self, unsafe_file_name: str): - self._file = None - - def delete_all_files(self): - self.delete_file("")