django/tests/mail/custombackend.py

16 lines
465 B
Python
Raw Normal View History

"""A custom backend for testing."""
from django.core.mail.backends.base import BaseEmailBackend
class EmailBackend(BaseEmailBackend):
def __init__(self, *args, **kwargs):
super(EmailBackend, self).__init__(*args, **kwargs)
self.test_outbox = []
def send_messages(self, email_messages):
2014-05-29 00:39:14 +00:00
# Messages are stored in an instance variable for testing.
self.test_outbox.extend(email_messages)
return len(email_messages)