[DOP-22266] - change celery queue name from queue.name to queue.slug

MobileTeleSystems · Dec 11, 2024 · 36f6fb3 · 36f6fb3
1 parent f51242b
commit 36f6fb3
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 6 deletions.
diff --git a/docker/Dockerfile.worker b/docker/Dockerfile.worker
@@ -17,6 +17,7 @@ RUN apt-get update && apt-get install -y \
         libffi-dev \
         libkrb5-dev \
         krb5-user \
+        curl \
     && rm -rf /var/lib/apt/lists/*
 
 RUN curl -sSL https://install.python-poetry.org | python3 - \

diff --git a/syncmaster/backend/api/v1/runs.py b/syncmaster/backend/api/v1/runs.py
@@ -135,7 +135,7 @@ async def start_run(
             celery.send_task,
             "run_transfer_task",
             kwargs={"run_id": run.id},
-            queue=transfer.queue.name,
+            queue=transfer.queue.slug,
         )
     except KombuError as e:
         async with unit_of_work:

diff --git a/syncmaster/scheduler/transfer_job_manager.py b/syncmaster/scheduler/transfer_job_manager.py
@@ -78,7 +78,7 @@ async def send_job_to_celery(transfer_id: int) -> None:
                     celery.send_task,
                     "run_transfer_task",
                     kwargs={"run_id": run.id},
-                    queue=transfer.queue.name,
+                    queue=transfer.queue.slug,
                 )
             except KombuError as e:
                 async with unit_of_work:

diff --git a/tests/test_unit/test_runs/test_create_run.py b/tests/test_unit/test_runs/test_create_run.py
@@ -61,7 +61,7 @@ async def test_developer_plus_can_create_run_of_transfer_his_group(
         mock_send_task,
         "run_transfer_task",
         kwargs={"run_id": run.id},
-        queue=group_transfer.queue.name,
+        queue=group_transfer.queue.slug,
     )
 
 
@@ -163,7 +163,7 @@ async def test_superuser_can_create_run(
         mock_send_task,
         "run_transfer_task",
         kwargs={"run_id": run.id},
-        queue=group_transfer.queue.name,
+        queue=group_transfer.queue.slug,
     )
 
 

diff --git a/tests/test_unit/test_scheduler/test_transfer_job_manager.py b/tests/test_unit/test_scheduler/test_transfer_job_manager.py
@@ -96,7 +96,7 @@ async def test_send_job_to_celery_with_success(
         mock_send_task,
         "run_transfer_task",
         kwargs={"run_id": run.id},
-        queue=group_transfer.queue.name,
+        queue=group_transfer.queue.slug,
     )
 
 

diff --git a/tests/test_unit/utils.py b/tests/test_unit/utils.py
@@ -101,7 +101,7 @@ async def create_queue(
         name=name,
         description=description,
         group_id=group_id,
-        slug=f"{group_id}-{name}",
+        slug=f"{name}",
     )
     session.add(queue)
     await session.commit()