From 9df498106f8b6be03f7a6a93ff8c22d84ab6ec8d Mon Sep 17 00:00:00 2001 From: Katie Smith Date: Tue, 29 Aug 2017 16:26:55 +0100 Subject: [PATCH] Add one_off argument to create_notification test function One off messages have no API key or job ID. If one_off is set to False, an API key will automatically be added. --- tests/app/db.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/app/db.py b/tests/app/db.py index 8283428c9..e61270b81 100644 --- a/tests/app/db.py +++ b/tests/app/db.py @@ -123,7 +123,8 @@ def create_notification( international=False, phone_prefix=None, scheduled_for=None, - normalised_to=None + normalised_to=None, + one_off=False, ): if created_at is None: created_at = datetime.utcnow() @@ -132,7 +133,7 @@ def create_notification( sent_at = sent_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 api_key = ApiKey.query.filter(ApiKey.service == template.service, ApiKey.key_type == key_type).first() if not api_key: