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