fix error model

Author: bisgaard-itis

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

@@ -31,4 +31,4 @@ class TaskSubmissionError(OsparcErrorMixin, Exception):
 
 
 class TaskNotFoundError(OsparcErrorMixin, Exception):
-    msg_template = "Task with key '{task_key}' was not found"
+    msg_template = "Task with uuid '{task_uuid}' and owner_metadata '{owner_metadata}' was not found"

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

@@ -95,7 +95,9 @@ async def cancel_task(
         ):
             task_key = owner_metadata.model_dump_task_key(task_uuid=task_uuid)
             if not await self.task_exists(task_key):
-                raise TaskNotFoundError(task_key=task_key)
+                raise TaskNotFoundError(
+                    task_uuid=task_uuid, owner_metadata=owner_metadata
+                )
 
             await self._task_info_store.remove_task(task_key)
             await self._forget_task(task_key)
@@ -117,7 +119,9 @@ async def get_task_result(
         ):
             task_key = owner_metadata.model_dump_task_key(task_uuid=task_uuid)
             if not await self.task_exists(task_key):
-                raise TaskNotFoundError(task_key=task_key)
+                raise TaskNotFoundError(
+                    task_uuid=task_uuid, owner_metadata=owner_metadata
+                )
 
             async_result = self._celery_app.AsyncResult(task_key)
             result = async_result.result
@@ -160,7 +164,9 @@ async def get_task_status(
         ):
             task_key = owner_metadata.model_dump_task_key(task_uuid=task_uuid)
             if not await self.task_exists(task_key):
-                raise TaskNotFoundError(task_key=task_key)
+                raise TaskNotFoundError(
+                    task_uuid=task_uuid, owner_metadata=owner_metadata
+                )
 
             task_state = await self._get_task_celery_state(task_key)
             return TaskStatus(

services/api-server/tests/unit/service/test_service_function_jobs_task_client.py

@@ -1,5 +1,6 @@
 # pylint: disable=redefined-outer-name
 
+import json
 from collections.abc import Callable
 
 import pytest
@@ -51,7 +52,11 @@ async def _raise(*args, **kwargs):
         )
         for state in list(TaskState)
     ]
-    + [TaskNotFoundError(task_key=_faker.uuid4())],
+    + [
+        TaskNotFoundError(
+            task_uuid=_faker.uuid4(), owner_metadata=json.dumps({"owner": "test-owner"})
+        )
+    ],
 )
 @pytest.mark.parametrize("job_creation_task_id", [_faker.uuid4(), None])
 async def test_celery_status_conversion(