From f1b338e5564a8458c906f903ee7e0383dae86287 Mon Sep 17 00:00:00 2001 From: Giovane Date: Fri, 19 Feb 2016 19:17:30 -0200 Subject: Filter keys from attachments #608 w/ @deniscostadsc - Filtering attachments by content type application/pgp-keys --- service/test/integration/test_leap_mailstore.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'service/test/integration') diff --git a/service/test/integration/test_leap_mailstore.py b/service/test/integration/test_leap_mailstore.py index 885aa62b..f52d7c9c 100644 --- a/service/test/integration/test_leap_mailstore.py +++ b/service/test/integration/test_leap_mailstore.py @@ -49,6 +49,19 @@ class LeapMailStoreTest(SoledadTestBase): self.assertEqual(expected_mail_dict['header'], mail.as_dict()['header']) self.assertEqual(expected_mail_dict['header'], fetched_mail.as_dict()['header']) + @defer.inlineCallbacks + def test_remove_key_from_attachments(self): + input_mail = MIMEMultipart() + input_mail.attach(MIMEText(u'a utf8 message', _charset='utf-8')) + attachment = MIMEApplication('pretend to be binary attachment data') + attachment.add_header('Content-Disposition', 'attachment', filename='pub.key') + attachment.replace_header('Content-Type', 'application/pgp-keys') + input_mail.attach(attachment) + + mail = yield self.mail_store.add_mail('INBOX', input_mail.as_string()) + fetched_mail = yield self.mail_store.get_mail(mail.ident, include_body=True) + self.assertEquals(fetched_mail.as_dict().get('attachments'), []) + @defer.inlineCallbacks def test_round_trip_through_soledad_keeps_attachment(self): input_mail = MIMEMultipart() @@ -59,6 +72,7 @@ class LeapMailStoreTest(SoledadTestBase): mail = yield self.mail_store.add_mail('INBOX', input_mail.as_string()) fetched_mail = yield self.mail_store.get_mail(mail.ident, include_body=True) + self.assertDictEqual(mail.as_dict(), fetched_mail.as_dict()) @defer.inlineCallbacks def test_all_mails(self): -- cgit v1.2.3