helpers for mock mixins, added test for r extended context

Author: dkainama

tests/helpers.py

@@ -0,0 +1,13 @@
+from django.views.generic.base import ContextMixin
+from templated_mail.mail import BaseEmailMessage
+
+
+class MockMailContext(ContextMixin):
+    def get_context_data(self, **kwargs):
+        context = super(MockMailContext, self).get_context_data(**kwargs)
+        context['thing'] = 42
+        return context
+
+
+class MockMail(BaseEmailMessage, MockMailContext):
+    pass

tests/test_mail.py

@@ -9,6 +9,8 @@
 from django.test import RequestFactory, TestCase
 from templated_mail.mail import BaseEmailMessage
 
+from .helpers import MockMail
+
 
 class TestBaseEmailMessage(TestCase):
     def setUp(self):
@@ -203,3 +205,13 @@ def test_mail_from_email_is_sent_with_valid_from_email(self):
         self.assertEqual(mail.outbox[0].body, 'Foobar email content')
         self.assertEqual(mail.outbox[0].alternatives, [])
         self.assertEqual(mail.outbox[0].content_subtype, 'plain')
+
+    def test_extending_mail_with_context_mixin(self):
+        email_message = MockMail(
+            template_name='text_mail.html', context={'foo': 'bar'}
+        )
+
+        context = email_message.get_context_data()
+
+        self.assertEquals(context['foo'], 'bar')
+        self.assertEquals(context['thing'], 42)