From bd79a6f770834226178f4f4d14ab19c33472d226 Mon Sep 17 00:00:00 2001 From: Ken Tsang Date: Tue, 22 Aug 2017 11:05:58 +0100 Subject: [PATCH] Improved tests on letter jobs by status --- tests/app/dao/test_jobs_dao.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/app/dao/test_jobs_dao.py b/tests/app/dao/test_jobs_dao.py index 4e2f3062f..72f3ece04 100644 --- a/tests/app/dao/test_jobs_dao.py +++ b/tests/app/dao/test_jobs_dao.py @@ -24,7 +24,7 @@ from app.dao.statistics_dao import create_or_update_job_sending_statistics, upda from app.models import ( Job, JobStatistics, EMAIL_TYPE, SMS_TYPE, LETTER_TYPE, - JOB_STATUS_READY_TO_SEND + JOB_STATUS_READY_TO_SEND, JOB_STATUS_SENT_TO_DVLA, JOB_STATUS_FINISHED, JOB_STATUS_PENDING ) from tests.app.conftest import sample_notification as create_notification @@ -557,7 +557,9 @@ def test_dao_get_letter_jobs_by_status(sample_service): jobs = [] jobs.append(create_db_job(letter_template, job_status=JOB_STATUS_READY_TO_SEND, original_file_name='1.csv')) jobs.append(create_db_job(letter_template, job_status=JOB_STATUS_READY_TO_SEND, original_file_name='2.csv')) - + create_db_job(letter_template, job_status=JOB_STATUS_SENT_TO_DVLA, original_file_name='3.csv') + create_db_job(letter_template, job_status=JOB_STATUS_FINISHED, original_file_name='4.csv') + create_db_job(letter_template, job_status=JOB_STATUS_PENDING, original_file_name='5.csv') result = dao_get_letter_jobs_by_status(JOB_STATUS_READY_TO_SEND) assert len(result) == 2