86 lines
3.2 KiB
Python
86 lines
3.2 KiB
Python
from __future__ import absolute_import, unicode_literals
|
|
|
|
from django.core import mail
|
|
|
|
from mayan.apps.documents.tests.base import GenericDocumentTestCase
|
|
|
|
from .literals import (
|
|
TEST_EMAIL_BODY_HTML, TEST_EMAIL_ADDRESS, TEST_EMAIL_FROM_ADDRESS,
|
|
TEST_RECIPIENTS_MULTIPLE_COMMA, TEST_RECIPIENTS_MULTIPLE_COMMA_RESULT,
|
|
TEST_RECIPIENTS_MULTIPLE_SEMICOLON,
|
|
TEST_RECIPIENTS_MULTIPLE_SEMICOLON_RESULT, TEST_RECIPIENTS_MULTIPLE_MIXED,
|
|
TEST_RECIPIENTS_MULTIPLE_MIXED_RESULT,
|
|
)
|
|
from .mixins import MailerTestMixin
|
|
|
|
|
|
class ModelTestCase(MailerTestMixin, GenericDocumentTestCase):
|
|
def test_send_simple(self):
|
|
self._create_test_user_mailer()
|
|
self.test_user_mailer.send(to=TEST_EMAIL_ADDRESS)
|
|
|
|
self.assertEqual(len(mail.outbox), 1)
|
|
self.assertEqual(mail.outbox[0].from_email, TEST_EMAIL_FROM_ADDRESS)
|
|
self.assertEqual(mail.outbox[0].to, [TEST_EMAIL_ADDRESS])
|
|
|
|
def test_send_simple_with_html(self):
|
|
self._create_test_user_mailer()
|
|
self.test_user_mailer.send(
|
|
to=TEST_EMAIL_ADDRESS, body=TEST_EMAIL_BODY_HTML
|
|
)
|
|
|
|
self.assertEqual(len(mail.outbox), 1)
|
|
self.assertEqual(mail.outbox[0].from_email, TEST_EMAIL_FROM_ADDRESS)
|
|
self.assertEqual(mail.outbox[0].to, [TEST_EMAIL_ADDRESS])
|
|
self.assertEqual(
|
|
mail.outbox[0].alternatives[0][0], TEST_EMAIL_BODY_HTML
|
|
)
|
|
|
|
def test_send_attachment(self):
|
|
self._create_test_user_mailer()
|
|
self.test_user_mailer.send_document(
|
|
to=TEST_EMAIL_ADDRESS, document=self.test_document,
|
|
as_attachment=True
|
|
)
|
|
|
|
self.assertEqual(len(mail.outbox), 1)
|
|
self.assertEqual(mail.outbox[0].from_email, TEST_EMAIL_FROM_ADDRESS)
|
|
self.assertEqual(mail.outbox[0].to, [TEST_EMAIL_ADDRESS])
|
|
with self.test_document.open() as file_object:
|
|
self.assertEqual(
|
|
mail.outbox[0].attachments[0], (
|
|
self.test_document.label, file_object.read(),
|
|
self.test_document.file_mimetype
|
|
)
|
|
)
|
|
|
|
def test_send_multiple_recipients_comma(self):
|
|
self._create_test_user_mailer()
|
|
self.test_user_mailer.send(to=TEST_RECIPIENTS_MULTIPLE_COMMA)
|
|
|
|
self.assertEqual(len(mail.outbox), 1)
|
|
self.assertEqual(mail.outbox[0].from_email, TEST_EMAIL_FROM_ADDRESS)
|
|
self.assertEqual(
|
|
mail.outbox[0].to, TEST_RECIPIENTS_MULTIPLE_COMMA_RESULT
|
|
)
|
|
|
|
def test_send_multiple_recipients_semicolon(self):
|
|
self._create_test_user_mailer()
|
|
self.test_user_mailer.send(to=TEST_RECIPIENTS_MULTIPLE_SEMICOLON)
|
|
|
|
self.assertEqual(len(mail.outbox), 1)
|
|
self.assertEqual(mail.outbox[0].from_email, TEST_EMAIL_FROM_ADDRESS)
|
|
self.assertEqual(
|
|
mail.outbox[0].to, TEST_RECIPIENTS_MULTIPLE_SEMICOLON_RESULT
|
|
)
|
|
|
|
def test_send_multiple_recipient_mixed(self):
|
|
self._create_test_user_mailer()
|
|
self.test_user_mailer.send(to=TEST_RECIPIENTS_MULTIPLE_MIXED)
|
|
|
|
self.assertEqual(len(mail.outbox), 1)
|
|
self.assertEqual(mail.outbox[0].from_email, TEST_EMAIL_FROM_ADDRESS)
|
|
self.assertEqual(
|
|
mail.outbox[0].to, TEST_RECIPIENTS_MULTIPLE_MIXED_RESULT
|
|
)
|