diff options
Diffstat (limited to 'service/test/unit/resources/test_attachments_resource.py')
-rw-r--r-- | service/test/unit/resources/test_attachments_resource.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/service/test/unit/resources/test_attachments_resource.py b/service/test/unit/resources/test_attachments_resource.py index 2afa208c..c44dab84 100644 --- a/service/test/unit/resources/test_attachments_resource.py +++ b/service/test/unit/resources/test_attachments_resource.py @@ -27,6 +27,7 @@ class AttachmentsResourceTest(unittest.TestCase): _file = MagicMock() _file.value = 'some mocked value' _file.type = 'some mocked type' + _file.filename = 'filename.txt' mock_fields.return_value = {'attachment': _file} when(self.mail_service).save_attachment('some mocked value', 'some mocked type').thenReturn(defer.succeed(attachment_id)) @@ -35,7 +36,9 @@ class AttachmentsResourceTest(unittest.TestCase): def assert_response(_): self.assertEqual(201, request.code) self.assertEqual('/attachment/%s' % attachment_id, request.headers['Location']) - self.assertEqual({'attachment_id': attachment_id}, json.loads(request.written[0])) + response_json = {'attachment_id': attachment_id, 'content-type': 'some mocked type', + 'filename': 'filename.txt', 'filesize': 17} + self.assertEqual(response_json, json.loads(request.written[0])) verify(self.mail_service).save_attachment('some mocked value', 'some mocked type') d.addCallback(assert_response) |