diff options
| author | Patrick Maia and Victor Shyba <pixelated-team+pmaia+vshyba@thoughtworks.com> | 2014-08-27 14:25:24 +0000 |
|---|---|---|
| committer | Patrick Maia <pmaia@thoughtworks.com> | 2014-08-27 14:25:24 +0000 |
| commit | 2745e1983aad57f2f18b745b31dbf6629590b26a (patch) | |
| tree | 1bec493de418ebdc7ac7eeeeea8816a1b23736b3 | |
| parent | 2c50e4495bfbdb04a05d07227c014ce5d0f28936 (diff) | |
uses Tag.from_flags method to extract tags on PixelatedMail
| -rw-r--r-- | service/pixelated/adapter/pixelated_mail.py | 4 | ||||
| -rw-r--r-- | service/test/adapter/pixelated_mail_test.py | 6 |
2 files changed, 6 insertions, 4 deletions
diff --git a/service/pixelated/adapter/pixelated_mail.py b/service/pixelated/adapter/pixelated_mail.py index ff2c5edb..c3f43a70 100644 --- a/service/pixelated/adapter/pixelated_mail.py +++ b/service/pixelated/adapter/pixelated_mail.py @@ -48,9 +48,7 @@ class PixelatedMail: return temporary_headers def _extract_tags(self): - flags = self.leap_mail.getFlags() - tags = set(Tag.from_flag(flag) for flag in flags) - return tags + return Tag.from_flags(self.leap_mail.getFlags()) def update_tags(self, tags): self.tags = [Tag(tag) for tag in tags] diff --git a/service/test/adapter/pixelated_mail_test.py b/service/test/adapter/pixelated_mail_test.py index 841a53f2..8d1b70df 100644 --- a/service/test/adapter/pixelated_mail_test.py +++ b/service/test/adapter/pixelated_mail_test.py @@ -17,7 +17,6 @@ import unittest from pixelated.adapter.pixelated_mail import PixelatedMail from pixelated.adapter.tag import Tag -from pixelated.adapter.status import Status import test_helper @@ -43,6 +42,11 @@ class TestPixelatedMail(unittest.TestCase): pixelated_mail = PixelatedMail.from_leap_mail(test_helper.leap_mail(extra_flags=['tag_tag_work_tag_'])) self.assertIn(Tag('tag_work_tag_'), pixelated_mail.tags) + def test_non_tags_flags_are_ignored(self): + pixelated_mail = PixelatedMail.from_leap_mail(test_helper.leap_mail(leap_flags=['\\Recent'], + extra_flags=['this_is_not_a_tag', 'tag_custom_tag'])) + self.assertEquals(set([Tag('custom_tag'), Tag('inbox')]), pixelated_mail.tags) + def test_from_dict(self): mail_dict = { 'body': 'Este \xe9 o corpo', |