fixed test

Author: Andrei Neagu

services/storage/src/simcore_service_storage/modules/celery/utils.py

@@ -3,25 +3,25 @@
 
 from .worker import CeleryWorkerClient
 
-_WORKER_KEY = "celery_worker_client"
+_WORKER_KEY = "celery_worker"
 _FASTAPI_APP_KEY = "fastapi_app"
 
 
+def set_celery_worker_client(celery_app: Celery, worker: CeleryWorkerClient) -> None:
+    celery_app.conf[_WORKER_KEY] = worker
+
+
 def get_celery_worker_client(celery_app: Celery) -> CeleryWorkerClient:
     worker = celery_app.conf[_WORKER_KEY]
     assert isinstance(worker, CeleryWorkerClient)
     return worker
 
 
+def set_fastapi_app(celery_app: Celery, fastapi_app: FastAPI) -> None:
+    celery_app.conf[_FASTAPI_APP_KEY] = fastapi_app
+
+
 def get_fastapi_app(celery_app: Celery) -> FastAPI:
     fastapi_app = celery_app.conf[_FASTAPI_APP_KEY]
     assert isinstance(fastapi_app, FastAPI)
     return fastapi_app
-
-
-def set_celery_worker_client(celery_app: Celery, worker: CeleryWorkerClient) -> None:
-    celery_app.conf[_WORKER_KEY] = worker
-
-
-def set_fastapi_app(celery_app: Celery, fastapi_app: FastAPI) -> None:
-    celery_app.conf[_FASTAPI_APP_KEY] = fastapi_app

services/storage/tests/unit/modules/celery/test_celery.py

@@ -48,7 +48,7 @@ def sync_archive(task: Task, files: list[str]) -> str:
     assert task.name
     _logger.info("Calling async_archive")
     return asyncio.run_coroutine_threadsafe(
-        _async_archive(task, task.request.id, files),
+        _async_archive(task.app, task, files),
         get_event_loop(get_fastapi_app(task.app)),
     ).result()