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

import pytest
from flask import current_app

from app import get_aws_sns_client


def test_send_sms_successful_returns_aws_sns_response(notify_api, mocker):

    to = "16135555555"
    content = reference = "foo"
    with notify_api.app_context():
        aws_sns_client = get_aws_sns_client()
        aws_sns_client.init_app(current_app)
        boto_mock = mocker.patch.object(aws_sns_client, "_client", create=True)
        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
):
    with notify_api.app_context():
        aws_sns_client = get_aws_sns_client()
        aws_sns_client.init_app(current_app)
        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)