Make sure we don't requeue tasks with excessive retries

alephdata · Jun 18, 2024 · 3fea636 · 3fea636
1 parent f94302e
commit 3fea636
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/tests/test_taskqueue.py b/tests/test_taskqueue.py
@@ -93,7 +93,15 @@ def test_task_queue(self):
             dataset.add_task(task_id, "test-op")
             channel.close()
             with self.assertLogs(level="ERROR") as ctx:
-                worker.process(blocking=False)
+                with patch.object(
+                    pika.channel.Channel,
+                    attribute="basic_nack",
+                    return_value=None,
+                ) as nack_fn:
+                    worker.process(blocking=False)
+                    nack_fn.assert_any_call(
+                        delivery_tag=1, multiple=False, requeue=False
+                    )
             assert "Max retries reached for task test-task. Aborting." in ctx.output[0]
             # Assert that retry count stays the same
             assert task.get_retry_count(conn) == 1