diff options
Diffstat (limited to 'service/test/unit/adapter/mail_test.py')
| -rw-r--r-- | service/test/unit/adapter/mail_test.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/service/test/unit/adapter/mail_test.py b/service/test/unit/adapter/mail_test.py index e8fd8755..e0b7a498 100644 --- a/service/test/unit/adapter/mail_test.py +++ b/service/test/unit/adapter/mail_test.py @@ -21,6 +21,8 @@ from mockito import * from test.support import test_helper import dateutil.parser as dateparser import base64 +from leap.mail.imap.fields import fields +from datetime import datetime class TestPixelatedMail(unittest.TestCase): @@ -69,6 +71,17 @@ class TestPixelatedMail(unittest.TestCase): self.assertEquals(mail.fdoc.content['flags'], []) + def test_get_for_save_adds_from(self): + InputMail.FROM_EMAIL_ADDRESS = 'me@pixelated.org' + headers = {'Subject': 'The subject', + 'Date': str(datetime.now()), + 'To': 'me@pixelated.org'} + + input_mail = InputMail() + input_mail.headers = headers + + self.assertEqual('me@pixelated.org', input_mail.get_for_save(1, 'SENT')[1][fields.HEADERS_KEY]['From']) + def test_as_dict(self): headers = {'Subject': 'The subject', 'From': 'someone@pixelated.org', |