darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-03 09:51:11 -05:00
Code Issues Packages Projects Releases Wiki Activity

Improved tests on letter jobs by status

Browse Source
This commit is contained in:
Ken Tsang
2017-08-22 11:05:58 +01:00
parent 18881cd580
commit bd79a6f770
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/app/dao/test_jobs_dao.py
View File

@@ -24,7 +24,7 @@ from app.dao.statistics_dao import create_or_update_job_sending_statistics, upda
from app.models import ( from app.models import (
Job, JobStatistics, Job, JobStatistics,
EMAIL_TYPE, SMS_TYPE, LETTER_TYPE, 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 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 = []
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='1.csv'))
jobs.append(create_db_job(letter_template, job_status=JOB_STATUS_READY_TO_SEND, original_file_name='2.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) result = dao_get_letter_jobs_by_status(JOB_STATUS_READY_TO_SEND)
assert len(result) == 2 assert len(result) == 2