Fixed processing of incomplete letter jobs

- `template` argument passed in to `job_complete` should be `template.template_type` otherwise the job status is incorrectly set
This commit is contained in:
Ken Tsang
2017-12-07 11:58:49 +00:00
parent a08411bde0
commit 68659bab1b
2 changed files with 3 additions and 5 deletions

View File

@@ -1439,6 +1439,7 @@ def test_process_incomplete_job_letter(mocker, sample_letter_template):
mocker.patch('app.celery.tasks.s3.get_job_from_s3', return_value=load_example_csv('multiple_letter'))
mock_letter_saver = mocker.patch('app.celery.tasks.save_letter.apply_async')
mock_build_dvla = mocker.patch('app.celery.tasks.build_dvla_file.apply_async')
job = create_job(template=sample_letter_template, notification_count=10,
created_at=datetime.utcnow() - timedelta(hours=2),
@@ -1453,8 +1454,5 @@ def test_process_incomplete_job_letter(mocker, sample_letter_template):
process_incomplete_job(str(job.id))
completed_job = Job.query.filter(Job.id == job.id).one()
assert completed_job.job_status == JOB_STATUS_FINISHED
assert mock_build_dvla.called
assert mock_letter_saver.call_count == 8