From 85264c9ab3052dd844016d3ec44cb8af43dd5869 Mon Sep 17 00:00:00 2001 From: Folker Bernitt Date: Wed, 8 Oct 2014 18:06:28 +0200 Subject: Added whoosh as search engine (issue #94). --- service/test/unit/user_agent_test.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) (limited to 'service/test/unit/user_agent_test.py') diff --git a/service/test/unit/user_agent_test.py b/service/test/unit/user_agent_test.py index 22199333..83a35246 100644 --- a/service/test/unit/user_agent_test.py +++ b/service/test/unit/user_agent_test.py @@ -35,10 +35,14 @@ class UserAgentTest(unittest.TestCase): self.app = pixelated.user_agent.app.test_client() self.mail_service = mock() self.tag_service = mock() - self.mail_service.tag_service = self.tag_service + self.draft_service = mock() + self.search_engine = mock() pixelated.user_agent.DISABLED_FEATURES = [] pixelated.user_agent.mail_service = self.mail_service + pixelated.user_agent.tag_service = self.tag_service + pixelated.user_agent.draft_service = self.draft_service + pixelated.user_agent.search_engine = self.search_engine self.input_mail = None pixelated.adapter.pixelated_mail.input_mail_from_dict = lambda x: self.input_mail @@ -50,7 +54,7 @@ class UserAgentTest(unittest.TestCase): self.app.post('/mails', data='{}', content_type="application/json") - verify(self.mail_service).create_draft(self.input_mail) + verify(self.draft_service).create_draft(self.input_mail) def test_create_or_send_draft_should_send_draft_if_mail_has_ident(self): self.input_mail = self.draft() @@ -61,6 +65,7 @@ class UserAgentTest(unittest.TestCase): def test_sending_mail_return_sent_mail_data_when_send_succeeds(self): self.input_mail = self.draft() + when(self.mail_service).send(1, self.input_mail).thenReturn(self.input_mail) self.input_mail.as_dict = lambda: {'header': {'from': 'a@a.a', 'to': 'b@b.b'}, 'ident': 1, 'tags': [], @@ -89,11 +94,11 @@ class UserAgentTest(unittest.TestCase): def test_update_draft(self): self.input_mail = self.draft() - when(self.mail_service).update_draft(1, self.input_mail).thenReturn(self.input_mail) + when(self.draft_service).update_draft(1, self.input_mail).thenReturn(self.input_mail) self.app.put('/mails', data='{"ident":1}', content_type="application/json") - verify(self.mail_service).update_draft(1, self.input_mail) + verify(self.draft_service).update_draft(1, self.input_mail) def draft(self): return test_helper.input_mail() -- cgit v1.2.3