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

[WIP]

Browse Source
This commit is contained in:
Rebecca Law
2019-12-11 15:24:31 +00:00
parent 22851c4ef0
commit 5482c03bca
4 changed files with 40 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tests/app/dao/test_returned_letters_dao.py
View File

@@ -2,7 +2,10 @@ from datetime import datetime, timedelta, date
from freezegun import freeze_time
from app.dao.returned_letters_dao import insert_or_update_returned_letters, get_returned_letter_summary
from app.dao.returned_letters_dao import (
insert_or_update_returned_letters, get_returned_letter_summary,
fetch_returned_letters
)
from app.models import ReturnedLetter
from tests.app.db import create_notification, create_notification_history, create_returned_letter
@@ -116,3 +119,16 @@ def test_get_returned_letter_summary_orders_by_reported_at(sample_service):
assert results[0].returned_letter_count == 3
assert results[1].reported_at == last_month.date()
assert results[1].returned_letter_count == 2
def test_fetch_returned_letters(sample_service):
today = datetime.now()
last_month = datetime.now() - timedelta(days=30)
create_returned_letter(service=sample_service, reported_at=today)
create_returned_letter(service=sample_service, reported_at=today)
create_returned_letter(service=sample_service, reported_at=last_month)
results = fetch_returned_letters(service_id=sample_service.id, report_date=today.date())
assert len(results) == 2