class TaskTest(TestCase):
def setUp(self):
self.redis = StrictRedis(**settings.REDISES['default'])
self.job_changes_pool_key = 'queue_job_changes'
self.job_sync = JobSync(self.redis)
def tearDown(self):
Task.objects.all().delete()
#self.redis.delete('active_jobs_pool')
self.redis.delete(self.job_changes_pool_key)
self.redis.delete('job_stats.1')
def testAddJob(self):
task = self.generateTask()
task.run_time = '{"days":0, "hours":0, "minutes":10, "seconds":0}'
task.save()
self.assertTrue(task.id > 0)
job = task.getJob()
now = Task.objects.now()
self.assertTrue(job.compute_next_run_time(now) is not None)
self.assertEqual(600, job.trigger.interval_length)
self.assertEqual(task.run_entry, job.func)
self.assertEqual(1, self.redis.llen(self.job_changes_pool_key))
def testGetJob(self):
task = self.generateTask()
task.save()
job = task.getJob()
self.assertEqual(task.id, job.id)
self.assertEqual(task.name, job.name)
self.assertEqual(task.run_entry, job.func)
self.assertEqual(str(task.getTrigger()), str(job.trigger))
def testDelete(self):
task = self.generateTask()
task.active = 1
task.save()
now = Task.objects.now()
job_id, change_type, job = self.job_sync.pop(now)
self.assertEqual(task.id, job_id)
self.assertEqual('add', change_type)
task_id = task.id
task.delete()
job_id, change_type, job = self.job_sync.pop(now)
self.assertEqual(task_id, job_id)
self.assertEqual('remove', change_type)
def testUpdate(self):
task = self.generateTask()
task.save()
now = Task.objects.now()
job_id, change_type, job = self.job_sync.pop(now)
task.run_time = '2013-09-10 00:00:00'
task.type = 'date'
task.save()
job_id, change_type, job = self.job_sync.pop(now)
self.assertEqual(task.id, job_id)
self.assertEqual('update', change_type)
def testEnable(self):
task = self.generateTask()
task.active = 0
task.save()
self.assertEqual(0, self.job_sync.count())
task.disable()
self.assertEqual(0, self.job_sync.count())
task.enable()
now = Task.objects.now()
job_id, change_type, job = self.job_sync.pop(now)
self.assertEqual(task.id, job_id)
self.assertEqual('add', change_type)
def generateTask(self):
task = Task()
task.name = 'Task1'
task.run_time = '{"days":0, "hours":0, "minutes":10, "seconds":0}'
task.run_entry = 'http://baidu.com'
task.run_method = 'post'
task.active = 1
return task
def testDisable(self):
task = self.generateTask()
task.active = 1
task.save()
now = Task.objects.now()
job_id, change_type, job = self.job_sync.pop(now)
self.assertEqual(task.id, job_id)
self.assertEqual('add', change_type)
task.enable()
self.assertEqual(0, self.job_sync.count())
task.disable()
job_id, change_type, job = self.job_sync.pop(now)
self.assertEqual(task.id, job_id)
self.assertEqual('remove', change_type)
def testRun(self):
task = self.generateTask()
task.save()
self.redis.delete('job_stats.%d' % task.id)
self.assertEqual(0,task.stats['total'])
task.run()
self.assertEqual(1,task.stats['total'])
