"""Test SNS utility functions."""

from accounting.util import sns


def test_get_sns_resources(mocker):
    """Test SNS resource helper."""
    mocked_sns = mocker.Mock()
    mocked_resource = mocker.patch('boto3.resource')
    mocked_resource.return_value = mocked_sns

    assert sns.get_sns_resource() == mocked_sns


def test_publish_sns_message_to_email(mocker):
    """Test SNS publish is being called."""
    mocked_sns = mocker.patch('boto3.resource')
    sns.publish_sns_message_to_email('message', 'topic', mocked_sns)
    mocked_sns.Topic.assert_called_once_with('topic')
    mocked_sns.Topic.return_value.publish.assert_called_once_with(
        Message='message', MessageStructure='string')