diff options
Diffstat (limited to 'service/test/unit/resources/test_attachments_resource.py')
-rw-r--r-- | service/test/unit/resources/test_attachments_resource.py | 10 |
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) |