tests

Author: giancarloromeo

packages/celery-library/src/celery_library/client.py

@@ -54,7 +54,7 @@ async def submit_task(
             self._celery_app.send_task(
                 task_metadata.name,
                 task_id=task_id,
-                kwargs=task_params,
+                kwargs={"task_id": task_id} | task_params,
                 queue=task_metadata.queue.value,
             )
 

packages/celery-library/src/celery_library/task.py

@@ -44,7 +44,7 @@ def _async_task_wrapper(
     Callable[Concatenate[AbortableTask, P], R],
 ]:
     def decorator(
-        coro: Callable[Concatenate[AbortableTask, TaskId, P], Coroutine[Any, Any, R]],
+        coro: Callable[Concatenate[AbortableTask, P], Coroutine[Any, Any, R]],
     ) -> Callable[Concatenate[AbortableTask, P], R]:
         @wraps(coro)
         def wrapper(task: AbortableTask, *args: P.args, **kwargs: P.kwargs) -> R:
@@ -56,7 +56,7 @@ async def run_task(task_id: TaskID) -> R:
                 try:
                     async with asyncio.TaskGroup() as tg:
                         main_task = tg.create_task(
-                            coro(task, task_id, *args, **kwargs),
+                            coro(task, *args, **kwargs),
                         )
 
                         async def abort_monitor():

services/storage/tests/unit/test_async_jobs.py

@@ -94,8 +94,9 @@ async def _process_action(action: str, payload: Any) -> Any:
     return None
 
 
-def sync_job(task: Task, action: Action, payload: Any) -> Any:
+def sync_job(task: Task, task_id: TaskID, action: Action, payload: Any) -> Any:
     _ = task
+    _ = task_id
     return asyncio.run(_process_action(action, payload))