darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-17 18:52:30 -05:00
Code Issues Packages Projects Releases Wiki Activity

Building tests for the tasks class

Browse Source
This commit is contained in:
Martyn Inglis
2016-02-16 17:17:02 +00:00
parent 18d63e241b
commit 0933e5c647
5 changed files with 81 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
app/celery/tasks.py
View File

@@ -1,12 +1,12 @@
from itsdangerous import URLSafeSerializer
from app import notify_celery, twilio_client, db, encryption
from app.clients.sms.twilio import TwilioClientException
from app.dao.templates_dao import get_model_templates
from app.dao.notifications_dao import save_notification
from app.models import Notification
from flask import current_app
@notify_celery.task(name="send-sms", bind="True")
@notify_celery.task(name="send-sms")
def send_sms(service_id, notification_id, encrypted_notification):
notification = encryption.decrypt(encrypted_notification)
template = get_model_templates(notification['template'])
@@ -14,7 +14,7 @@ def send_sms(service_id, notification_id, encrypted_notification):
status = 'sent'
try:
twilio_client.send_sms(notification, template.content)
twilio_client.send_sms(notification['to'], template.content)
except TwilioClientException as e:
current_app.logger.info(e)
status = 'failed'
@@ -27,5 +27,4 @@ def send_sms(service_id, notification_id, encrypted_notification):
status=status
)
db.session.add(notification_db_object)
db.session.commit()
save_notification(notification_db_object)