diff options
| author | Folker Bernitt <fbernitt@thoughtworks.com> | 2014-10-08 18:06:28 +0200 |
|---|---|---|
| committer | Folker Bernitt <fbernitt@thoughtworks.com> | 2014-10-09 18:24:41 +0200 |
| commit | 85264c9ab3052dd844016d3ec44cb8af43dd5869 (patch) | |
| tree | d6a5cb6511039407fdef40804c5338347b71c551 /service/test/unit/user_agent_test.py | |
| parent | fb8782d3f66b1068153a9948a01864ee52e39b10 (diff) | |
Added whoosh as search engine (issue #94).
Diffstat (limited to 'service/test/unit/user_agent_test.py')
| -rw-r--r-- | service/test/unit/user_agent_test.py | 13 |
1 files changed, 9 insertions, 4 deletions
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() |