diff options
| author | Patrick Maia <pmaia@thoughtworks.com> | 2014-08-28 15:06:21 +0000 |
|---|---|---|
| committer | Patrick Maia <pmaia@thoughtworks.com> | 2014-08-28 15:06:21 +0000 |
| commit | dc3f7d1ae9a56192676ae4efb8286b0968ec9726 (patch) | |
| tree | c33b7801f57716a30a5f2829749b4cff0895f78f /service/test | |
| parent | de9c26359ed448d4ee4937fa0d241469a7d76cf1 (diff) | |
#53 - adds ability to remove tags from email
Diffstat (limited to 'service/test')
| -rw-r--r-- | service/test/adapter/mail_service_test.py | 8 | ||||
| -rw-r--r-- | service/test/adapter/pixelated_mail_test.py | 6 |
2 files changed, 11 insertions, 3 deletions
diff --git a/service/test/adapter/mail_service_test.py b/service/test/adapter/mail_service_test.py index 6972f6f6..7b3f4e96 100644 --- a/service/test/adapter/mail_service_test.py +++ b/service/test/adapter/mail_service_test.py @@ -24,8 +24,9 @@ from pixelated.adapter.pixelated_mailbox import PixelatedMailbox class TestMailService(unittest.TestCase): - @patch.object(MailService, '_set_mail_flags', return_value=None) - def test_custom_tags_get_created_if_not_exists(self, mockSetFlags): + @patch.object(MailService, '_append_mail_flags', return_value=None) + @patch.object(MailService, '_remove_mail_flags', return_value=None) + def test_custom_tags_get_created_if_not_exists(self, mockRemoveFlags, mockAppendFlags): MailService._open_leap_session = lambda self: None MailService.mailbox = PixelatedMailbox(test_helper.leap_mailbox(leap_flags=['\\Recent'])) MailService.account = Mock(return_value=MagicMock()) @@ -37,4 +38,5 @@ class TestMailService(unittest.TestCase): 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']) + mockAppendFlags.assert_called_with(6, ['tag_test']) + mockRemoveFlags.assert_called_with(6, []) diff --git a/service/test/adapter/pixelated_mail_test.py b/service/test/adapter/pixelated_mail_test.py index 8d1b70df..81e9f340 100644 --- a/service/test/adapter/pixelated_mail_test.py +++ b/service/test/adapter/pixelated_mail_test.py @@ -69,3 +69,9 @@ class TestPixelatedMail(unittest.TestCase): self.assertEqual(mail.ident, '') self.assertEqual(mail.tags, ['sent']) self.assertEqual(mail.body, 'Este \xe9 o corpo') + + def test_update_tags_return_a_set_for_current_tags_and_a_set_for_removed(self): + pixelated_mail = PixelatedMail.from_leap_mail(test_helper.leap_mail(leap_flags=[], extra_flags=['tag_custom_1', 'tag_custom_2'])) + current_tags, removed_tags = pixelated_mail.update_tags(set([Tag('custom_1'), Tag('custom_3')])) + self.assertEquals(set([Tag('custom_3'), Tag('custom_1')]), current_tags) + self.assertEquals(set([Tag('custom_2')]), removed_tags) |