Made LeapMail compatible with SearchEngine.

1 files changed, 9 insertions, 9 deletions

diff --git a/service/test/integration/test_leap_mailstore.py b/service/test/integration/test_leap_mailstore.py
index 614d75cc..f111d2b1 100644
--- a/service/test/integration/test_leap_mailstore.py
+++ b/service/test/integration/test_leap_mailstore.py
@@ -29,7 +29,7 @@ class LeapMailStoreTest(SoledadTestBase):
         mail_id = yield self._create_mail_in_soledad(mail)
         expected_mail_dict = {'body': u'Dignissimos ducimus veritatis. Est tenetur consequatur quia occaecati. Vel sit sit voluptas.\n\nEarum distinctio eos. Accusantium qui sint ut quia assumenda. Facere dignissimos inventore autem sit amet. Pariatur voluptatem sint est.\n\nUt recusandae praesentium aspernatur. Exercitationem amet placeat deserunt quae consequatur eum. Unde doloremque suscipit quia.\n\n', 'header': {u'date': u'Tue, 21 Apr 2015 08:43:27 +0000 (UTC)', u'to': u'carmel@murazikortiz.name', u'x-tw-pixelated-tags': u'nite, macro, trash', u'from': u'darby.senger@zemlak.biz', u'subject': u'Itaque consequatur repellendus provident sunt quia.'}, 'ident': mail_id, 'tags': set([])}
 
-        result = yield self.store.get_mail(mail_id, include_body=True)
+        result = yield self.mail_store.get_mail(mail_id, include_body=True)
         self.assertIsNotNone(result)
         self.assertEqual(expected_mail_dict, result.as_dict())
 
@@ -38,7 +38,7 @@ class LeapMailStoreTest(SoledadTestBase):
         mail = load_mail_from_file('mbox00000000')
         yield self._create_mail_in_soledad(mail)
 
-        mails = yield self.store.all_mails()
+        mails = yield self.mail_store.all_mails()
 
         self.assertEqual(1, len(mails))
         self.assertEqual('Itaque consequatur repellendus provident sunt quia.', mails[0].subject)
@@ -47,23 +47,23 @@ class LeapMailStoreTest(SoledadTestBase):
     def test_add_and_remove_mail(self):
         yield self.adaptor.initialize_store(self.soledad)
         mail = load_mail_from_file('mbox00000000')
-        yield self.store.add_mailbox('INBOX')
+        yield self.mail_store.add_mailbox('INBOX')
 
-        msg = yield self.store.add_mail('INBOX', mail.as_string())
+        msg = yield self.mail_store.add_mail('INBOX', mail.as_string())
 
-        yield self.store.delete_mail(msg.mail_id)
+        yield self.mail_store.delete_mail(msg.mail_id)
 
-        deleted_msg = yield self.store.get_mail(msg.mail_id)
+        deleted_msg = yield self.mail_store.get_mail(msg.mail_id)
 
         self.assertIsNone(deleted_msg)
 
     @defer.inlineCallbacks
     def test_get_mailbox_mail_ids(self):
         mail = load_mail_from_file('mbox00000000')
-        yield self.store.add_mailbox('INBOX')
-        mail = yield self.store.add_mail('INBOX', mail.as_string())
+        yield self.mail_store.add_mailbox('INBOX')
+        mail = yield self.mail_store.add_mail('INBOX', mail.as_string())
 
-        mails = yield self.store.get_mailbox_mail_ids('INBOX')
+        mails = yield self.mail_store.get_mailbox_mail_ids('INBOX')
 
         self.assertEqual(1, len(mails))
         self.assertEqual(mail.mail_id, mails[0])