diff options
Diffstat (limited to 'service/test/adapter/mail_service_test.py')
-rw-r--r-- | service/test/adapter/mail_service_test.py | 33 |
1 files changed, 7 insertions, 26 deletions
diff --git a/service/test/adapter/mail_service_test.py b/service/test/adapter/mail_service_test.py index 0c6538dc..1219307e 100644 --- a/service/test/adapter/mail_service_test.py +++ b/service/test/adapter/mail_service_test.py @@ -5,38 +5,19 @@ from mock import Mock, MagicMock, patch import test_helper from pixelated.tags import Tag - class TestMailService(unittest.TestCase): - 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" - } - @patch.object(MailService, 'set_flags', return_value=None) - def test_custom_tags_get_created_if_not_exists(self, setFlags): + def test_custom_tags_get_created_if_not_exists(self, mockSetFlags): MailService._open_leap_session = lambda self: None - MailService.mailbox = Mock(messages=[test_helper.leap_mail(uid=6)]) + MailService.mailbox = Mock(messages=[test_helper.leap_mail(uid=6, leap_flags=['\\Recent'])]) MailService.account = Mock(return_value=MagicMock()) mailservice = MailService() - new_tags = ['test'] - mails = mailservice.update_tags(6, new_tags) - - for tag in mailservice.all_tags(): - print tag.name + new_tags = ['test', 'inbox'] + updated_tags = mailservice.update_tags(6, new_tags) - self.assertIn(Tag('test'), mailservice.all_tags()) + self.assertEquals(set([Tag('test'), Tag('inbox')]), set(updated_tags)) + # make sure that special tags are skipped when setting leap flags (eg.: tag_inbox) + mockSetFlags.assert_called_with(6, ['tag_test']) |