Island: raise RemovalError in Local...FileRepository.delete_all_files()

monkey/monkey_island/cc/repository/file_storage/i_file_repository.py

@@ -56,5 +56,7 @@ class IFileRepository(metaclass=abc.ABCMeta):
     def delete_all_files(self):
         """
         Delete all files that have been stored using this service.
+
+        :raises RemovalError: If an error was encountered while attempting to remove a file
         """
         pass

monkey/monkey_island/cc/repository/file_storage/local_storage_file_repository.py

@@ -82,6 +82,9 @@ class LocalStorageFileRepository(IFileRepository):
         return safe_file_path
 
     def delete_all_files(self):
-        for file in get_all_regular_files_in_directory(self._storage_directory):
-            logger.debug(f"Deleting {file}")
-            file.unlink()
+        try:
+            for file in get_all_regular_files_in_directory(self._storage_directory):
+                logger.debug(f"Deleting {file}")
+                file.unlink()
+        except Exception as err:
+            raise RemovalError(f"Error while attempting to clear the repository: {err}")