fixed tests

Author: Andrei Neagu

services/storage/tests/unit/api/_worker_tasks/conftest.py

@@ -24,10 +24,10 @@ def mock_task_progress(mocker: MockerFixture) -> list[ProgressReport]:
     progress_updates = []
 
     async def _progress(*args, **_) -> None:
-        progress_updates.append(args[1])
+        progress_updates.append(args[2])
 
     mocker.patch(
-        "simcore_service_storage.modules.celery.worker.CeleryWorkerClient.set_task_progress",
+        "simcore_service_storage.modules.celery.worker.CeleryTaskQueueWorker.set_task_progress",
         side_effect=_progress,
     )
     return progress_updates

services/storage/tests/unit/api/_worker_tasks/test__data_export.py

@@ -1,22 +1,36 @@
+# pylint:disable=redefined-outer-name
+
 import re
 
 import pytest
 from celery import Task
+from faker import Faker
 from models_library.progress_bar import ProgressReport, ProgressStructuredMessage
 from models_library.users import UserID
 from simcore_service_storage.api._worker_tasks._data_export import data_export
+from simcore_service_storage.modules.celery.models import TaskID
 
 pytest_simcore_core_services_selection = [
     "postgres",
     "rabbit",
 ]
 
 
+@pytest.fixture
+def fake_task_id(faker: Faker) -> TaskID:
+    return f"{faker.uuid4()}"
+
+
 @pytest.mark.usefixtures("celery_worker")
 async def test_data_export(
-    mock_task_progress: list[ProgressReport], fake_celery_task: Task, user_id: UserID
+    mock_task_progress: list[ProgressReport],
+    fake_celery_task: Task,
+    user_id: UserID,
+    fake_task_id: TaskID,
 ):
-    result = await data_export(fake_celery_task, user_id=user_id, paths_to_export=[])
+    result = await data_export(
+        fake_celery_task, fake_task_id, user_id=user_id, paths_to_export=[]
+    )
     assert re.fullmatch(
         rf"^exports/{user_id}/[0-9a-fA-F]{{8}}-[0-9a-fA-F]{{4}}-[0-9a-fA-F]{{4}}-[0-9a-fA-F]{{4}}-[0-9a-fA-F]{{12}}\.zip$",
         result,
@@ -29,11 +43,7 @@ async def test_data_export(
             attempt=0,
             unit=None,
             message=ProgressStructuredMessage(
-                description="create and upload export",
-                current=0.0,
-                total=1,
-                unit=None,
-                sub=None,
+                description="data export", current=0.0, total=1, unit=None, sub=None
             ),
         ),
         ProgressReport(
@@ -42,11 +52,7 @@ async def test_data_export(
             attempt=0,
             unit=None,
             message=ProgressStructuredMessage(
-                description="create and upload export",
-                current=1.0,
-                total=1,
-                unit=None,
-                sub=None,
+                description="data export", current=1.0, total=1, unit=None, sub=None
             ),
         ),
     ]