def test_set_task_status(self):
c = MockClient()
job = Job(self.job_file)
task = job.assign_next_task(c)
self.assertEqual(task.startframe, 1, 'start frame is correct')
self.assertEqual(task.endframe, 2, 'end frame is correct')
job.set_task_status(1, 'complete', 10)
task = job.assign_next_task(c)
self.assertEqual(task.startframe, 3, 'start frame is correct')
self.assertEqual(task.endframe, 4, 'end frame is correct')
self.assertEqual(job.tasks[0].status, 'complete')
self.assertEqual(job.tasks[1].status, 'rendering')
def test_next_step(self):
c = MockClient()
job = Job(self.job_file)
task = job.assign_next_task(c)
self.assertEqual(task.startframe, 1, 'start frame is correct')
self.assertEqual(task.endframe, 2, 'end frame is correct')
self.assertEqual(task.client, c, 'client is correct')
task.status = 'complete'
task = job.assign_next_task(c)
self.assertEqual(task.startframe, 3, 'start frame is correct')
self.assertEqual(task.endframe, 4, 'end frame is correct')
task.status = 'complete'
task = job.assign_next_task(c)
self.assertEqual(task.startframe, 5, 'start frame is correct')
self.assertEqual(task.endframe, 6, 'end frame is correct')
task.status = 'complete'
task = job.assign_next_task(c)
self.assertEqual(task, None, 'None returned when all tasks complete')
self.job_dict['tasksize'] = 5
self.update_job_file()
job = Job(self.job_file)
task = job.assign_next_task(c)
self.assertEqual(task.startframe, 1, 'start frame is correct')
self.assertEqual(task.endframe, 5, 'end frame is correct')
task.status = 'complete'
task = job.assign_next_task(c)
self.assertEqual(task.startframe, 6, 'start frame is correct')
self.assertEqual(task.endframe, 6, 'end frame is correct')
self.job_dict['tasksize'] = 1
self.job_dict['startframe'] = 1
self.job_dict['endframe'] = 1
self.update_job_file()
job = Job(self.job_file)
task = job.assign_next_task(c)
self.assertEqual(task.startframe, 1, 'start frame is correct')
self.assertEqual(task.endframe, 1, 'end frame is correct')
self.job_dict['tasksize'] = 3
self.job_dict['startframe'] = 1
self.job_dict['endframe'] = 1
self.update_job_file()
job = Job(self.job_file)
task = job.assign_next_task(c)
self.assertEqual(task.startframe, 1, 'start frame is correct')
self.assertEqual(task.endframe, 1, 'end frame is correct')
