~ubuntu-branches/ubuntu/quantal/python-django/quantal-security

 

>>> import os

>>> import shutil

>>> import tempfile

>>> from StringIO import StringIO

>>> 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

# 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()

>>> message['From']

'from@example.com'

>>> message = email.message()

>>> message['From']

'from@example.com'

 

# 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'])

>>> 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'])

>>> email.message()['To']

'=?utf-8?q?S=C3=BCrname=2C_Firstname?= <to@example.com>, other@example.com'

 

# Regression for #6918 - When a header contains unicode,

# make sure headers can be set with a different encoding than utf-8

>>> email = EmailMessage('Message from Firstname Sürname', 'Content', 'from@example.com', ['"Sürname, Firstname" <to@example.com>','other@example.com'])

>>> email.encoding = 'iso-8859-1'

>>> email.message()['To']

'=?iso-8859-1?q?S=FCrname=2C_Firstname?= <to@example.com>, other@example.com'

>>> email.message()['Subject'].encode()

u'=?iso-8859-1?q?Message_from_Firstname_S=FCrname?='

 

# 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'

>>> msg.message()['To']

'=?iso-8859-1?q?S=FCrname=2C_Firstname?= <to@example.com>'

>>> msg.message()['Subject'].encode()

u'=?iso-8859-1?q?Message_from_Firstname_S=FCrname?='

 

# 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()

>>> message.as_string()

'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\nDate: ...\nMessage-ID: <...>\n\nFirstname 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")

>>> 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.'

>>> 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>'

 

# Make sure that the console backend writes to stdout by default

>>> connection = console.EmailBackend()

>>> email = EmailMessage('Subject', 'Content', 'bounce@example.com', ['to@example.com'], headers={'From': 'from@example.com'})

>>> connection.send_messages([email])

Content-Type: text/plain; charset="utf-8"

MIME-Version: 1.0

Content-Transfer-Encoding: quoted-printable

Subject: Subject

From: from@example.com

To: to@example.com

Date: ...

Message-ID: ...

 

Content

-------------------------------------------------------------------------------

1

 

# 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)

1

>>> print s.getvalue()

Content-Type: text/plain; charset="utf-8"

MIME-Version: 1.0

Content-Transfer-Encoding: quoted-printable

Subject: Subject

From: from@example.com

To: to@example.com

Date: ...

Message-ID: ...

 

Content

-------------------------------------------------------------------------------

 

# 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'})

>>> connection.send_messages([email, email, email])

3

 

# Make sure that 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])

2

>>> len(mail.outbox)

2

>>> mail.outbox[0].subject

'Subject'

>>> mail.outbox[1].subject

'Subject 2'

 

# Make sure that multiple locmem connections share mail.outbox

>>> mail.outbox = []

>>> connection1 = locmem.EmailBackend()

>>> connection2 = locmem.EmailBackend()

>>> email = EmailMessage('Subject', 'Content', 'bounce@example.com', ['to@example.com'], headers={'From': 'from@example.com'})

>>> connection1.send_messages([email])

1

>>> connection2.send_messages([email])

1

>>> len(mail.outbox)

2

 

# Make sure that the file backend write to the right location

>>> tmp_dir = tempfile.mkdtemp()

>>> connection = filebased.EmailBackend(file_path=tmp_dir)

>>> email = EmailMessage('Subject', 'Content', 'bounce@example.com', ['to@example.com'], headers={'From': 'from@example.com'})

>>> connection.send_messages([email])

1

>>> len(os.listdir(tmp_dir))

1

>>> print open(os.path.join(tmp_dir, os.listdir(tmp_dir)[0])).read()

Content-Type: text/plain; charset="utf-8"

MIME-Version: 1.0

Content-Transfer-Encoding: quoted-printable

Subject: Subject

From: from@example.com

To: to@example.com

Date: ...

Message-ID: ...

 

Content

-------------------------------------------------------------------------------

 

>>> connection2 = filebased.EmailBackend(file_path=tmp_dir)

>>> connection2.send_messages([email])

1

>>> len(os.listdir(tmp_dir))

2

>>> connection.send_messages([email])

1

>>> len(os.listdir(tmp_dir))

2

>>> email.connection = filebased.EmailBackend(file_path=tmp_dir)

>>> connection_created = connection.open()

>>> num_sent = email.send()

>>> len(os.listdir(tmp_dir))

3

>>> num_sent = email.send()

>>> len(os.listdir(tmp_dir))

3

>>> connection.close()

>>> shutil.rmtree(tmp_dir)

 

# Make sure that get_connection() accepts arbitrary keyword that might be

# used with custom backends.

>>> c = mail.get_connection(fail_silently=True, foo='bar')

>>> c.fail_silently

True

 

# Test custom backend defined in this suite.

>>> conn = mail.get_connection('regressiontests.mail.custombackend.EmailBackend')

>>> hasattr(conn, 'test_outbox')

True

>>> email = EmailMessage('Subject', 'Content', 'bounce@example.com', ['to@example.com'], headers={'From': 'from@example.com'})

>>> conn.send_messages([email])

1

>>> len(conn.test_outbox)

1

 

# Test backend argument of mail.get_connection()

>>> isinstance(mail.get_connection('django.core.mail.backends.smtp.EmailBackend'), smtp.EmailBackend)

True

>>> isinstance(mail.get_connection('django.core.mail.backends.locmem.EmailBackend'), locmem.EmailBackend)

True

>>> isinstance(mail.get_connection('django.core.mail.backends.dummy.EmailBackend'), dummy.EmailBackend)

True

>>> isinstance(mail.get_connection('django.core.mail.backends.console.EmailBackend'), console.EmailBackend)

True

>>> tmp_dir = tempfile.mkdtemp()

>>> isinstance(mail.get_connection('django.core.mail.backends.filebased.EmailBackend', file_path=tmp_dir), filebased.EmailBackend)

True

>>> shutil.rmtree(tmp_dir)

>>> isinstance(mail.get_connection(), locmem.EmailBackend)

True

 

# Test connection argument of send_mail() et al

>>> connection = mail.get_connection('django.core.mail.backends.console.EmailBackend')

>>> send_mail('Subject', 'Content', 'from@example.com', ['to@example.com'], connection=connection)

Content-Type: text/plain; charset="utf-8"

MIME-Version: 1.0

Content-Transfer-Encoding: quoted-printable

Subject: Subject

From: from@example.com

To: to@example.com

Date: ...

Message-ID: ...

 

Content

-------------------------------------------------------------------------------

1

 

>>> send_mass_mail([

...         ('Subject1', 'Content1', 'from1@example.com', ['to1@example.com']),

...         ('Subject2', 'Content2', 'from2@example.com', ['to2@example.com'])

...     ], connection=connection)

Content-Type: text/plain; charset="utf-8"

MIME-Version: 1.0

Content-Transfer-Encoding: quoted-printable

Subject: Subject1

From: from1@example.com

To: to1@example.com

Date: ...

Message-ID: ...

 

Content1

-------------------------------------------------------------------------------

Content-Type: text/plain; charset="utf-8"

MIME-Version: 1.0

Content-Transfer-Encoding: quoted-printable

Subject: Subject2

From: from2@example.com

To: to2@example.com

Date: ...

Message-ID: ...

 

Content2

-------------------------------------------------------------------------------

2

 

>>> mail_admins('Subject', 'Content', connection=connection)

Content-Type: text/plain; charset="utf-8"

MIME-Version: 1.0

Content-Transfer-Encoding: quoted-printable

Subject: [Django] Subject

From: root@localhost

To: nobody@example.com

Date: ...

Message-ID: ...

 

Content

-------------------------------------------------------------------------------

 

>>> mail_managers('Subject', 'Content', connection=connection)

Content-Type: text/plain; charset="utf-8"

MIME-Version: 1.0

Content-Transfer-Encoding: quoted-printable

Subject: [Django] Subject

From: root@localhost

To: nobody@example.com

Date: ...

Message-ID: ...

 

Content

-------------------------------------------------------------------------------

 

>>> settings.ADMINS = old_admins

>>> settings.MANAGERS = old_managers