mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-20 15:31:15 -05:00
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
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)
|
|
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",
|
|
},
|
|
"AWS.MM.SMS.OriginationNumber": {
|
|
"DataType": "String",
|
|
"StringValue": "+18556438890",
|
|
},
|
|
},
|
|
)
|
|
|
|
|
|
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)
|
|
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)
|