from email.mime.multipart import MIMEMultipart
from unittest.mock import Mock

import pytest

from ...src import aws


class TestSendRawEmail:

    @pytest.fixture(autouse=True)
    def patcher_ses_client(self, mocker):
        """Patch the SES client."""
        return mocker.patch.object(aws, "new_ses_client")

    def test_send_raw_email(self, patcher_ses_client):
        """Test that the send_raw_email function returns a dict."""
        mock_msg = Mock(spec=MIMEMultipart)
        aws.send_raw_email(
            ["john.doe@theorchard.com", "jane.doe@theorchard.com"],
            mock_msg,
        )
        assert patcher_ses_client.return_value.send_raw_email.called