notifications-api/tests/app/clients/test_aws_sns.py

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)