OnroerendErfgoed / oe_utils

tests/test_email.py

# -*- coding: utf-8 -*-
import unittest

from oe_utils.email import get_oe_email_footer_html, get_oe_email_footer_plain
from oe_utils.email.smtp import SMTPClient
from tests import load_fixture

try:
    from unittest.mock import Mock, patch, MagicMock, call, ANY
except:
    from mock import Mock, patch, MagicMock, call, ANY

exception = Exception('test')
smtp_obj_mock2 = MagicMock()
smtp_obj_mock2.sendmail = Mock(side_effect=exception)


class EmailUtilTests(unittest.TestCase):
    def setUp(self):
        self.smtp_obj_mock = MagicMock()
        self.smtp_obj_mock.sendmail = Mock(return_value=True)
        self.email_content = u"""Dear,

Thank you for your e-mail. Your message has been forwarded to Darth Vader for action as I am in a galaxy far far away.

Kind regards,

The Emperor
    """

    def test_sendmail(self):
        with patch('oe_utils.email.smtp.SMTPClient.smtp_obj') as smtp_mock:
            smtp_mock.__get__ = Mock(return_value=self.smtp_obj_mock)
            email_engine = SMTPClient('smtp', 'test_sender@email.com')
            email_engine.send_mail(['boba@fett.com'], 'test subject', self.email_content)
            self.assertEqual(self.smtp_obj_mock.sendmail.call_count, 1)
            self.smtp_obj_mock.sendmail.assert_called_once_with('test_sender@email.com', ['boba@fett.com'], ANY)

    def test_sendmail_html(self):
        with patch('oe_utils.email.smtp.SMTPClient.smtp_obj') as smtp_mock:
            smtp_mock.__get__ = Mock(return_value=self.smtp_obj_mock)
            email_engine = SMTPClient('smtp', 'test_sender@email.com')
            email_engine.send_mail(['boba@fett.com'], 'test subject', self.email_content, self.email_content)
            self.assertEqual(self.smtp_obj_mock.sendmail.call_count, 1)
            self.smtp_obj_mock.sendmail.assert_called_once_with('test_sender@email.com', ['boba@fett.com'], ANY)

    def test_sendmail_exception(self):
        with patch('oe_utils.email.smtp.SMTPClient.smtp_obj') as smtp_mock:
            smtp_mock.__get__ = Mock(return_value=smtp_obj_mock2)
            email_engine = SMTPClient('smtp', 'test_sender@email.com')
            self.assertRaises(Exception, email_engine.send_mail, ['boba@fett.com'], 'test subject', self.email_content)
            self.assertEqual(smtp_obj_mock2.sendmail.call_count, 1)

    def test_sendmail_cc(self):

This code seems to be duplicated in your project.

        with patch('oe_utils.email.smtp.SMTPClient.smtp_obj') as smtp_mock:
            smtp_mock.__get__ = Mock(return_value=self.smtp_obj_mock)
            email_engine = SMTPClient('smtp', 'test_sender@email.com')
            email_engine.send_mail(['boba@fett.com'], 'test subject', self.email_content,
                                   cc=['vader@galatic-empire.com'])
            self.assertEqual(self.smtp_obj_mock.sendmail.call_count, 1)
            content = self.smtp_obj_mock.sendmail.mock_calls[0][1][2]
            self.smtp_obj_mock.sendmail.assert_called_once_with('test_sender@email.com',
                                                                ['boba@fett.com', 'vader@galatic-empire.com'], ANY)
            self.assertIn('CC', content)
            self.assertIn('galatic-empire', content)

    def test_sendmail_bcc(self):

This code seems to be duplicated in your project.

        with patch('oe_utils.email.smtp.SMTPClient.smtp_obj') as smtp_mock:
            smtp_mock.__get__ = Mock(return_value=self.smtp_obj_mock)
            email_engine = SMTPClient('smtp', 'test_sender@email.com')
            email_engine.send_mail(['boba@fett.com'], 'test subject', self.email_content,
                                   bcc=['vader@galatic-empire.com'])
            self.assertEqual(self.smtp_obj_mock.sendmail.call_count, 1)
            content = self.smtp_obj_mock.sendmail.mock_calls[0][1][2]
            self.smtp_obj_mock.sendmail.assert_called_once_with('test_sender@email.com',
                                                                ['boba@fett.com', 'vader@galatic-empire.com'], ANY)
            self.assertIn('BCC', content)
            self.assertIn('galatic-empire', content)

    def test_sendmail_attachment(self):
        with patch('oe_utils.email.smtp.SMTPClient.smtp_obj') as smtp_mock:
            attachments = [
                {'name': 'test.pdf', 'mime': 'application/pdf', 'data': load_fixture('fixtures/Test_document.pdf')}]
            smtp_mock.__get__ = Mock(return_value=self.smtp_obj_mock)
            email_engine = SMTPClient('smtp', 'test_sender@email.com')
            email_engine.send_mail(['boba@fett.com'], 'test subject', self.email_content, files=attachments)
            self.assertEqual(self.smtp_obj_mock.sendmail.call_count, 1)
            self.smtp_obj_mock.sendmail.assert_called_once_with('test_sender@email.com', ['boba@fett.com'], ANY)

    def test_get_oe_email_footer_html(self):
        template = get_oe_email_footer_html()
        expected = """
    <div style="float:left; width:100%;">
        Onroerend Erfgoed<br/>
        Herman Teirlinckgebouw  |  Havenlaan 88 bus 5  |  1000 Brussel <br/>
        <div style="width:100%">
            <div style="float:left; width:100%">
                <a href='https://www.onroerenderfgoed.be'>www.onroerenderfgoed.be</a>
            </div>
            <div style="float:left; width:100%">
                <img src="https://www.onroerenderfgoed.be/assets/img/logo-vlaanderen.png" height="100">
            </div>
        </div>
    </div>
    """
        self.assertEqual(template, expected)

    def test_get_oe_email_footer_plain(self):
        template = get_oe_email_footer_plain()
        expected = """
    \n
    Onroerend Erfgoed\n
    Herman Teirlinckgebouw  |  Havenlaan 88 bus 5  |  1000 Brussel\n
    www.onroerenderfgoed.be\n
    \n
    """
        self.assertEqual(template, expected)