federico/django-mailbox
1
0
Fork 0
forked from mirror/django-mailbox
Code Activity

Switch from nose to py.test.

Browse source
This commit is contained in:
Adam Coddington 2016-05-14 22:06:32 -07:00
parent 7bb816b9ce
commit b4a3cdc843
11 changed files with 140 additions and 156 deletions
Download patch file Download diff file Expand all files Collapse all files

18
django_mailbox/tests/test_message_flattening.py
View file

@ -1,4 +1,8 @@
from django_mailbox import models
import copy
import mock
from django_mailbox import models, utils
from django_mailbox.models import Message
from django_mailbox.tests.base import EmailMessageTestCase
@ -74,10 +78,16 @@ class TestMessageFlattening(EmailMessageTestCase):
expected_email_object = self._get_email_object(
'message_with_many_multiparts_stripped_html.eml',
)
models.STRIP_UNALLOWED_MIMETYPES = True
models.ALLOWED_MIMETYPES = ['text/plain']
default_settings = utils.get_settings()
msg = self.mailbox.process_incoming_message(incoming_email_object)
with mock.patch('django_mailbox.utils.get_settings') as get_settings:
altered = copy.deepcopy(default_settings)
altered['strip_unallowed_mimetypes'] = True
altered['allowed_mimetypes'] = ['text/plain']
get_settings.return_value = altered
msg = self.mailbox.process_incoming_message(incoming_email_object)
actual_email_object = msg.get_email_object()