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')