migrating AttachmentsResources to use the service factory

Issue #576

1 files changed, 8 insertions, 2 deletions

diff --git a/service/test/unit/resources/test_attachments_resource.py b/service/test/unit/resources/test_attachments_resource.py
index 8beb02fa..f3d8ce53 100644
--- a/service/test/unit/resources/test_attachments_resource.py
+++ b/service/test/unit/resources/test_attachments_resource.py
@@ -3,7 +3,7 @@ import unittest
 import logging
 
 from mock import patch, MagicMock
-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
 
@@ -17,7 +17,13 @@ class AttachmentsResourceTest(unittest.TestCase):
 
     def setUp(self):
         self.mail_service = mock()
-        self.mails_resource = AttachmentsResource(self.mail_service)
+        self.servicesFactory = mock()
+        self.services = mock()
+        self.services.mail_service = self.mail_service
+        self.servicesFactory._services_by_user = {'someuserid': self.mail_service}
+        when(self.servicesFactory).services(ANY()).thenReturn(self.services)
+
+        self.mails_resource = AttachmentsResource(self.servicesFactory)
         self.mails_resource.isLeaf = True
         self.web = DummySite(self.mails_resource)