diff options
Diffstat (limited to 'service/test/adapter')
-rw-r--r-- | service/test/adapter/mail_service_test.py | 42 |
1 files changed, 32 insertions, 10 deletions
diff --git a/service/test/adapter/mail_service_test.py b/service/test/adapter/mail_service_test.py index 0828f20a..4751413d 100644 --- a/service/test/adapter/mail_service_test.py +++ b/service/test/adapter/mail_service_test.py @@ -1,17 +1,39 @@ import unittest from app.adapter.mail_service import MailService - - -class MailboxCollision(Exception): - pass - - -class MailboxException(Exception): - pass +from mock import Mock, MagicMock, patch +import test_helper +from app.tags import Tag class TestMailService(unittest.TestCase): - def setUp(self): - self.mail_service = MailService() + def _raw_mail(self): + return { + "header": + { + "to":["p@k.s"], + "from":"a@y.t", + "subject":"Test", + "date":"2007-09-28T06:11:03-03:00" + }, + "ident":6, + "tags":["instagramer"], + "status":[], + "security_casing":{}, + "draft_reply_for":[], + "body":"teste" + } + + def test_custom_tags_get_created_if_not_exists(self): + MailService._open_leap_session = lambda self: None + MailService.mailbox = Mock(messages=[test_helper.leap_mail(uid=6)]) + MailService.account = Mock(return_value=MagicMock()) + + mailservice = MailService() + + raw_mail = self._raw_mail() + raw_mail['tags'].append('my_new_tag') + mails = mailservice.update_mail(raw_mail) + + self.assertIn(Tag('my_new_tag'), mailservice.all_tags()) |