diff --git a/tests/test_registry.py b/tests/test_registry.py index 57f3e1d..0dd7772 100644 --- a/tests/test_registry.py +++ b/tests/test_registry.py @@ -66,10 +66,10 @@ class TestRegistry(RQTestCase): job = queue.enqueue(say_hello) registry.add(job, 5) - self.assertEqual( - registry.get_expiration_time(job), - (datetime.utcnow() + timedelta(seconds=5)).replace(microsecond=0) - ) + time = registry.get_expiration_time(job) + expected_time = (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): """Adding and removing job to StartedJobRegistry.""" diff --git a/tests/test_retry.py b/tests/test_retry.py index ad01b37..e8fddeb 100644 --- a/tests/test_retry.py +++ b/tests/test_retry.py @@ -114,7 +114,7 @@ class TestRetry(RQTestCase): # Scheduled time is roughly 5 seconds from now scheduled_time = registry.get_scheduled_time(job) 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): """Expired jobs should also be retried"""