Also test task locks & queue locks

closeio · Apr 24, 2024 · a829c07 · a829c07
1 parent 6fe6daf
commit a829c07
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tests/test_workers.py b/tests/test_workers.py
@@ -188,15 +188,20 @@ def test_handles_timeout(self, tiger, ensure_queues):
         ensure_queues(error={"default": 1})
 
     def test_heartbeat(self, tiger):
-        task = Task(tiger, sleep_task)
+        # Test both task heartbeat and lock renewal.
+        task = Task(tiger, sleep_task, lock=True)
         task.delay()
 
         # Start a worker and wait until it starts processing.
         worker = Process(
             target=external_worker,
             kwargs={
                 "patch_config": {"ACTIVE_TASK_UPDATE_TIMER": DELAY / 2},
-                "worker_kwargs": {"executor_class": SyncExecutor},
+                "worker_kwargs": {
+                    # Test queue lock.
+                    "max_workers_per_queue": 1,
+                    "executor_class": SyncExecutor,
+                },
             },
         )
         worker.start()