Added load-mails maintenance command.

- Issue #335

3 files changed, 53 insertions, 2 deletions

diff --git a/service/test/unit/fixtures/mailset/mbox00000000 b/service/test/unit/fixtures/mailset/mbox00000000
new file mode 100644
index 00000000..3d01c203
--- /dev/null
+++ b/service/test/unit/fixtures/mailset/mbox00000000
@@ -0,0 +1,12 @@
+From darby.senger@zemlak.biz
+Subject: Itaque consequatur repellendus provident sunt quia.
+To: carmel@murazikortiz.name
+X-TW-Pixelated-Tags: nite, macro, trash
+Date: Tue, 21 Apr 2015 08:43:27 +0000 (UTC)
+
+Dignissimos ducimus veritatis. Est tenetur consequatur quia occaecati. Vel sit sit voluptas.
+
+Earum distinctio eos. Accusantium qui sint ut quia assumenda. Facere dignissimos inventore autem sit amet. Pariatur voluptatem sint est.
+
+Ut recusandae praesentium aspernatur. Exercitationem amet placeat deserunt quae consequatur eum. Unde doloremque suscipit quia.
+
diff --git a/service/test/unit/fixtures/mailset/mbox00000001 b/service/test/unit/fixtures/mailset/mbox00000001
new file mode 100644
index 00000000..fc76bba2
--- /dev/null
+++ b/service/test/unit/fixtures/mailset/mbox00000001
@@ -0,0 +1,12 @@
+From madeline.littel@sanfordruel.com
+Subject: Error illum dignissimos autem eos aspernatur.
+To: phyllis@stiedemann.net
+X-TW-Pixelated-Tags: instadaily, inspiration
+Date: Tue, 21 Apr 2015 08:43:27 +0000 (UTC)
+
+Et inventore placeat aut. Sint eveniet labore perferendis nulla. Maiores rerum sunt perferendis. Voluptate iure hic et ut blanditiis ad veritatis. Labore occaecati rerum.
+
+Sit fugiat aliquam voluptates ipsum non. Dolor quo sapiente. Itaque sed odit velit. Qui et aspernatur et fugiat voluptas eum est. Et expedita eos rerum nisi ut eum vero.
+
+Ab et est cumque. Qui nostrum perferendis. Labore est tempora porro est quia deleniti consequatur. Fugit quis ipsa.
+
diff --git a/service/test/unit/maintenance/test_commands.py b/service/test/unit/maintenance/test_commands.py
index a20b0383..f23655d8 100644
--- a/service/test/unit/maintenance/test_commands.py
+++ b/service/test/unit/maintenance/test_commands.py
@@ -14,12 +14,15 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with Pixelated. If not, see <http://www.gnu.org/licenses/>.
 import unittest
+import email
 
-from pixelated.maintenance import delete_all_mails
+from pixelated.maintenance import delete_all_mails, load_mails
 from pixelated.bitmask_libraries.session import LeapSession
+from leap.mail.imap.account import SoledadBackedAccount
 from leap.soledad.client import Soledad
 from leap.soledad.common.document import SoledadDocument
-from mock import MagicMock
+from mock import MagicMock, ANY
+from os.path import join, dirname
 
 
 class TestCommands(unittest.TestCase):
@@ -27,6 +30,11 @@ class TestCommands(unittest.TestCase):
         def setUp(self):
             self.leap_session = MagicMock(spec=LeapSession)
             self.soledad = MagicMock(spec=Soledad)
+            self.account = MagicMock(spec=SoledadBackedAccount)
+            self.mailbox = MagicMock()
+            self.leap_session.account = self.account
+            self.account.getMailbox.return_value = self.mailbox
+
             self.args = (self.leap_session, self.soledad)
 
         def test_delete_all_mails_supports_empty_doclist(self):
@@ -63,3 +71,22 @@ class TestCommands(unittest.TestCase):
             doc = MagicMock(spec=SoledadDocument)
             doc.content = {'type': doc_type}
             return doc
+
+        def test_load_mails_empty_path_list(self):
+            load_mails(self.args, [])
+
+            self.assertFalse(self.mailbox.called)
+
+        def test_load_mails_adds_mails(self):
+            mail_root = join(dirname(__file__), '..', 'fixtures', 'mailset')
+
+            foo = load_mails(self.args, [mail_root])
+
+            self.assertTrue(self.mailbox.addMessage.called)
+            self.mailbox.addMessage.assert_any_call(self._mail_content(join(mail_root, 'mbox00000000')), flags=("\\RECENT",), notify_on_disk=False)
+            self.mailbox.addMessage.assert_any_call(self._mail_content(join(mail_root, 'mbox00000001')), flags=("\\RECENT",), notify_on_disk=False)
+
+        def _mail_content(self, mail_file):
+            with open(mail_file, 'r') as fp:
+                m = email.message_from_file(fp)
+                return m.as_string()