migrating mails resources to use services factory

Issue #576

1 files changed, 10 insertions, 4 deletions

diff --git a/service/test/unit/resources/test_mails_resource.py b/service/test/unit/resources/test_mails_resource.py
index efa37b27..e36e4f5f 100644
--- a/service/test/unit/resources/test_mails_resource.py
+++ b/service/test/unit/resources/test_mails_resource.py
@@ -17,7 +17,7 @@
 import unittest
 
 from mock import patch
-from mockito import mock, when, verify
+from mockito import mock, when, verify, any as ANY
 from twisted.internet import defer
 from twisted.web.test.requesthelper import DummyRequest
 
@@ -28,6 +28,12 @@ from test.unit.resources import DummySite
 class TestMailsResource(unittest.TestCase):
     def setUp(self):
         self.mail_service = mock()
+        self.servicesFactory = mock()
+        self.services = mock()
+        self.services.mail_service = self.mail_service
+        self.services.draft_service = mock()
+        self.servicesFactory._services_by_user = {'someuserid': self.mail_service}
+        when(self.servicesFactory).services(ANY()).thenReturn(self.services)
 
     @patch('leap.common.events.register')
     def test_render_GET_should_unicode_mails_search_query(self, mock_register):
@@ -40,7 +46,7 @@ class TestMailsResource(unittest.TestCase):
         unicodified_search_term = u'coração'
         when(self.mail_service).mails(unicodified_search_term, 25, 1).thenReturn(defer.Deferred())
 
-        mails_resource = MailsResource(self.mail_service, mock())
+        mails_resource = MailsResource(self.servicesFactory)
         mails_resource.isLeaf = True
         web = DummySite(mails_resource)
         d = web.get(request)
@@ -60,7 +66,7 @@ class TestMailsResource(unittest.TestCase):
         when(self.mail_service).attachment('some fake attachment id').thenReturn(defer.Deferred())
         request.content = content
 
-        mails_resource = MailsResource(self.mail_service, mock())
+        mails_resource = MailsResource(self.servicesFactory)
         mails_resource.isLeaf = True
         web = DummySite(mails_resource)
         d = web.get(request)
@@ -84,7 +90,7 @@ class TestMailsResource(unittest.TestCase):
             .thenReturn(defer.succeed(as_dictable))
         request.content = content
 
-        mails_resource = MailsResource(self.mail_service, mock())
+        mails_resource = MailsResource(self.servicesFactory)
         mails_resource.isLeaf = True
         web = DummySite(mails_resource)
         d = web.get(request)