From 8b38782075fab1410a5f45c79c2cc252ff3d84db Mon Sep 17 00:00:00 2001 From: Folker Bernitt Date: Thu, 6 Aug 2015 09:56:12 +0200 Subject: Added move mail to indexing mail store. --- service/pixelated/adapter/mailstore/searchable_mailstore.py | 7 +++++++ .../test/unit/adapter/mailstore/test_searchable_mailstore.py | 12 ++++++++++++ 2 files changed, 19 insertions(+) (limited to 'service') diff --git a/service/pixelated/adapter/mailstore/searchable_mailstore.py b/service/pixelated/adapter/mailstore/searchable_mailstore.py index efb611fb..7339bd18 100644 --- a/service/pixelated/adapter/mailstore/searchable_mailstore.py +++ b/service/pixelated/adapter/mailstore/searchable_mailstore.py @@ -47,6 +47,13 @@ class SearchableMailStore(object): # implementes MailStore yield self._delegate.update_mail(mail) self._search_engine.index_mail(mail) + @defer.inlineCallbacks + def move_mail_to_mailbox(self, mail_id, mailbox_name): + moved_mail = yield self._delegate.move_mail_to_mailbox(mail_id, mailbox_name) + self._search_engine.remove_from_index(mail_id) + self._search_engine.index_mail(moved_mail) + defer.returnValue(moved_mail) + def __getattr__(self, name): """ Acts like method missing. If a method of MailStore is not implemented in this class, diff --git a/service/test/unit/adapter/mailstore/test_searchable_mailstore.py b/service/test/unit/adapter/mailstore/test_searchable_mailstore.py index da0755c9..00695a19 100644 --- a/service/test/unit/adapter/mailstore/test_searchable_mailstore.py +++ b/service/test/unit/adapter/mailstore/test_searchable_mailstore.py @@ -76,6 +76,18 @@ class TestSearchableMailStore(TestCase): verify(self.delegate_mail_store).update_mail(leap_mail) verify(self.search_index).index_mail(leap_mail) + @defer.inlineCallbacks + def test_move_mail_delegates_to_mail_store_and_updates_index(self): + moved_mail = LeapMail('new id', ANY_MAILBOX) + when(self.delegate_mail_store).move_mail_to_mailbox('mail id', ANY_MAILBOX).thenReturn(defer.succeed(moved_mail)) + + result = yield self.store.move_mail_to_mailbox('mail id', ANY_MAILBOX) + + verify(self.delegate_mail_store).move_mail_to_mailbox('mail id', ANY_MAILBOX) + verify(self.search_index).remove_from_index('mail id') + verify(self.search_index).index_mail(moved_mail) + self.assertEqual(moved_mail, result) + @defer.inlineCallbacks def test_other_methods_are_delegated(self): mail = LeapMail('mail id', ANY_MAILBOX) -- cgit v1.2.3