Fix tests for priority RabbitMQ implementation

alephdata · Nov 20, 2023 · 3ddfebe · 3ddfebe
1 parent f0622a3
commit 3ddfebe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_taskqueue.py b/tests/test_taskqueue.py
@@ -31,18 +31,19 @@ def test_task_queue(self):
         conn = get_fakeredis()
         collection_id = 2
         task_id = "test-task"
+        priority = randrange(1, settings.RABBITMQ_MAX_PRIORITY + 1)
         body = {
             "collection_id": 2,
             "job_id": "test-job",
             "task_id": "test-task",
             "operation": "test-op",
             "context": {},
             "payload": {},
+            "priority": priority,
         }
         connection = get_rabbitmq_connection()
         channel = connection.channel()
         channel.queue_purge(settings.QUEUE_INGEST)
-        priority = randrange(1, settings.RABBITMQ_MAX_PRIORITY + 1)
         channel.basic_publish(
             properties=pika.BasicProperties(priority=priority),
             exchange="",
@@ -72,7 +73,6 @@ def test_task_queue(self):
         assert task.get_retry_count(conn) == 1
 
         with patch("servicelayer.settings.WORKER_RETRY", 0):
-            priority = randrange(1, settings.RABBITMQ_MAX_PRIORITY + 1)
             channel = connection.channel()
             channel.queue_purge(settings.QUEUE_INGEST)
             channel.basic_publish(