Make unstable tests more stable (#1523)

Co-authored-by: rpkak <rpkak@users.noreply.github.com>
main
rpkak 3 years ago committed by GitHub
parent 643a26b991
commit 9737ffa95f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -66,10 +66,10 @@ class TestRegistry(RQTestCase):
job = queue.enqueue(say_hello) job = queue.enqueue(say_hello)
registry.add(job, 5) registry.add(job, 5)
self.assertEqual( time = registry.get_expiration_time(job)
registry.get_expiration_time(job), expected_time = (datetime.utcnow() + timedelta(seconds=5)).replace(microsecond=0)
(datetime.utcnow() + timedelta(seconds=5)).replace(microsecond=0) self.assertGreaterEqual(time, expected_time - timedelta(seconds=2))
) self.assertLessEqual(time, expected_time + timedelta(seconds=2))
def test_add_and_remove(self): def test_add_and_remove(self):
"""Adding and removing job to StartedJobRegistry.""" """Adding and removing job to StartedJobRegistry."""

@ -114,7 +114,7 @@ class TestRetry(RQTestCase):
# Scheduled time is roughly 5 seconds from now # Scheduled time is roughly 5 seconds from now
scheduled_time = registry.get_scheduled_time(job) scheduled_time = registry.get_scheduled_time(job)
now = datetime.now(timezone.utc) now = datetime.now(timezone.utc)
self.assertTrue(now + timedelta(seconds=4) < scheduled_time < now + timedelta(seconds=6)) self.assertTrue(now + timedelta(seconds=4) < scheduled_time < now + timedelta(seconds=10))
def test_cleanup_handles_retries(self): def test_cleanup_handles_retries(self):
"""Expired jobs should also be retried""" """Expired jobs should also be retried"""

Loading…
Cancel
Save