rename

Author: Andrei Neagu

packages/service-library/src/servicelib/long_running_tasks/_redis_store.py

@@ -112,7 +112,7 @@ async def delete_task_data(self, task_id: TaskId) -> None:
 
     # to cancel
 
-    async def mark_task_for_removal(self, task_id: TaskId) -> None:
+    async def mark_for_removal(self, task_id: TaskId) -> None:
         await handle_redis_returns_union_types(
             self._redis.hset(
                 self._get_redis_task_data_key(task_id),

packages/service-library/src/servicelib/long_running_tasks/task.py

@@ -503,7 +503,7 @@ async def remove_task(
 
         tracked_task = await self._get_tracked_task(task_id, with_task_context)
 
-        await self._tasks_data.mark_task_for_removal(tracked_task.task_id)
+        await self._tasks_data.mark_for_removal(tracked_task.task_id)
 
         if not wait_for_removal:
             return

packages/service-library/tests/long_running_tasks/test_long_running_tasks__redis_store.py

@@ -65,7 +65,7 @@ async def test_workflow(store: RedisStore, task_data: TaskData) -> None:
 
     assert await store.is_marked_for_removal(task_data.task_id) is False
 
-    await store.mark_task_for_removal(task_data.task_id)
+    await store.mark_for_removal(task_data.task_id)
 
     assert await store.is_marked_for_removal(task_data.task_id) is True
 
@@ -103,7 +103,7 @@ async def test_workflow_multiple_redis_stores_with_different_namespaces(
 
     for store in redis_stores:
         await store.add_task_data(task_data.task_id, task_data)
-        await store.mark_task_for_removal(task_data.task_id)
+        await store.mark_for_removal(task_data.task_id)
 
     marked_as_removed_task_data = deepcopy(task_data)
     marked_as_removed_task_data.marked_for_removal = True

packages/service-library/tests/long_running_tasks/test_long_running_tasks_client_long_running_manager.py

@@ -65,7 +65,7 @@ async def test_cleanup_namespace(
 ) -> None:
     # create entries in both sides
     await store.add_task_data(task_data.task_id, task_data)
-    await store.mark_task_for_removal(task_data.task_id)
+    await store.mark_for_removal(task_data.task_id)
 
     # entries exit
     marked_for_removal = deepcopy(task_data)

packages/service-library/tests/long_running_tasks/test_long_running_tasks_task.py

@@ -548,7 +548,7 @@ async def test__cancelled_tasks_worker_equivalent_of_cancellation_from_a_differe
         total_sleep=10,
         task_context=empty_context,
     )
-    await long_running_manager.tasks_manager._tasks_data.mark_task_for_removal(  # noqa: SLF001
+    await long_running_manager.tasks_manager._tasks_data.mark_for_removal(  # noqa: SLF001
         task_id
     )