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

import pytest

from app.clients.sms import SmsClient, SmsClientResponseException


@pytest.fixture
def fake_client(notify_api):
    class FakeSmsClient(SmsClient):
        @property
        def name(self):
            return "fake"

        def init_app(self, current_app, *args, **kwargs):
            pass

        def send_sms(self, *args, **kwargs):
            pass

    fake_client = FakeSmsClient()
    # fake_client.init_app(notify_api)
    return fake_client


def test_send_sms(fake_client, mocker):
    mock_send = mocker.patch.object(fake_client, "send_sms")

    fake_client.send_sms(
        to="to",
        content="content",
        reference="reference",
        international=False,
        sender="testing",
    )

    mock_send.assert_called_with(
        to="to",
        content="content",
        reference="reference",
        international=False,
        sender="testing",
    )


def test_send_sms_error(fake_client, mocker):
    mocker.patch.object(
        fake_client, "send_sms", side_effect=SmsClientResponseException("error")
    )

    with pytest.raises(SmsClientResponseException):
        fake_client.send_sms(
            to="to",
            content="content",
            reference="reference",
            international=False,
            sender=None,
        )