fixed tests that used the new state.

This commit is contained in:
Martyn Inglis
2017-04-18 14:40:48 +01:00
parent 5081ffc675
commit 2b4043515f
2 changed files with 5 additions and 2 deletions

View File

@@ -1060,6 +1060,8 @@ def test_update_dvla_job_to_error(sample_letter_template, sample_letter_job):
update_dvla_job_to_error(job_id=sample_letter_job.id)
updated_notifications = Notification.query.all()
assert [(n.status == 'created', n.sent_by == 'dvla') for n in updated_notifications]
for n in updated_notifications:
assert n.status == 'created'
assert not n.sent_by
assert 'error' == Job.query.filter_by(id=sample_letter_job.id).one().job_status

View File

@@ -673,7 +673,7 @@ def test_get_jobs_accepts_page_parameter(
@pytest.mark.parametrize('statuses_filter, expected_statuses', [
('', JOB_STATUS_TYPES),
('pending', [JOB_STATUS_PENDING]),
('pending, in progress, finished, sending limits exceeded, scheduled, cancelled, ready to send, sent to dvla',
('pending, in progress, finished, sending limits exceeded, scheduled, cancelled, ready to send, sent to dvla, error', # noqa
JOB_STATUS_TYPES),
# bad statuses are accepted, just return no data
('foo', [])
@@ -694,6 +694,7 @@ def test_get_jobs_can_filter_on_statuses(
create_job(notify_db, notify_db_session, job_status='cancelled')
create_job(notify_db, notify_db_session, job_status='ready to send')
create_job(notify_db, notify_db_session, job_status='sent to dvla')
create_job(notify_db, notify_db_session, job_status='error')
path = '/service/{}/job'.format(sample_service.id)
response = client.get(