diff options
| -rw-r--r-- | service/pixelated/adapter/pixelated_mail.py | 7 | ||||
| -rw-r--r-- | service/test/adapter/pixelated_mail_test.py | 10 |
2 files changed, 11 insertions, 6 deletions
diff --git a/service/pixelated/adapter/pixelated_mail.py b/service/pixelated/adapter/pixelated_mail.py index 3e4e415b..ff2c5edb 100644 --- a/service/pixelated/adapter/pixelated_mail.py +++ b/service/pixelated/adapter/pixelated_mail.py @@ -73,4 +73,9 @@ class PixelatedMail: @staticmethod def from_dict(mail_dict): - return PixelatedMail() + mail = PixelatedMail() + mail.headers = mail_dict['header'] + mail.body = mail_dict['body'] + mail.ident = mail_dict['ident'] + mail.tags = mail_dict['tags'] + return mail diff --git a/service/test/adapter/pixelated_mail_test.py b/service/test/adapter/pixelated_mail_test.py index c7fe9112..841a53f2 100644 --- a/service/test/adapter/pixelated_mail_test.py +++ b/service/test/adapter/pixelated_mail_test.py @@ -43,7 +43,7 @@ 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 from_dict(self): + def test_from_dict(self): mail_dict = { 'body': 'Este \xe9 o corpo', 'header': { @@ -58,10 +58,10 @@ class TestPixelatedMail(unittest.TestCase): mail = PixelatedMail.from_dict(mail_dict) - self.assertEqual(mail.headers.cc, ['cc@pixelated.com']) - self.assertEqual(mail.headers.to, ['to@pixelated.com']) - self.assertEqual(mail.headers.bcc, ['bcc@pixelated.com']) - self.assertEqual(mail.headers.subject, 'Oi') + self.assertEqual(mail.headers['cc'], ['cc@pixelated.com']) + self.assertEqual(mail.headers['to'], ['to@pixelated.com']) + self.assertEqual(mail.headers['bcc'], ['bcc@pixelated.com']) + self.assertEqual(mail.headers['subject'], 'Oi') self.assertEqual(mail.ident, '') self.assertEqual(mail.tags, ['sent']) self.assertEqual(mail.body, 'Este \xe9 o corpo') |