1 files changed, 89 insertions, 0 deletions
diff --git a/service/test/unit/adapter/search/test_search.py b/service/test/unit/adapter/search/test_search.py
new file mode 100644
index 00000000..d57b8227
--- /dev/null
+++ b/service/test/unit/adapter/search/test_search.py
@@ -0,0 +1,89 @@
+#
+# Copyright (c) 2014 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 <http://www.gnu.org/licenses/>.
+
+
+import unittest
+from mockito import mock, when
+from pixelated.adapter.search import SearchEngine
+from tempdir import TempDir
+from test.support import test_helper
+
+INDEX_KEY = '\xde3?\x87\xff\xd9\xd3\x14\xf0\xa7>\x1f%C{\x16.\\\xae\x8c\x13\xa7\xfb\x04\xd4]+\x8d_\xed\xd1\x8d\x0bI' \
+    '\x8a\x0e\xa4tm\xab\xbf\xb4\xa5\x99\x00d\xd5w\x9f\x18\xbc\x1d\xd4_W\xd2\xb6\xe8H\x83\x1b\xd8\x9d\xad'
+
+
+class LockStub(object):
+    def __init__(self):
+        self.called = False
+
+    def __enter__(self):
+        self.called = True
+        return self
+
+    def __exit__(self, type, value, traceback):
+        return False
+
+
+class SearchEngineTest(unittest.TestCase):
+    def setUp(self):
+        self.tempdir = TempDir()
+        self.agent_home = self.tempdir.name
+
+    def tearDown(self):
+        self.tempdir.dissolve()
+
+    def test_index_mail_secured_by_lock(self):
+        # given
+        soledad_querier = mock()
+        lock_stub = LockStub()
+        when(soledad_querier).get_index_masterkey().thenReturn(INDEX_KEY)
+
+        self.assertEqual(INDEX_KEY, soledad_querier.get_index_masterkey())
+        se = SearchEngine(soledad_querier, self.agent_home)
+        se._write_lock = lock_stub
+
+        headers = {
+            'From': 'from@bar.tld',
+            'To': 'to@bar.tld',
+            'Subject': 'Some test mail',
+        }
+
+        # when
+        se.index_mail(test_helper.pixelated_mail(extra_headers=headers))
+
+        # then
+        self.assertTrue(lock_stub.called)
+
+    def test_encoding(self):
+        # given
+        soledad_querier = mock()
+        when(soledad_querier).get_index_masterkey().thenReturn(INDEX_KEY)
+
+        se = SearchEngine(soledad_querier, self.agent_home)
+
+        headers = {
+            'From': 'foo@bar.tld',
+            'To': '=?utf-8?b?IsOEw7zDtiDDlsO8w6QiIDxmb2xrZXJAcGl4ZWxhdGVkLXByb2plY3Qub3Jn?=\n =?utf-8?b?PiwgRsO2bGtlciA8Zm9sa2VyQHBpeGVsYXRlZC1wcm9qZWN0Lm9yZz4=?=',
+            'Cc': '=?utf-8?b?IsOEw7zDtiDDlsO8w6QiIDxmb2xrZXJAcGl4ZWxhdGVkLXByb2plY3Qub3Jn?=\n =?utf-8?b?PiwgRsO2bGtlciA8Zm9sa2VyQHBpeGVsYXRlZC1wcm9qZWN0Lm9yZz4=?=',
+            'Subject': 'Some test mail',
+        }
+
+        # when
+        se.index_mail(test_helper.pixelated_mail(extra_headers=headers, chash='mailid'))
+
+        result = se.search('folker')
+
+        self.assertEqual((['mailid'], 1), result)