added drafts.

4 files changed, 30 insertions, 1 deletions

diff --git a/service/test/adapter/mail_service_test.py b/service/test/adapter/mail_service_test.py
index e96b7e11..96359d30 100644
--- a/service/test/adapter/mail_service_test.py
+++ b/service/test/adapter/mail_service_test.py
@@ -38,3 +38,11 @@ class TestMailService(unittest.TestCase):
         self.mail_service.mark_as_read(1)
 
         verify(mail).mark_as_read()
+
+    def test_create_draft(self):
+        drafts_mailbox = mock()
+        when(self.mailboxes).drafts().thenReturn(drafts_mailbox)
+
+        self.mail_service.create_draft('a new draft')
+
+        verify(drafts_mailbox).add('a new draft')
diff --git a/service/test/adapter/pixelated_mail_test.py b/service/test/adapter/pixelated_mail_test.py
index 9d7176a4..922c147d 100644
--- a/service/test/adapter/pixelated_mail_test.py
+++ b/service/test/adapter/pixelated_mail_test.py
@@ -91,9 +91,10 @@ class TestPixelatedMail(unittest.TestCase):
         self.assertRegexpMatches(mime_multipart.as_string(), "\nEste \xe9 o corpo")
 
     def test_smtp_format(self):
+        PixelatedMail.from_email_address = 'pixelated@org'
         mail = PixelatedMail.from_dict(self.mail_dict)
 
-        smtp_format = mail.to_smtp_format(_from='pixelated@org')
+        smtp_format = mail.to_smtp_format()
 
         self.assertRegexpMatches(smtp_format, "\nFrom: pixelated@org")
 
diff --git a/service/test/adapter/pixelated_mailbox_test.py b/service/test/adapter/pixelated_mailbox_test.py
index 6574d407..ae63b157 100644
--- a/service/test/adapter/pixelated_mailbox_test.py
+++ b/service/test/adapter/pixelated_mailbox_test.py
@@ -15,8 +15,12 @@
 # along with Pixelated. If not, see <http://www.gnu.org/licenses/>.
 import unittest
 from mockito import *
+import leap
+import os
+from pixelated.adapter.pixelated_mail import PixelatedMail
 import test_helper
 from pixelated.adapter.pixelated_mailbox import PixelatedMailbox
+from mockito import *
 
 
 class TestPixelatedMailbox(unittest.TestCase):
@@ -31,3 +35,18 @@ class TestPixelatedMailbox(unittest.TestCase):
         mails = self.mailbox.mails()
 
         self.assertIn('sent', mails[0].tags)
+
+    def test_add_message_to_mailbox(self):
+        PixelatedMail.from_email_address = 'pixel@ted.org'
+        mail = PixelatedMail.from_dict(test_helper.mail_dict())
+        mail.to_smtp_format = lambda: 'the mail in smtp format'
+
+        leap_mailbox_messages = mock()
+        leap_mailbox = test_helper.leap_mailbox(messages=[])
+        mailbox = PixelatedMailbox(leap_mailbox, self.db_file_path)
+        mailbox.leap_mailbox.messages = leap_mailbox_messages
+
+        mailbox.add(mail)
+
+        verify(leap_mailbox_messages).add_msg('the mail in smtp format')
+        
diff --git a/service/test/adapter/pixelated_mailboxes_test.py b/service/test/adapter/pixelated_mailboxes_test.py
index 1ced8369..eebdc904 100644
--- a/service/test/adapter/pixelated_mailboxes_test.py
+++ b/service/test/adapter/pixelated_mailboxes_test.py
@@ -40,3 +40,4 @@ class PixelatedMailboxesTest(unittest.TestCase):
         # then
         self.assertEqual(1, len(mails))
         self.assertEqual("mail", mails[0])
+