source: OpenWorkouts-current/ow/tests/test_mail.py @ b8ef4ab

from unittest.mock import Mock, patch

import pytest

from pyramid.testing import DummyRequest
from pyramid_mailer.message import Message

from ow.models.root import OpenWorkouts
from ow.models.user import User
from ow.utilities import get_verification_token
from ow.mail import idna_encode_recipients, send_verification_email


class TestMail(object):

    @pytest.fixture
    def root(self):
        root = OpenWorkouts()
        user = User(email='user@example.net')
        user.verification_token = get_verification_token()
        root.add_user(user)
        return root

    def test_idna_encode_recipients(self):
        message = Message(subject='s', recipients=['a@a.com'], body='b')
        assert message.recipients == ['a@a.com']
        message = idna_encode_recipients(message)
        assert message.recipients == ['a@a.com']
        message.recipients.append(u'a@\xfc.de')
        assert message.recipients == ['a@a.com', u'a@\xfc.de']
        message = idna_encode_recipients(message)
        assert message.recipients == ['a@a.com', 'a@xn--tda.de']

    @patch('ow.mail.get_mailer')
    @patch('ow.mail.Message')
    @patch('ow.mail.render')
    def test_send_verification_email(self, r, m, gm, root):
        mailer = Mock()
        gm.return_value = mailer
        message = Mock()
        message.recipients = ['user@example.net']
        m.return_value = message

        txt_body = Mock()
        html_body = Mock()
        r.side_effect = [txt_body, html_body]

        request = DummyRequest()
        request.root = root
        user = root.users[0]
        send_verification_email(request, user)
        verify_link = request.resource_url(
            user, 'verify', user.verification_token)

        # two render calls
        assert r.call_count == 2

        # first call renders the text version of the email
        assert r.call_args_list[0][0][0] == (
            'ow:templates/mail_verify_account_txt.pt')
        assert r.call_args_list[0][0][1] == (
            {'user': user, 'verify_link': verify_link})
        assert r.call_args_list[0][0][2] == request

        # second call renders the html version of the email
        assert r.call_args_list[1][0][0] == (
            'ow:templates/mail_verify_account_html.pt')
        assert r.call_args_list[1][0][1] == (
            {'user': user, 'verify_link': verify_link})
        assert r.call_args_list[1][0][2] == request

        m.assert_called_once
        m.call_args_list[0][1]['recipients'] == user.email
        m.call_args_list[0][1]['body'] == txt_body
        m.call_args_list[0][1]['html'] == html_body
        mailer.send_to_queue.assert_called_once_with(message)