mirror of https://github.com/rq/rq.git
52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
import time
|
|
|
|
from rq import Queue, SimpleWorker
|
|
from rq.registry import FailedJobRegistry, FinishedJobRegistry
|
|
from rq.timeouts import TimerDeathPenalty
|
|
from tests import RQTestCase
|
|
|
|
|
|
class TimerBasedWorker(SimpleWorker):
|
|
death_penalty_class = TimerDeathPenalty
|
|
|
|
|
|
def thread_friendly_sleep_func(seconds):
|
|
end_at = time.time() + seconds
|
|
while True:
|
|
if time.time() > end_at:
|
|
break
|
|
|
|
|
|
class TestTimeouts(RQTestCase):
|
|
def test_timer_death_penalty(self):
|
|
"""Ensure TimerDeathPenalty works correctly."""
|
|
q = Queue(connection=self.connection)
|
|
q.empty()
|
|
finished_job_registry = FinishedJobRegistry(connection=self.connection)
|
|
failed_job_registry = FailedJobRegistry(connection=self.connection)
|
|
|
|
# make sure death_penalty_class persists
|
|
w = TimerBasedWorker([q], connection=self.connection)
|
|
self.assertIsNotNone(w)
|
|
self.assertEqual(w.death_penalty_class, TimerDeathPenalty)
|
|
|
|
# Test short-running job doesn't raise JobTimeoutException
|
|
job = q.enqueue(thread_friendly_sleep_func, args=(1,), job_timeout=3)
|
|
w.work(burst=True)
|
|
job.refresh()
|
|
self.assertIn(job, finished_job_registry)
|
|
|
|
# Test long-running job raises JobTimeoutException
|
|
job = q.enqueue(thread_friendly_sleep_func, args=(5,), job_timeout=3)
|
|
w.work(burst=True)
|
|
self.assertIn(job, failed_job_registry)
|
|
job.refresh()
|
|
self.assertIn("rq.timeouts.JobTimeoutException", job.exc_info)
|
|
|
|
# Test negative timeout doesn't raise JobTimeoutException,
|
|
# which implies an unintended immediate timeout.
|
|
job = q.enqueue(thread_friendly_sleep_func, args=(1,), job_timeout=-1)
|
|
w.work(burst=True)
|
|
job.refresh()
|
|
self.assertIn(job, finished_job_registry)
|