pass retry param to enqueue_at func (#1343)

main
Vladimir Ulupov 4 years ago committed by GitHub
parent a31d37f1ea
commit 237e69123a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -430,7 +430,7 @@ nd
job = self.create_job(f, status=JobStatus.SCHEDULED, args=args, kwargs=kwargs, job = self.create_job(f, status=JobStatus.SCHEDULED, args=args, kwargs=kwargs,
timeout=timeout, result_ttl=result_ttl, ttl=ttl, timeout=timeout, result_ttl=result_ttl, ttl=ttl,
failure_ttl=failure_ttl, description=description, failure_ttl=failure_ttl, description=description,
depends_on=depends_on, job_id=job_id, meta=meta) depends_on=depends_on, job_id=job_id, meta=meta, retry=retry)
return self.schedule_job(job, datetime) return self.schedule_job(job, datetime)

@ -7,7 +7,7 @@ from multiprocessing import Process
from rq import Queue from rq import Queue
from rq.compat import utc, PY2 from rq.compat import utc, PY2
from rq.exceptions import NoSuchJobError from rq.exceptions import NoSuchJobError
from rq.job import Job from rq.job import Job, Retry
from rq.registry import FinishedJobRegistry, ScheduledJobRegistry from rq.registry import FinishedJobRegistry, ScheduledJobRegistry
from rq.scheduler import RQScheduler from rq.scheduler import RQScheduler
from rq.utils import current_timestamp from rq.utils import current_timestamp
@ -321,3 +321,12 @@ class TestQueue(RQTestCase):
self.assertTrue( self.assertTrue(
now + timedelta(seconds=28) < scheduled_time < now + timedelta(seconds=32) now + timedelta(seconds=28) < scheduled_time < now + timedelta(seconds=32)
) )
def test_enqueue_in_with_retry(self):
""" Ensure that the retry parameter is passed
to the enqueue_at function from enqueue_in.
"""
queue = Queue(connection=self.testconn)
job = queue.enqueue_in(timedelta(seconds=30), say_hello, retry=Retry(3, [2]))
self.assertEqual(job.retries_left, 3)
self.assertEqual(job.retry_intervals, [2])

Loading…
Cancel
Save