def test_restart_job_in_record_output_step(self):
self.job.state = Job.JOB_STATE_ERROR
self.job.step = Job.JOB_STEP_RECORD_OUTPUT_PROJECT
self.job.save()
job = LandoJob(self.job.id, self.user)
with self.assertRaises(ValidationError) as raised_error:
job.restart()
def test_restart_job(self, mock_give_download_permissions,
mock_has_download_permissions, mock_make_client):
self.job.state = Job.JOB_STATE_ERROR
self.job.step = Job.JOB_STEP_RUNNING
self.job.save()
mock_has_download_permissions.return_value = False
job = LandoJob(self.job.id, self.user)
job.restart()
mock_make_client().restart_job.assert_called()
mock_give_download_permissions.assert_has_calls(
[call(self.user, '1234', '5432'),
call(self.user, '1235', '5432')],
any_order=True)