diff options
Diffstat (limited to 'service/test')
| -rw-r--r-- | service/test/integration/test_retrieve_attachment.py | 12 | ||||
| -rw-r--r-- | service/test/support/integration/app_test_client.py | 9 |
2 files changed, 16 insertions, 5 deletions
diff --git a/service/test/integration/test_retrieve_attachment.py b/service/test/integration/test_retrieve_attachment.py index 7de03c59..4aaeadc2 100644 --- a/service/test/integration/test_retrieve_attachment.py +++ b/service/test/integration/test_retrieve_attachment.py @@ -27,22 +27,28 @@ from test.support.integration.soledad_test_base import SoledadTestBase class RetrieveAttachmentTest(SoledadTestBase): - @defer.inlineCallbacks def test_attachment_content_is_retrieved(self): attachment_id, input_mail = self._create_mail_with_attachment() yield self.mail_store.add_mail('INBOX', input_mail.as_string()) - attachment, req = yield self.get_attachment(attachment_id, 'base64') + requested_filename = "file name with space" + expected_content_type = 'text/plain' + expected_content_disposition = 'attachment; filename="file name with space"' + + attachment, req = yield self.get_attachment(attachment_id, 'base64', filename=requested_filename, content_type=expected_content_type) self.assertEqual(200, req.code) self.assertEquals('pretend to be binary attachment data', attachment) + self.assertEquals(expected_content_disposition, req.outgoingHeaders['content-disposition']) + self.assertEquals(expected_content_type, req.outgoingHeaders['content-type']) def _create_mail_with_attachment(self): input_mail = MIMEMultipart() input_mail.attach(MIMEText(u'a utf8 message', _charset='utf-8')) attachment = MIMEApplication('pretend to be binary attachment data') - attachment.add_header('Content-Disposition', 'attachment', filename='filename.txt') + attachment.add_header('Content-Disposition', 'attachment', filename='file name.txt') + attachment.add_header('Content-Type', 'text/plain') input_mail.attach(attachment) attachment_id = 'B5B4ED80AC3B894523D72E375DACAA2FC6606C18EDF680FE95903086C8B5E14A' return attachment_id, input_mail diff --git a/service/test/support/integration/app_test_client.py b/service/test/support/integration/app_test_client.py index 99f4ebc7..8ab58397 100644 --- a/service/test/support/integration/app_test_client.py +++ b/service/test/support/integration/app_test_client.py @@ -322,8 +322,13 @@ class AppTestClient(object): defer.returnValue(mails) @defer.inlineCallbacks - def get_attachment(self, ident, encoding): - deferred_result, req = self.get("/attachment/%s" % ident, {'encoding': [encoding]}, as_json=False) + def get_attachment(self, ident, encoding, filename=None, content_type=None): + params = {'encoding': [encoding]} + if filename: + params['filename'] = [filename] + if content_type: + params['content_type'] = [content_type] + deferred_result, req = self.get("/attachment/%s" % ident, params, as_json=False) res = yield deferred_result defer.returnValue((res, req)) |