added reset cleanup functionality (#211)

mipengine/controller/cleaner.py

@@ -120,6 +120,11 @@ def _get_node_info_by_id(self, node_id: str) -> _NodeInfoDTO:
 
         raise KeyError(f"Node with id '{node_id}' is not currently available.")
 
+    # This is only supposed to be called from a test.
+    # In all other circumstances cleanup should not be reset manually
+    def _reset_cleanup(self):
+        self._cleanup_file_processor._delete_cleanup_file()
+
 
 def _create_node_task_handler(node_info: _NodeInfoDTO) -> NodeTasksHandlerCelery:
     return NodeTasksHandlerCelery(
@@ -233,3 +238,9 @@ def _write_to_cleanup_file(self, toml_string: str):
         # renaming is atomic. Of course if more Controllers are spawn the file is very
         # likely to become corrupted
         os.rename(self._cleanup_file_tmp_path, self._cleanup_file_path)
+
+    def _delete_cleanup_file(self):
+        cleanup_file_path = Path(
+            controller_config.cleanup.contextids_cleanup_folder
+        ).joinpath(Path(CONTEXT_ID_CLEANUP_FILE))
+        cleanup_file_path.unlink()

tests/standalone_tests/test_cleanup_after_algorithm_execution.py

@@ -234,6 +234,7 @@ async def test_cleanup_after_uninterrupted_algorithm_execution(
         await asyncio.sleep(2)
 
     # Start the cleanup loop
+    controller._cleaner._reset_cleanup()
     controller.start_cleanup_loop()
     request_id = get_a_uniqueid()
     context_id = get_a_uniqueid()
@@ -383,6 +384,7 @@ async def test_cleanup_after_uninterrupted_algorithm_execution_without_releasing
         await asyncio.sleep(2)
 
     # Start the cleanup loop
+    controller._cleaner._reset_cleanup()
     controller.start_cleanup_loop()
 
     request_id = get_a_uniqueid()
@@ -528,6 +530,7 @@ async def test_cleanup_rabbitmq_down_algorithm_execution(
         await asyncio.sleep(2)
 
     # Start the cleanup loop
+    controller._cleaner._reset_cleanup()
     controller.start_cleanup_loop()
 
     request_id = get_a_uniqueid()
@@ -699,6 +702,7 @@ async def test_cleanup_node_service_down_algorithm_execution(
         await asyncio.sleep(2)
 
     # Start the cleanup loop
+    controller._cleaner._reset_cleanup()
     controller.start_cleanup_loop()
 
     request_id = get_a_uniqueid()
@@ -862,6 +866,7 @@ async def test_cleanup_controller_restart(
         await asyncio.sleep(2)
 
     # Start the cleanup loop
+    controller._cleaner._reset_cleanup()
     controller.start_cleanup_loop()
 
     request_id = get_a_uniqueid()