def test__top_job(self):
queue = Queue(DBH.job)
if len(queue.jobs()) > 0:
for j in queue.jobs():
j.remove()
self.assertEqual(len(queue.jobs()), 0)
self.assertRaises(QueueEmptyError, queue.top_job)
job_1 = Job(DBH.job, start='2010-01-01 10:00:00', priority=0).add()
job_2 = Job(DBH.job, start='2010-01-01 10:00:01', priority=-1).add()
job_3 = Job(DBH.job, start='2010-01-01 10:00:02', priority=1).add()
job_4 = Job(DBH.job, start='2999-12-31 23:59:59', priority=1).add()
job = queue.top_job()
self.assertEqual(job.id, job_3.id)
job_1.remove()
job_2.remove()
job_3.remove()
job_4.remove()
def test__jobs(self):
queue = Queue(DBH.job)
# Remove existing data
for j in Job(DBH.job).set_.get():
j.remove()
self.assertEqual(len(queue.jobs()), 0)
# (number, start, priority, status)
job_data = [
('2010-01-01 10:00:00', 0, 'a'),
('2010-01-01 10:00:00', 0, 'd'),
('2010-01-01 10:00:01', -1, 'a'),
('2010-01-01 10:00:01', -1, 'd'),
('2010-01-01 10:00:02', 1, 'a'),
('2010-01-01 10:00:02', 1, 'd'),
]
jobs = []
for j in job_data:
job_d = dict(command='pwd', start=j[0], priority=j[1], status=j[2])
jobs.append(Job(DBH.job, **job_d).add())
job_set = queue.jobs()
self.assertEqual(len(job_set), 3)
self.assertEqual([x.id for x in job_set],
[jobs[0].id, jobs[2].id, jobs[4].id])
job_set = queue.jobs(maximum_start='2010-01-01 10:00:01')
self.assertEqual(len(job_set), 2)
self.assertEqual([x.id for x in job_set],
[jobs[0].id, jobs[2].id])
# Test orderby
# Orderby priority ASC
job_set = queue.jobs(orderby=DBH.job.priority)
self.assertEqual(len(job_set), 3)
self.assertEqual([x.id for x in job_set],
[jobs[2].id, jobs[0].id, jobs[4].id])
# Orderby priority DESC
job_set = queue.jobs(orderby=~DBH.job.priority)
self.assertEqual(len(job_set), 3)
self.assertEqual([x.id for x in job_set],
[jobs[4].id, jobs[0].id, jobs[2].id])
# Test limitby
# Highest priority job
job_set = queue.jobs(orderby=~DBH.job.priority, limitby=1)
self.assertEqual(len(job_set), 1)
self.assertEqual([x.id for x in job_set], [jobs[4].id])
for j in jobs:
j.remove()
