darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-22 08:21:13 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add one_off argument to create_notification test function

Browse Source 
One off messages have no API key or job ID. If one_off is set to False,
an API key will automatically be added.
This commit is contained in:
Katie Smith
2017-08-29 16:26:55 +01:00
parent 5c25b12b0f
commit 9df498106f
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/app/db.py
View File

@@ -123,7 +123,8 @@ def create_notification(
international=False, international=False,
phone_prefix=None, phone_prefix=None,
scheduled_for=None, scheduled_for=None,
normalised_to=None normalised_to=None,
one_off=False,
): ):
if created_at is None: if created_at is None:
created_at = datetime.utcnow() created_at = datetime.utcnow()
@@ -132,7 +133,7 @@ def create_notification(
sent_at = sent_at or datetime.utcnow() sent_at = sent_at or datetime.utcnow()
updated_at = updated_at or datetime.utcnow() updated_at = updated_at or datetime.utcnow()
if job is None and api_key is None: if not one_off and (job is None and api_key is None):
# we didn't specify in test - lets create it # we didn't specify in test - lets create it
api_key = ApiKey.query.filter(ApiKey.service == template.service, ApiKey.key_type == key_type).first() api_key = ApiKey.query.filter(ApiKey.service == template.service, ApiKey.key_type == key_type).first()
if not api_key: if not api_key: