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

Add dao method to get notification by reference with tests

Browse Source
This commit is contained in:
Imdad Ahad
2016-12-07 14:08:22 +00:00
parent f02869c1e6
commit d4d733f997
3 changed files with 25 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/app/conftest.py
View File

@@ -411,7 +411,8 @@ def sample_notification(notify_db,
personalisation=None,
api_key_id=None,
key_type=KEY_TYPE_NORMAL,
sent_by=None):
sent_by=None,
client_reference=None):
if created_at is None:
created_at = datetime.utcnow()
if service is None:
@@ -445,7 +446,8 @@ def sample_notification(notify_db,
'api_key_id': api_key_id,
'key_type': key_type,
'sent_by': sent_by,
'updated_at': created_at if status in NOTIFICATION_STATUS_TYPES_COMPLETED else None
'updated_at': created_at if status in NOTIFICATION_STATUS_TYPES_COMPLETED else None,
'client_reference': client_reference
}
if job_row_number:
data['job_row_number'] = job_row_number