class TestScheduleJobs(TransactionTestCase): def setUp(self): self.q = Queue() self.w = Worker.create([self.q]) def test_shedule_call(self): """Schedule to fire now""" job = self.q.schedule_call(now(), do_nothing) self.w.work(burst=True) with self.assertRaises(Job.DoesNotExist) as exc: Job.objects.get(queue_id='default', pk=job.id) def test_schedule_future_call(self): """Schedule to fire in the distant future""" job = self.q.schedule_call(datetime(2999,12,1, tzinfo=utc), do_nothing) self.w.work(burst=True) # check it is still in the queue self.assertIsNotNone(Job.objects.get(queue_id='default', pk=job.id))
class Test_get_job_or_promise(TransactionTestCase): """Test the Job._get_job_or_promise classmethod""" def setUp(self): self.q = Queue(scheduled=True) # simulate the default worker timeout self.timeout = 60 future = now() + timedelta(seconds=self.timeout/2) # enqueue a job for 30 seconds time in the future self.job = self.q.schedule_call(future, do_nothing) def test_get_job_or_promise(self): """Test get a promise of a job in the future""" job, promise, timeout = Job._get_job_or_promise( self.q.connection, self.q, self.timeout) self.assertLessEqual(timeout, self.timeout) self.assertIsNone(job) self.assertEqual(promise, self.q.name) def test_get_no_job_no_promise(self): """Test get no job and no promise""" # job is in the future beyond the current # worker timeout job, promise, timeout = Job._get_job_or_promise( self.q.connection, self.q, 1) self.assertEqual(timeout, 1) self.assertIsNone(job) self.assertIsNone(promise) def test_get_earlier_job_no_promise(self): """Test get earlier job and no promise""" # Job enqueue after the first scheduled job # but to be exec ahead of the scheduled job now_job = self.q.enqueue(do_nothing) job, promise, timeout = Job._get_job_or_promise( self.q.connection, self.q, 60) # timeout should remain the same self.assertEqual(timeout, 60) self.assertEqual(now_job.id, job.id) self.assertIsNone(promise)
class Test_get_job_no_promise(TransactionTestCase): def setUp(self): # setup a job in the very near future which # should execute self.q = Queue() # simulate the default worker timeout self.timeout = 60 future = now() + timedelta(seconds=1) # enqueue a job for 1 second time in the future self.job = self.q.schedule_call(future, do_nothing) time.sleep(1) def test_get_job_no_promise(self): """Test get job and no promise""" job, promise, timeout = Job._get_job_or_promise( self.q.connection, self.q, self.timeout) self.assertEqual(timeout, self.timeout) self.assertEquals(job.id, self.job.id) self.assertIsNone(promise)
class TestGetJobNoPromise(TransactionTestCase): def setUp(self): # setup a job in the very near future which # should execute self.q = Queue(scheduled=True) # simulate the default worker timeout self.timeout = 60 future = now() + timedelta(seconds=1) # enqueue a job for 1 second time in the future self.job = self.q.schedule_call(future, do_nothing) time.sleep(1) def test_get_job_no_promise(self): """Test get job and no promise""" job, promise, timeout = Job._get_job_or_promise( self.q.connection, self.q, self.timeout) self.assertEqual(timeout, self.timeout) self.assertEqual(job.id, self.job.id) self.assertIsNone(promise)
class Test_get_job_or_promise(TransactionTestCase): """Test the Job._get_job_or_promise classmethod""" def setUp(self): self.q = Queue() # simulate the default worker timeout self.timeout = 60 future = now() + timedelta(seconds=self.timeout / 2) # enqueue a job for 30 seconds time in the future self.job = self.q.schedule_call(future, do_nothing) def test_get_job_or_promise(self): """Test get a promise of a job in the future""" job, promise, timeout = Job._get_job_or_promise( self.q.connection, self.q, self.timeout) self.assertLessEqual(timeout, self.timeout) self.assertIsNone(job) self.assertEqual(promise, self.q.name) def test_get_no_job_no_promise(self): """Test get no job and no promise""" # job is in the future beyond the current # worker timeout job, promise, timeout = Job._get_job_or_promise( self.q.connection, self.q, 1) self.assertEqual(timeout, 1) self.assertIsNone(job) self.assertIsNone(promise) def test_get_earlier_job_no_promise(self): """Test get earlier job and no promise""" # Job enqueue after the first scheduled job # but to be exec ahead of the scheduled job now_job = self.q.enqueue(do_nothing) job, promise, timeout = Job._get_job_or_promise( self.q.connection, self.q, 60) # timeout should remain the same self.assertEqual(timeout, 60) self.assertEqual(now_job.id, job.id) self.assertIsNone(promise)