mirror of https://github.com/rq/rq.git
125 lines
4.5 KiB
Python
125 lines
4.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import absolute_import
|
|
|
|
from rq.job import Job
|
|
from rq.queue import FailedQueue, Queue
|
|
from rq.utils import current_timestamp
|
|
from rq.worker import Worker
|
|
from rq.registry import FinishedJobRegistry, StartedJobRegistry
|
|
|
|
from tests import RQTestCase
|
|
from tests.fixtures import div_by_zero, say_hello
|
|
|
|
|
|
class TestRegistry(RQTestCase):
|
|
|
|
def setUp(self):
|
|
super(TestRegistry, self).setUp()
|
|
self.registry = StartedJobRegistry(connection=self.testconn)
|
|
|
|
def test_add_and_remove(self):
|
|
"""Adding and removing job to StartedJobRegistry."""
|
|
timestamp = current_timestamp()
|
|
job = Job()
|
|
|
|
# Test that job is added with the right score
|
|
self.registry.add(job, 1000)
|
|
self.assertLess(self.testconn.zscore(self.registry.key, job.id),
|
|
timestamp + 1002)
|
|
|
|
# Ensure that a timeout of -1 results in a score of -1
|
|
self.registry.add(job, -1)
|
|
self.assertEqual(self.testconn.zscore(self.registry.key, job.id), -1)
|
|
|
|
# Ensure that job is properly removed from sorted set
|
|
self.registry.remove(job)
|
|
self.assertIsNone(self.testconn.zscore(self.registry.key, job.id))
|
|
|
|
def test_get_job_ids(self):
|
|
"""Getting job ids from StartedJobRegistry."""
|
|
timestamp = current_timestamp()
|
|
self.testconn.zadd(self.registry.key, timestamp + 10, 'foo')
|
|
self.testconn.zadd(self.registry.key, timestamp + 20, 'bar')
|
|
self.assertEqual(self.registry.get_job_ids(), ['foo', 'bar'])
|
|
|
|
def test_get_expired_job_ids(self):
|
|
"""Getting expired job ids form StartedJobRegistry."""
|
|
timestamp = current_timestamp()
|
|
|
|
self.testconn.zadd(self.registry.key, 1, 'foo')
|
|
self.testconn.zadd(self.registry.key, timestamp + 10, 'bar')
|
|
|
|
self.assertEqual(self.registry.get_expired_job_ids(), ['foo'])
|
|
|
|
def test_cleanup(self):
|
|
"""Moving expired jobs to FailedQueue."""
|
|
failed_queue = FailedQueue(connection=self.testconn)
|
|
self.assertTrue(failed_queue.is_empty())
|
|
self.testconn.zadd(self.registry.key, 1, 'foo')
|
|
self.registry.cleanup()
|
|
self.assertIn('foo', failed_queue.job_ids)
|
|
self.assertEqual(self.testconn.zscore(self.registry.key, 'foo'), None)
|
|
|
|
def test_job_execution(self):
|
|
"""Job is removed from StartedJobRegistry after execution."""
|
|
registry = StartedJobRegistry(connection=self.testconn)
|
|
queue = Queue(connection=self.testconn)
|
|
worker = Worker([queue])
|
|
|
|
job = queue.enqueue(say_hello)
|
|
|
|
worker.prepare_job_execution(job)
|
|
self.assertIn(job.id, registry.get_job_ids())
|
|
|
|
worker.perform_job(job)
|
|
self.assertNotIn(job.id, registry.get_job_ids())
|
|
|
|
# Job that fails
|
|
job = queue.enqueue(div_by_zero)
|
|
|
|
worker.prepare_job_execution(job)
|
|
self.assertIn(job.id, registry.get_job_ids())
|
|
|
|
worker.perform_job(job)
|
|
self.assertNotIn(job.id, registry.get_job_ids())
|
|
|
|
def test_get_job_count(self):
|
|
"""StartedJobRegistry returns the right number of job count."""
|
|
timestamp = current_timestamp() + 10
|
|
self.testconn.zadd(self.registry.key, timestamp, 'foo')
|
|
self.testconn.zadd(self.registry.key, timestamp, 'bar')
|
|
self.assertEqual(self.registry.count, 2)
|
|
self.assertEqual(len(self.registry), 2)
|
|
|
|
|
|
class TestFinishedJobRegistry(RQTestCase):
|
|
|
|
def setUp(self):
|
|
super(TestFinishedJobRegistry, self).setUp()
|
|
self.registry = FinishedJobRegistry(connection=self.testconn)
|
|
|
|
def test_cleanup(self):
|
|
"""Finished job registry removes expired jobs."""
|
|
timestamp = current_timestamp()
|
|
self.testconn.zadd(self.registry.key, 1, 'foo')
|
|
self.testconn.zadd(self.registry.key, timestamp + 10, 'bar')
|
|
|
|
self.registry.cleanup()
|
|
self.assertEqual(self.registry.get_job_ids(), ['bar'])
|
|
|
|
def test_jobs_are_put_in_registry(self):
|
|
"""Completed jobs are added to FinishedJobRegistry."""
|
|
self.assertEqual(self.registry.get_job_ids(), [])
|
|
queue = Queue(connection=self.testconn)
|
|
worker = Worker([queue])
|
|
|
|
# Completed jobs are put in FinishedJobRegistry
|
|
job = queue.enqueue(say_hello)
|
|
worker.perform_job(job)
|
|
self.assertEqual(self.registry.get_job_ids(), [job.id])
|
|
|
|
# Failed jobs are not put in FinishedJobRegistry
|
|
failed_job = queue.enqueue(div_by_zero)
|
|
worker.perform_job(failed_job)
|
|
self.assertEqual(self.registry.get_job_ids(), [job.id])
|