--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/parts/django/tests/regressiontests/mail/tests.py Sat Jan 08 11:20:57 2011 +0530
@@ -0,0 +1,375 @@
+# coding: utf-8
+import email
+import os
+import shutil
+import sys
+import tempfile
+from StringIO import StringIO
+from django.conf import settings
+from django.core import mail
+from django.core.mail import EmailMessage, mail_admins, mail_managers, EmailMultiAlternatives
+from django.core.mail import send_mail, send_mass_mail
+from django.core.mail.backends.base import BaseEmailBackend
+from django.core.mail.backends import console, dummy, locmem, filebased, smtp
+from django.core.mail.message import BadHeaderError
+from django.test import TestCase
+from django.utils.translation import ugettext_lazy
+
+class MailTests(TestCase):
+
+ def test_ascii(self):
+ email = EmailMessage('Subject', 'Content', 'from@example.com', ['to@example.com'])
+ message = email.message()
+ self.assertEqual(message['Subject'].encode(), 'Subject')
+ self.assertEqual(message.get_payload(), 'Content')
+ self.assertEqual(message['From'], 'from@example.com')
+ self.assertEqual(message['To'], 'to@example.com')
+
+ def test_multiple_recipients(self):
+ email = EmailMessage('Subject', 'Content', 'from@example.com', ['to@example.com','other@example.com'])
+ message = email.message()
+ self.assertEqual(message['Subject'].encode(), 'Subject')
+ self.assertEqual(message.get_payload(), 'Content')
+ self.assertEqual(message['From'], 'from@example.com')
+ self.assertEqual(message['To'], 'to@example.com, other@example.com')
+
+ def test_header_injection(self):
+ email = EmailMessage('Subject\nInjection Test', 'Content', 'from@example.com', ['to@example.com'])
+ self.assertRaises(BadHeaderError, email.message)
+ email = EmailMessage(ugettext_lazy('Subject\nInjection Test'), 'Content', 'from@example.com', ['to@example.com'])
+ self.assertRaises(BadHeaderError, email.message)
+
+ def test_space_continuation(self):
+ """
+ Test for space continuation character in long (ascii) subject headers (#7747)
+ """
+ email = EmailMessage('Long subject lines that get wrapped should use a space continuation character to get expected behaviour in Outlook and Thunderbird', 'Content', 'from@example.com', ['to@example.com'])
+ message = email.message()
+ self.assertEqual(message['Subject'], 'Long subject lines that get wrapped should use a space continuation\n character to get expected behaviour in Outlook and Thunderbird')
+
+ def test_message_header_overrides(self):
+ """
+ Specifying dates or message-ids in the extra headers overrides the
+ default values (#9233)
+ """
+ headers = {"date": "Fri, 09 Nov 2001 01:08:47 -0000", "Message-ID": "foo"}
+ email = EmailMessage('subject', 'content', 'from@example.com', ['to@example.com'], headers=headers)
+ self.assertEqual(email.message().as_string(), 'Content-Type: text/plain; charset="utf-8"\nMIME-Version: 1.0\nContent-Transfer-Encoding: quoted-printable\nSubject: subject\nFrom: from@example.com\nTo: to@example.com\ndate: Fri, 09 Nov 2001 01:08:47 -0000\nMessage-ID: foo\n\ncontent')
+
+ def test_empty_admins(self):
+ """
+ Test that mail_admins/mail_managers doesn't connect to the mail server
+ if there are no recipients (#9383)
+ """
+ old_admins = settings.ADMINS
+ old_managers = settings.MANAGERS
+
+ settings.ADMINS = settings.MANAGERS = [('nobody','nobody@example.com')]
+ mail.outbox = []
+ mail_admins('hi', 'there')
+ self.assertEqual(len(mail.outbox), 1)
+ mail.outbox = []
+ mail_managers('hi', 'there')
+ self.assertEqual(len(mail.outbox), 1)
+
+ settings.ADMINS = settings.MANAGERS = []
+ mail.outbox = []
+ mail_admins('hi', 'there')
+ self.assertEqual(len(mail.outbox), 0)
+ mail.outbox = []
+ mail_managers('hi', 'there')
+ self.assertEqual(len(mail.outbox), 0)
+
+ settings.ADMINS = old_admins
+ settings.MANAGERS = old_managers
+
+ def test_from_header(self):
+ """
+ Make sure we can manually set the From header (#9214)
+ """
+ email = EmailMessage('Subject', 'Content', 'bounce@example.com', ['to@example.com'], headers={'From': 'from@example.com'})
+ message = email.message()
+ self.assertEqual(message['From'], 'from@example.com')
+
+ def test_multiple_message_call(self):
+ """
+ Regression for #13259 - Make sure that headers are not changed when
+ calling EmailMessage.message()
+ """
+ email = EmailMessage('Subject', 'Content', 'bounce@example.com', ['to@example.com'], headers={'From': 'from@example.com'})
+ message = email.message()
+ self.assertEqual(message['From'], 'from@example.com')
+ message = email.message()
+ self.assertEqual(message['From'], 'from@example.com')
+
+ def test_unicode_header(self):
+ """
+ Regression for #11144 - When a to/from/cc header contains unicode,
+ make sure the email addresses are parsed correctly (especially with
+ regards to commas)
+ """
+ email = EmailMessage('Subject', 'Content', 'from@example.com', ['"Firstname Sürname" <to@example.com>','other@example.com'])
+ self.assertEqual(email.message()['To'], '=?utf-8?q?Firstname_S=C3=BCrname?= <to@example.com>, other@example.com')
+ email = EmailMessage('Subject', 'Content', 'from@example.com', ['"Sürname, Firstname" <to@example.com>','other@example.com'])
+ self.assertEqual(email.message()['To'], '=?utf-8?q?S=C3=BCrname=2C_Firstname?= <to@example.com>, other@example.com')
+
+ def test_safe_mime_multipart(self):
+ """
+ Make sure headers can be set with a different encoding than utf-8 in
+ SafeMIMEMultipart as well
+ """
+ headers = {"Date": "Fri, 09 Nov 2001 01:08:47 -0000", "Message-ID": "foo"}
+ subject, from_email, to = 'hello', 'from@example.com', '"Sürname, Firstname" <to@example.com>'
+ text_content = 'This is an important message.'
+ html_content = '<p>This is an <strong>important</strong> message.</p>'
+ msg = EmailMultiAlternatives('Message from Firstname Sürname', text_content, from_email, [to], headers=headers)
+ msg.attach_alternative(html_content, "text/html")
+ msg.encoding = 'iso-8859-1'
+ self.assertEqual(msg.message()['To'], '=?iso-8859-1?q?S=FCrname=2C_Firstname?= <to@example.com>')
+ self.assertEqual(msg.message()['Subject'].encode(), u'=?iso-8859-1?q?Message_from_Firstname_S=FCrname?=')
+
+ def test_encoding(self):
+ """
+ Regression for #12791 - Encode body correctly with other encodings
+ than utf-8
+ """
+ email = EmailMessage('Subject', 'Firstname Sürname is a great guy.', 'from@example.com', ['other@example.com'])
+ email.encoding = 'iso-8859-1'
+ message = email.message()
+ self.assertTrue(message.as_string().startswith('Content-Type: text/plain; charset="iso-8859-1"\nMIME-Version: 1.0\nContent-Transfer-Encoding: quoted-printable\nSubject: Subject\nFrom: from@example.com\nTo: other@example.com'))
+ self.assertEqual(message.get_payload(), 'Firstname S=FCrname is a great guy.')
+
+ # Make sure MIME attachments also works correctly with other encodings than utf-8
+ text_content = 'Firstname Sürname is a great guy.'
+ html_content = '<p>Firstname Sürname is a <strong>great</strong> guy.</p>'
+ msg = EmailMultiAlternatives('Subject', text_content, 'from@example.com', ['to@example.com'])
+ msg.encoding = 'iso-8859-1'
+ msg.attach_alternative(html_content, "text/html")
+ self.assertEqual(msg.message().get_payload(0).as_string(), 'Content-Type: text/plain; charset="iso-8859-1"\nMIME-Version: 1.0\nContent-Transfer-Encoding: quoted-printable\n\nFirstname S=FCrname is a great guy.')
+ self.assertEqual(msg.message().get_payload(1).as_string(), 'Content-Type: text/html; charset="iso-8859-1"\nMIME-Version: 1.0\nContent-Transfer-Encoding: quoted-printable\n\n<p>Firstname S=FCrname is a <strong>great</strong> guy.</p>')
+
+ def test_attachments(self):
+ """Regression test for #9367"""
+ headers = {"Date": "Fri, 09 Nov 2001 01:08:47 -0000", "Message-ID": "foo"}
+ subject, from_email, to = 'hello', 'from@example.com', 'to@example.com'
+ text_content = 'This is an important message.'
+ html_content = '<p>This is an <strong>important</strong> message.</p>'
+ msg = EmailMultiAlternatives(subject, text_content, from_email, [to], headers=headers)
+ msg.attach_alternative(html_content, "text/html")
+ msg.attach("an attachment.pdf", "%PDF-1.4.%...", mimetype="application/pdf")
+ msg_str = msg.message().as_string()
+ message = email.message_from_string(msg_str)
+ self.assertTrue(message.is_multipart())
+ self.assertEqual(message.get_content_type(), 'multipart/mixed')
+ self.assertEqual(message.get_default_type(), 'text/plain')
+ payload = message.get_payload()
+ self.assertEqual(payload[0].get_content_type(), 'multipart/alternative')
+ self.assertEqual(payload[1].get_content_type(), 'application/pdf')
+
+ def test_arbitrary_stream(self):
+ """
+ Test that the console backend can be pointed at an arbitrary stream.
+ """
+ s = StringIO()
+ connection = mail.get_connection('django.core.mail.backends.console.EmailBackend', stream=s)
+ send_mail('Subject', 'Content', 'from@example.com', ['to@example.com'], connection=connection)
+ self.assertTrue(s.getvalue().startswith('Content-Type: text/plain; charset="utf-8"\nMIME-Version: 1.0\nContent-Transfer-Encoding: quoted-printable\nSubject: Subject\nFrom: from@example.com\nTo: to@example.com\nDate: '))
+
+ def test_stdout(self):
+ """Make sure that the console backend writes to stdout by default"""
+ old_stdout = sys.stdout
+ sys.stdout = StringIO()
+ connection = console.EmailBackend()
+ email = EmailMessage('Subject', 'Content', 'bounce@example.com', ['to@example.com'], headers={'From': 'from@example.com'})
+ connection.send_messages([email])
+ self.assertTrue(sys.stdout.getvalue().startswith('Content-Type: text/plain; charset="utf-8"\nMIME-Version: 1.0\nContent-Transfer-Encoding: quoted-printable\nSubject: Subject\nFrom: from@example.com\nTo: to@example.com\nDate: '))
+ sys.stdout = old_stdout
+
+ def test_dummy(self):
+ """
+ Make sure that dummy backends returns correct number of sent messages
+ """
+ connection = dummy.EmailBackend()
+ email = EmailMessage('Subject', 'Content', 'bounce@example.com', ['to@example.com'], headers={'From': 'from@example.com'})
+ self.assertEqual(connection.send_messages([email, email, email]), 3)
+
+ def test_locmem(self):
+ """
+ Make sure that the locmen backend populates the outbox.
+ """
+ mail.outbox = []
+ connection = locmem.EmailBackend()
+ email1 = EmailMessage('Subject', 'Content', 'bounce@example.com', ['to@example.com'], headers={'From': 'from@example.com'})
+ email2 = EmailMessage('Subject 2', 'Content', 'bounce@example.com', ['to@example.com'], headers={'From': 'from@example.com'})
+ connection.send_messages([email1, email2])
+ self.assertEqual(len(mail.outbox), 2)
+ self.assertEqual(mail.outbox[0].subject, 'Subject')
+ self.assertEqual(mail.outbox[1].subject, 'Subject 2')
+
+ # Make sure that multiple locmem connections share mail.outbox
+ mail.outbox = []
+ connection2 = locmem.EmailBackend()
+ email = EmailMessage('Subject', 'Content', 'bounce@example.com', ['to@example.com'], headers={'From': 'from@example.com'})
+ connection.send_messages([email])
+ connection2.send_messages([email])
+ self.assertEqual(len(mail.outbox), 2)
+
+ def test_file_backend(self):
+ tmp_dir = tempfile.mkdtemp()
+ connection = filebased.EmailBackend(file_path=tmp_dir)
+ email1 = EmailMessage('Subject', 'Content', 'bounce@example.com', ['to@example.com'], headers={'From': 'from@example.com'})
+ connection.send_messages([email1])
+ self.assertEqual(len(os.listdir(tmp_dir)), 1)
+ message = email.message_from_file(open(os.path.join(tmp_dir, os.listdir(tmp_dir)[0])))
+ self.assertEqual(message.get_content_type(), 'text/plain')
+ self.assertEqual(message.get('subject'), 'Subject')
+ self.assertEqual(message.get('from'), 'from@example.com')
+ self.assertEqual(message.get('to'), 'to@example.com')
+ connection2 = filebased.EmailBackend(file_path=tmp_dir)
+ connection2.send_messages([email1])
+ self.assertEqual(len(os.listdir(tmp_dir)), 2)
+ connection.send_messages([email1])
+ self.assertEqual(len(os.listdir(tmp_dir)), 2)
+ email1.connection = filebased.EmailBackend(file_path=tmp_dir)
+ connection_created = connection.open()
+ email1.send()
+ self.assertEqual(len(os.listdir(tmp_dir)), 3)
+ email1.send()
+ self.assertEqual(len(os.listdir(tmp_dir)), 3)
+ connection.close()
+ shutil.rmtree(tmp_dir)
+
+ def test_arbitrary_keyword(self):
+ """
+ Make sure that get_connection() accepts arbitrary keyword that might be
+ used with custom backends.
+ """
+ c = mail.get_connection(fail_silently=True, foo='bar')
+ self.assertTrue(c.fail_silently)
+
+ def test_custom_backend(self):
+ """Test custom backend defined in this suite."""
+ conn = mail.get_connection('regressiontests.mail.custombackend.EmailBackend')
+ self.assertTrue(hasattr(conn, 'test_outbox'))
+ email = EmailMessage('Subject', 'Content', 'bounce@example.com', ['to@example.com'], headers={'From': 'from@example.com'})
+ conn.send_messages([email])
+ self.assertEqual(len(conn.test_outbox), 1)
+
+ def test_backend_arg(self):
+ """Test backend argument of mail.get_connection()"""
+ self.assertTrue(isinstance(mail.get_connection('django.core.mail.backends.smtp.EmailBackend'), smtp.EmailBackend))
+ self.assertTrue(isinstance(mail.get_connection('django.core.mail.backends.locmem.EmailBackend'), locmem.EmailBackend))
+ self.assertTrue(isinstance(mail.get_connection('django.core.mail.backends.dummy.EmailBackend'), dummy.EmailBackend))
+ self.assertTrue(isinstance(mail.get_connection('django.core.mail.backends.console.EmailBackend'), console.EmailBackend))
+ tmp_dir = tempfile.mkdtemp()
+ self.assertTrue(isinstance(mail.get_connection('django.core.mail.backends.filebased.EmailBackend', file_path=tmp_dir), filebased.EmailBackend))
+ shutil.rmtree(tmp_dir)
+ self.assertTrue(isinstance(mail.get_connection(), locmem.EmailBackend))
+
+ def test_connection_arg(self):
+ """Test connection argument to send_mail(), et. al."""
+ connection = mail.get_connection('django.core.mail.backends.locmem.EmailBackend')
+
+ mail.outbox = []
+ send_mail('Subject', 'Content', 'from@example.com', ['to@example.com'], connection=connection)
+ self.assertEqual(len(mail.outbox), 1)
+ message = mail.outbox[0]
+ self.assertEqual(message.subject, 'Subject')
+ self.assertEqual(message.from_email, 'from@example.com')
+ self.assertEqual(message.to, ['to@example.com'])
+
+ mail.outbox = []
+ send_mass_mail([
+ ('Subject1', 'Content1', 'from1@example.com', ['to1@example.com']),
+ ('Subject2', 'Content2', 'from2@example.com', ['to2@example.com'])
+ ], connection=connection)
+ self.assertEqual(len(mail.outbox), 2)
+ message = mail.outbox[0]
+ self.assertEqual(message.subject, 'Subject1')
+ self.assertEqual(message.from_email, 'from1@example.com')
+ self.assertEqual(message.to, ['to1@example.com'])
+ message = mail.outbox[1]
+ self.assertEqual(message.subject, 'Subject2')
+ self.assertEqual(message.from_email, 'from2@example.com')
+ self.assertEqual(message.to, ['to2@example.com'])
+
+ old_admins = settings.ADMINS
+ old_managers = settings.MANAGERS
+ settings.ADMINS = settings.MANAGERS = [('nobody','nobody@example.com')]
+
+ mail.outbox = []
+ mail_admins('Subject', 'Content', connection=connection)
+ self.assertEqual(len(mail.outbox), 1)
+ message = mail.outbox[0]
+ self.assertEqual(message.subject, '[Django] Subject')
+ self.assertEqual(message.from_email, 'root@localhost')
+ self.assertEqual(message.to, ['nobody@example.com'])
+
+ mail.outbox = []
+ mail_managers('Subject', 'Content', connection=connection)
+ self.assertEqual(len(mail.outbox), 1)
+ message = mail.outbox[0]
+ self.assertEqual(message.subject, '[Django] Subject')
+ self.assertEqual(message.from_email, 'root@localhost')
+ self.assertEqual(message.to, ['nobody@example.com'])
+
+ settings.ADMINS = old_admins
+ settings.MANAGERS = old_managers
+
+ def test_mail_prefix(self):
+ """Test prefix argument in manager/admin mail."""
+ # Regression for #13494.
+ old_admins = settings.ADMINS
+ old_managers = settings.MANAGERS
+ settings.ADMINS = settings.MANAGERS = [('nobody','nobody@example.com')]
+
+ mail_managers(ugettext_lazy('Subject'), 'Content')
+ self.assertEqual(len(mail.outbox), 1)
+ message = mail.outbox[0]
+ self.assertEqual(message.subject, '[Django] Subject')
+
+ mail.outbox = []
+ mail_admins(ugettext_lazy('Subject'), 'Content')
+ self.assertEqual(len(mail.outbox), 1)
+ message = mail.outbox[0]
+ self.assertEqual(message.subject, '[Django] Subject')
+
+ settings.ADMINS = old_admins
+ settings.MANAGERS = old_managers
+
+ def test_idn_validation(self):
+ """Test internationalized email adresses"""
+ # Regression for #14301.
+ mail.outbox = []
+ from_email = u'fröm@öäü.com'
+ to_email = u'tö@öäü.com'
+ connection = mail.get_connection('django.core.mail.backends.locmem.EmailBackend')
+ send_mail('Subject', 'Content', from_email, [to_email], connection=connection)
+ self.assertEqual(len(mail.outbox), 1)
+ message = mail.outbox[0]
+ self.assertEqual(message.subject, 'Subject')
+ self.assertEqual(message.from_email, from_email)
+ self.assertEqual(message.to, [to_email])
+ self.assertTrue(message.message().as_string().startswith('Content-Type: text/plain; charset="utf-8"\nMIME-Version: 1.0\nContent-Transfer-Encoding: quoted-printable\nSubject: Subject\nFrom: =?utf-8?b?ZnLDtm1Aw7bDpMO8LmNvbQ==?=\nTo: =?utf-8?b?dMO2QMO2w6TDvC5jb20=?='))
+
+ def test_idn_smtp_send(self):
+ import smtplib
+ smtplib.SMTP = MockSMTP
+ from_email = u'fröm@öäü.com'
+ to_email = u'tö@öäü.com'
+ connection = mail.get_connection('django.core.mail.backends.smtp.EmailBackend')
+ self.assertTrue(send_mail('Subject', 'Content', from_email, [to_email], connection=connection))
+
+class MockSMTP(object):
+ def __init__(self, host='', port=0, local_hostname=None,
+ timeout=1):
+ pass
+
+ def sendmail(self, from_addr, to_addrs, msg, mail_options=[],
+ rcpt_options=[]):
+ for addr in to_addrs:
+ str(addr.split('@', 1)[-1])
+ return {}
+
+ def quit(self):
+ return 0