diff options
Diffstat (limited to 'service/test/adapter/pixelated_mail_test.py')
-rw-r--r-- | service/test/adapter/pixelated_mail_test.py | 158 |
1 files changed, 62 insertions, 96 deletions
diff --git a/service/test/adapter/pixelated_mail_test.py b/service/test/adapter/pixelated_mail_test.py index 0a103cce..c8c23b8a 100644 --- a/service/test/adapter/pixelated_mail_test.py +++ b/service/test/adapter/pixelated_mail_test.py @@ -18,81 +18,83 @@ import os import pixelated.support.date import test_helper -from pixelated.adapter.pixelated_mail import PixelatedMail +from pixelated.adapter.pixelated_mail import PixelatedMail, InputMail from pixelated.adapter.tag_service import TagService from pixelated.adapter.tag_index import TagIndex from pixelated.adapter.tag import Tag -from mock import Mock - +from mockito import * class TestPixelatedMail(unittest.TestCase): - mail_dict = lambda x: { - 'body': 'Este \xe9 o corpo', - 'header': { - 'cc': ['cc@pixelated.org', 'anothercc@pixelated.org'], - 'to': ['to@pixelated.org', 'anotherto@pixelated.org'], - 'bcc': ['bcc@pixelated.org', 'anotherbcc@pixelated.org'], - 'subject': 'Oi' - }, - 'ident': '', - 'tags': ['sent'] - } - def test_parse_date_from_leap_mail_uses_date_header_if_available(self): + def setUp(self): + self.querier = mock() + + def test_parse_date_from_soledad_uses_date_header_if_available(self): leap_mail_date = 'Wed, 3 Sep 2014 12:36:17 -0300' leap_mail_date_in_iso_format = "2014-09-03T12:36:17-03:00" leap_mail = test_helper.leap_mail(headers={'date': leap_mail_date}) - mail = PixelatedMail.from_leap_mail(leap_mail) + mail = PixelatedMail.from_soledad(*leap_mail, soledad_querier=self.querier) - self.assertEqual(str(mail.headers['date']), leap_mail_date_in_iso_format) + self.assertEqual(str(mail.headers['Date']), leap_mail_date_in_iso_format) - def test_parse_date_from_leap_mail_fallback_to_received_header_if_date_header_isnt_available(self): + def test_parse_date_from_soledad_fallback_to_received_header_if_date_header_isnt_available(self): leap_mail_date = "Wed, 03 Sep 2014 13:11:15 -0300" leap_mail_date_in_iso_format = "2014-09-03T13:11:15-03:00" leap_mail_received_header = "by bitmask.local from 127.0.0.1 with ESMTP ;\n " + leap_mail_date leap_mail = test_helper.leap_mail(headers={'received': leap_mail_received_header}) - mail = PixelatedMail.from_leap_mail(leap_mail) + mail = PixelatedMail.from_soledad(*leap_mail, soledad_querier=self.querier) - self.assertEqual(str(mail.headers['date']), leap_mail_date_in_iso_format) + self.assertEqual(str(mail.headers['Date']), leap_mail_date_in_iso_format) - def test_from_dict(self): - mail = PixelatedMail.from_dict(self.mail_dict()) + def test_update_tags_return_a_set_with_the_current_tags(self): + soledad_docs = test_helper.leap_mail(extra_headers={'X-tags': '["custom_1", "custom_2"]'}) + pixelated_mail = PixelatedMail.from_soledad(*soledad_docs, soledad_querier=self.querier) - self.assertEqual(mail.headers['cc'], ['cc@pixelated.org', 'anothercc@pixelated.org']) - self.assertEqual(mail.headers['to'], ['to@pixelated.org', 'anotherto@pixelated.org']) - self.assertEqual(mail.headers['bcc'], ['bcc@pixelated.org', 'anotherbcc@pixelated.org']) - self.assertEqual(mail.headers['subject'], 'Oi') - self.assertEqual(mail.ident, None) - self.assertEqual(mail.tags, set(['sent'])) - self.assertEqual(mail.body, 'Este \xe9 o corpo') + current_tags = pixelated_mail.update_tags({'custom_1', 'custom_3'}) + self.assertEquals({'custom_3', 'custom_1'}, current_tags) - def test_from_dict_adds_current_date(self): - pixelated.support.date.iso_now = lambda: 'date now' + def test_mark_as_read(self): + mail = PixelatedMail.from_soledad(*test_helper.leap_mail(flags=[]), soledad_querier=self.querier) - mail = PixelatedMail.from_dict(self.mail_dict()) + mail.mark_as_read() - self.assertEqual('date now', mail.headers['date']) + self.assertEquals(mail.fdoc.content['flags'], ['\\Seen']) - def test_update_tags_return_a_set_with_the_current_tags(self): - pixelated_mail = PixelatedMail.from_leap_mail(test_helper.leap_mail(extra_headers={'X-tags': '["custom_1", "custom_2"]'}), Mock()) - current_tags = pixelated_mail.update_tags(set(['custom_1', 'custom_3'])) - self.assertEquals(set(['custom_3', 'custom_1']), current_tags) + def test_mark_as_not_recent(self): + mail = PixelatedMail.from_soledad(*test_helper.leap_mail(flags=['\\Recent']), soledad_querier=self.querier) - def test_to_mime_multipart(self): - pixelated.support.date.iso_now = lambda: 'date now' + mail.mark_as_not_recent() - mime_multipart = PixelatedMail.from_dict(self.mail_dict()).to_mime_multipart() + self.assertEquals(mail.fdoc.content['flags'], []) + + def test_update_tags_notifies_tag_service(self): + db_path = '/tmp/test_update_tags_notifies_tag_service' + tag_service = TagService(TagIndex(db_path)) + + mail = PixelatedMail.from_soledad(*test_helper.leap_mail(), soledad_querier=self.querier) + + mail.update_tags({'new_tag'}) + self.assertIn(Tag('new_tag'), tag_service.all_tags()) + + os.remove(db_path + '.db') + +class InputMailTest(unittest.TestCase): + mail_dict = lambda x: { + 'body': 'Este \xe9 o corpo', + 'header': { + 'cc': ['cc@pixelated.org', 'anothercc@pixelated.org'], + 'to': ['to@pixelated.org', 'anotherto@pixelated.org'], + 'bcc': ['bcc@pixelated.org', 'anotherbcc@pixelated.org'], + 'subject': 'Oi' + }, + 'ident': '', + 'tags': ['sent'] + } - self.assertRegexpMatches(mime_multipart.as_string(), "\nTo: to@pixelated.org, anotherto@pixelated.org\n") - self.assertRegexpMatches(mime_multipart.as_string(), "\nCc: cc@pixelated.org, anothercc@pixelated.org\n") - self.assertRegexpMatches(mime_multipart.as_string(), "\nBcc: bcc@pixelated.org, anotherbcc@pixelated.org\n") - self.assertRegexpMatches(mime_multipart.as_string(), "\nDate: date now\n") - self.assertRegexpMatches(mime_multipart.as_string(), "\nSubject: Oi\n") - self.assertRegexpMatches(mime_multipart.as_string(), "\nEste \xe9 o corpo") def test_to_mime_multipart_should_add_blank_fields(self): pixelated.support.date.iso_now = lambda: 'date now' @@ -103,66 +105,30 @@ class TestPixelatedMail(unittest.TestCase): mail_dict['header']['cc'] = '' mail_dict['header']['subject'] = '' - mime_multipart = PixelatedMail.from_dict(mail_dict).to_mime_multipart() + mime_multipart = InputMail.from_dict(mail_dict).to_mime_multipart() self.assertNotRegexpMatches(mime_multipart.as_string(), "\nTo: \n") self.assertNotRegexpMatches(mime_multipart.as_string(), "\nBcc: \n") self.assertNotRegexpMatches(mime_multipart.as_string(), "\nCc: \n") self.assertNotRegexpMatches(mime_multipart.as_string(), "\nSubject: \n") - def test_smtp_format(self): - PixelatedMail.from_email_address = 'pixelated@org' - mail = PixelatedMail.from_dict(self.mail_dict()) - - smtp_format = mail.to_smtp_format() - - self.assertRegexpMatches(smtp_format, "\nFrom: pixelated@org") - - def test_extract_headers_should_break_header_in_multiple_recipients(self): - headers = test_helper.DEFAULT_HEADERS.copy() - - headers['to'] = "nlima@example.com, Duda Dornelles <ddornelles@example.com>" - headers['bcc'] = "ddornelles@example.com, Neissi Lima <nlima@example.com>" - headers['cc'] = "nlima@example.com, Duda Dornelles <ddornelles@example.com>" - - leap_mail = test_helper.leap_mail(headers=headers) - - pixelated_mail = PixelatedMail.from_leap_mail(leap_mail) - - self.assertEquals(pixelated_mail.headers['to'], ["nlima@example.com", "Duda Dornelles <ddornelles@example.com>"]) - self.assertEquals(pixelated_mail.headers['bcc'], ["ddornelles@example.com", "Neissi Lima <nlima@example.com>"]) - self.assertEquals(pixelated_mail.headers['cc'], ["nlima@example.com", "Duda Dornelles <ddornelles@example.com>"]) - - def test_mark_as_read(self): - mail = PixelatedMail.from_leap_mail(test_helper.leap_mail(flags=[])) - - mail.mark_as_read() - - self.assertEquals(mail.leap_mail.setFlags.call_args[0], (('\\Seen',), 1)) - def test_mark_as_not_recent(self): - mail = PixelatedMail.from_leap_mail(test_helper.leap_mail(flags=[])) - - mail.mark_as_not_recent() - - self.assertEquals(mail.leap_mail.setFlags.call_args[0], (('\\Recent',), -1)) - - def test_mark_as_deleted(self): - mail = PixelatedMail.from_leap_mail(test_helper.leap_mail(extra_headers={'X-Tags': '["skinka", "altoids"]'}, flags=[]), Mock()) + def test_to_mime_multipart(self): + pixelated.support.date.iso_now = lambda: 'date now' - self.assertEquals(set(['skinka', 'altoids']), mail.tags) + mime_multipart = InputMail.from_dict(self.mail_dict()).to_mime_multipart() - mail.mark_as_deleted() + self.assertRegexpMatches(mime_multipart.as_string(), "\nTo: to@pixelated.org, anotherto@pixelated.org\n") + self.assertRegexpMatches(mime_multipart.as_string(), "\nCc: cc@pixelated.org, anothercc@pixelated.org\n") + self.assertRegexpMatches(mime_multipart.as_string(), "\nBcc: bcc@pixelated.org, anotherbcc@pixelated.org\n") + self.assertRegexpMatches(mime_multipart.as_string(), "\nDate: date now\n") + self.assertRegexpMatches(mime_multipart.as_string(), "\nSubject: Oi\n") + self.assertRegexpMatches(mime_multipart.as_string(), "\nEste \xe9 o corpo") - self.assertEquals(set([]), mail.tags) - self.assertEquals(mail.leap_mail.setFlags.call_args[0], ((u'\\Deleted',), 1)) - def test_update_tags_notifies_tag_service(self): - db_path = '/tmp/test_update_tags_notifies_tag_service' - tag_service = TagService(TagIndex(db_path)) - mail = PixelatedMail.from_leap_mail(test_helper.leap_mail(), tag_service) + def test_smtp_format(self): + PixelatedMail.from_email_address = 'pixelated@org' - mail.update_tags(set(['new_tag'])) - self.assertIn(Tag('new_tag'), tag_service.all_tags()) + smtp_format = InputMail.from_dict(self.mail_dict()).to_smtp_format() - os.remove(db_path + '.db') + self.assertRegexpMatches(smtp_format, "\nFrom: pixelated@org")
\ No newline at end of file |