mirror of
https://github.com/GSA/notifications-api.git
synced 2026-02-01 07:35:34 -05:00
implement SNS
This commit is contained in:
24
tests/app/clients/test_aws_sns.py
Normal file
24
tests/app/clients/test_aws_sns.py
Normal file
@@ -0,0 +1,24 @@
|
||||
import pytest
|
||||
from app import aws_sns_client
|
||||
def test_send_sms_successful_returns_aws_sns_response(notify_api, mocker):
|
||||
boto_mock = mocker.patch.object(aws_sns_client, '_client', create=True)
|
||||
mocker.patch.object(aws_sns_client, 'statsd_client', create=True)
|
||||
to = "6135555555"
|
||||
content = reference = 'foo'
|
||||
with notify_api.app_context():
|
||||
aws_sns_client.send_sms(to, content, reference)
|
||||
boto_mock.publish.assert_called_once_with(
|
||||
PhoneNumber="+16135555555",
|
||||
Message=content,
|
||||
MessageAttributes={'AWS.SNS.SMS.SMSType': {'DataType': 'String', 'StringValue': 'Transactional'}}
|
||||
)
|
||||
|
||||
|
||||
def test_send_sms_returns_raises_error_if_there_is_no_valid_number_is_found(notify_api, mocker):
|
||||
mocker.patch.object(aws_sns_client, '_client', create=True)
|
||||
mocker.patch.object(aws_sns_client, 'statsd_client', create=True)
|
||||
to = ""
|
||||
content = reference = 'foo'
|
||||
with pytest.raises(ValueError) as excinfo:
|
||||
aws_sns_client.send_sms(to, content, reference)
|
||||
assert 'No valid numbers found for SMS delivery' in str(excinfo.value)
|
||||
Reference in New Issue
Block a user