mirror of
https://github.com/pacnpal/django-anymail.git
synced 2025-12-20 03:41:05 -05:00
1bec172611
Cherry-picked from: 8c26807a - Add runtests.py for testing separately from other Django apps cd8504b1 - Make tests compatible with setuptools 4ac65b78 - Set up testing on the backend API calls, using mock
118 lines
4.7 KiB
Python
118 lines
4.7 KiB
Python
from mock import patch
|
|
|
|
from django.conf import settings
|
|
from django.core import mail
|
|
from django.core.exceptions import ImproperlyConfigured
|
|
from django.test import TestCase
|
|
from django.utils import simplejson as json
|
|
|
|
from djrill.mail import DjrillMessage
|
|
|
|
class DjrillBackendMockAPITestCase(TestCase):
|
|
"""TestCase that uses Djrill EmailBackend with a mocked Mandrill API"""
|
|
|
|
class MockResponse:
|
|
"""requests.post return value mock sufficient for DjrillBackend"""
|
|
def __init__(self, status_code=200):
|
|
self.status_code = status_code
|
|
|
|
def setUp(self):
|
|
self.patch = patch('requests.post')
|
|
self.mock_post = self.patch.start()
|
|
self.mock_post.return_value = self.MockResponse()
|
|
|
|
settings.MANDRILL_API_KEY = "FAKE_API_KEY_FOR_TESTING"
|
|
settings.MANDRILL_API_URL = "http://mandrillapp.com/api/1.0"
|
|
|
|
# Django TestCase sets up locmem EmailBackend; override it here
|
|
self.original_email_backend = settings.EMAIL_BACKEND
|
|
settings.EMAIL_BACKEND = "djrill.mail.backends.djrill.DjrillBackend"
|
|
|
|
def tearDown(self):
|
|
self.patch.stop()
|
|
settings.EMAIL_BACKEND = self.original_email_backend
|
|
|
|
def get_api_call_data(self):
|
|
"""Returns the data posted to the Mandrill API.
|
|
|
|
Fails test if API wasn't called.
|
|
"""
|
|
if self.mock_post.call_args is None:
|
|
raise AssertionError("Mandrill API was not called")
|
|
(args, kwargs) = self.mock_post.call_args
|
|
if 'data' not in kwargs:
|
|
raise AssertionError("requests.post was called without data kwarg "
|
|
"-- Maybe tests need to be updated for backend changes?")
|
|
return json.loads(kwargs['data'])
|
|
|
|
|
|
class DjrillBackendTests(DjrillBackendMockAPITestCase):
|
|
"""Test Djrill's support for Django mail wrappers"""
|
|
|
|
def test_send_mail(self):
|
|
mail.send_mail('Subject here', 'Here is the message.',
|
|
'from@example.com', ['to@example.com'], fail_silently=False)
|
|
data = self.get_api_call_data()
|
|
self.assertEqual(data['message']['subject'], "Subject here")
|
|
self.assertEqual(data['message']['text'], "Here is the message.")
|
|
self.assertEqual(data['message']['from_email'], "from@example.com")
|
|
self.assertEqual(len(data['message']['to']), 1)
|
|
self.assertEqual(data['message']['to'][0]['email'], "to@example.com")
|
|
|
|
def test_missing_api_key(self):
|
|
del settings.MANDRILL_API_KEY
|
|
with self.assertRaises(ImproperlyConfigured):
|
|
mail.send_mail('Subject', 'Message', 'from@example.com',
|
|
['to@example.com'])
|
|
|
|
|
|
class DjrillMessageTests(TestCase):
|
|
def setUp(self):
|
|
self.subject = "Djrill baby djrill."
|
|
self.from_name = "Tarzan"
|
|
self.from_email = "test@example"
|
|
self.to = ["King Kong <kingkong@example.com>",
|
|
"Cheetah <cheetah@example.com", "bubbles@example.com"]
|
|
self.text_content = "Wonderful fallback text content."
|
|
self.html_content = "<h1>That's a nice HTML email right there.</h1>"
|
|
self.headers = {"Reply-To": "tarzan@example.com"}
|
|
self.tags = ["track", "this"]
|
|
|
|
def test_djrill_message_success(self):
|
|
msg = DjrillMessage(self.subject, self.text_content, self.from_email,
|
|
self.to, tags=self.tags, headers=self.headers,
|
|
from_name=self.from_name)
|
|
|
|
self.assertIsInstance(msg, DjrillMessage)
|
|
self.assertEqual(msg.body, self.text_content)
|
|
self.assertEqual(msg.recipients(), self.to)
|
|
self.assertEqual(msg.tags, self.tags)
|
|
self.assertEqual(msg.extra_headers, self.headers)
|
|
self.assertEqual(msg.from_name, self.from_name)
|
|
|
|
def test_djrill_message_html_success(self):
|
|
msg = DjrillMessage(self.subject, self.text_content, self.from_email,
|
|
self.to, tags=self.tags)
|
|
msg.attach_alternative(self.html_content, "text/html")
|
|
|
|
self.assertEqual(msg.alternatives[0][0], self.html_content)
|
|
|
|
def test_djrill_message_tag_failure(self):
|
|
with self.assertRaises(ImproperlyConfigured):
|
|
DjrillMessage(self.subject, self.text_content, self.from_email,
|
|
self.to, tags=["_fail"])
|
|
|
|
def test_djrill_message_tag_skip(self):
|
|
"""
|
|
Test that tags over 50 chars are not included in the tags list.
|
|
"""
|
|
tags = ["works", "awesomesauce",
|
|
"iwilltestmycodeiwilltestmycodeiwilltestmycodeiwilltestmycode"]
|
|
msg = DjrillMessage(self.subject, self.text_content, self.from_email,
|
|
self.to, tags=tags)
|
|
|
|
self.assertIsInstance(msg, DjrillMessage)
|
|
self.assertIn(tags[0], msg.tags)
|
|
self.assertIn(tags[1], msg.tags)
|
|
self.assertNotIn(tags[2], msg.tags)
|