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

Celery task to send the sms verify code.

Browse Source 
Each celery task will use it's own queue.
This commit is contained in:
Rebecca Law
2016-02-17 15:41:33 +00:00
parent d6bea65626
commit d022d036dc
7 changed files with 57 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
tests/app/celery/test_tasks.py
View File

@@ -1,7 +1,7 @@
import uuid
import pytest
from app.celery.tasks import send_sms
from app import twilio_client
from app.celery.tasks import (send_sms, send_sms_code)
from app import twilio_client, encryption
from app.clients.sms.twilio import TwilioClientException
from app.dao import notifications_dao
from sqlalchemy.exc import SQLAlchemyError
@@ -74,3 +74,24 @@ def test_should_not_send_sms_if_db_peristance_failed(sample_template, mocker):
with pytest.raises(NoResultFound) as e:
notifications_dao.get_notification(sample_template.service_id, notification_id)
assert 'No row was found for one' in str(e.value)
def test_should_send_sms_code(mocker):
notification = {'to': '+441234123123',
'secret_code': '12345'}
encrypted_notification = encryption.encrypt(notification)
mocker.patch('app.twilio_client.send_sms')
send_sms_code(encrypted_notification)
twilio_client.send_sms.assert_called_once_with(notification['to'], notification['secret_code'])
def test_should_throw_twilio_exception(mocker):
notification = {'to': '+441234123123',
'secret_code': '12345'}
encrypted_notification = encryption.encrypt(notification)
mocker.patch('app.twilio_client.send_sms', side_effect=TwilioClientException)
send_sms_code(encrypted_notification)
twilio_client.send_sms.assert_called_once_with(notification['to'], notification['secret_code'])