p15693087
/
django
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixed #27036 -- Made locmem email backend's send_messages() accept generators.

This commit is contained in:
Antoine Fontaine 2016-08-09 00:29:55 +02:00 committed by Tim Graham
parent 35225e2ade
commit 004ba05bca
2 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
django/core/mail/backends/locmem.py
View File

@ -24,6 +24,6 @@ class EmailBackend(BaseEmailBackend):
msg_count = 0 msg_count = 0
for message in messages: # .message() triggers header validation for message in messages: # .message() triggers header validation
message.message() message.message()
mail.outbox.append(message)
msg_count += 1 msg_count += 1
mail.outbox.extend(messages)
return msg_count return msg_count

10
tests/mail/tests.py
View File

@ -748,12 +748,16 @@ class BaseEmailBackendTests(HeadersCheckMixin, object):
def test_send_many(self): def test_send_many(self):
email1 = EmailMessage('Subject', 'Content1', 'from@example.com', ['to@example.com']) email1 = EmailMessage('Subject', 'Content1', 'from@example.com', ['to@example.com'])
email2 = EmailMessage('Subject', 'Content2', 'from@example.com', ['to@example.com']) email2 = EmailMessage('Subject', 'Content2', 'from@example.com', ['to@example.com'])
num_sent = mail.get_connection().send_messages([email1, email2]) # send_messages() may take a list or a generator.
emails_lists = ([email1, email2], (email for email in [email1, email2]))
for emails_list in emails_lists:
num_sent = mail.get_connection().send_messages(emails_list)
self.assertEqual(num_sent, 2) self.assertEqual(num_sent, 2)
messages = self.get_mailbox_content() messages = self.get_mailbox_content()
self.assertEqual(len(messages), 2) self.assertEqual(len(messages), 2)
self.assertEqual(messages[0].get_payload(), "Content1") self.assertEqual(messages[0].get_payload(), 'Content1')
self.assertEqual(messages[1].get_payload(), "Content2") self.assertEqual(messages[1].get_payload(), 'Content2')
self.flush_mailbox()
def test_send_verbose_name(self): def test_send_verbose_name(self):
email = EmailMessage("Subject", "Content", '"Firstname Sürname" <from@example.com>', email = EmailMessage("Subject", "Content", '"Firstname Sürname" <from@example.com>',