Update mail with new tags should be working now

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())