From d2e5af008d012c24716c6515490ea4639b8dd9a1 Mon Sep 17 00:00:00 2001 From: Folker Bernitt Date: Wed, 5 Aug 2015 14:06:40 +0200 Subject: Started work on searchable mail store. --- .../adapter/mailstore/searchable_mailstore.py | 27 ++++++++++++++ .../adapter/mailstore/test_searchable_mailstore.py | 42 ++++++++++++++++++++++ 2 files changed, 69 insertions(+) create mode 100644 service/pixelated/adapter/mailstore/searchable_mailstore.py create mode 100644 service/test/unit/adapter/mailstore/test_searchable_mailstore.py diff --git a/service/pixelated/adapter/mailstore/searchable_mailstore.py b/service/pixelated/adapter/mailstore/searchable_mailstore.py new file mode 100644 index 00000000..ad5fca33 --- /dev/null +++ b/service/pixelated/adapter/mailstore/searchable_mailstore.py @@ -0,0 +1,27 @@ +# +# Copyright (c) 2015 ThoughtWorks, Inc. +# +# Pixelated is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# Pixelated is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Affero General Public License for more details. +# +# You should have received a copy of the GNU Affero General Public License +# along with Pixelated. If not, see . +from pixelated.adapter.mailstore.mailstore import MailStore + + +class SearchableMailStore(MailStore): + + def __init__(self, delegate, search_engine): + self._delegate = delegate + self._search_engine = search_engine + + def add_mail(self, mailbox_name, mail): + self._delegate.add_mail(mailbox_name, mail) + self._search_engine.index_mail(mail) diff --git a/service/test/unit/adapter/mailstore/test_searchable_mailstore.py b/service/test/unit/adapter/mailstore/test_searchable_mailstore.py new file mode 100644 index 00000000..9ba61e32 --- /dev/null +++ b/service/test/unit/adapter/mailstore/test_searchable_mailstore.py @@ -0,0 +1,42 @@ +# +# Copyright (c) 2015 ThoughtWorks, Inc. +# +# Pixelated is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# Pixelated is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Affero General Public License for more details. +# +# You should have received a copy of the GNU Affero General Public License +# along with Pixelated. If not, see . +from email.parser import Parser +import os +from mockito import verify, mock +import pkg_resources +from twisted.trial.unittest import TestCase +from pixelated.adapter.mailstore.searchable_mailstore import SearchableMailStore +from pixelated.adapter.search import SearchEngine + + +class TestLeapMail(TestCase): + def test_add_mail_delegates_to_mail_store_and_updates_index(self): + mail = self._load_mail_from_file('mbox00000000') + search_index = mock(mocked_obj=SearchEngine) + delegate_mail_store = mock() + store = SearchableMailStore(delegate_mail_store, search_index) + + store.add_mail('INBOX', mail) + + verify(search_index).index_mail(mail) + verify(delegate_mail_store).add_mail('INBOX', mail) + + def _load_mail_from_file(self, mail_file): + mailset_dir = pkg_resources.resource_filename('test.unit.fixtures', 'mailset') + mail_file = os.path.join(mailset_dir, 'new', mail_file) + with open(mail_file) as f: + mail = Parser().parse(f) + return mail -- cgit v1.2.3