Git-Mirrors/mat2-web
1
0
Fork 0
mirror of https://0xacab.org/jvoisin/mat2-web.git synced 2025-05-10 18:25:03 -04:00
Code Issues Projects Releases Packages Wiki Activity

Content disposition header

Browse source
This commit is contained in:
jfriedli 2019-10-03 04:29:42 -07:00
parent c36fc9c20e
commit f544d94a8c
2 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
main.py
View file

@ -50,7 +50,7 @@ def create_app(test_config=None):
def remove_file(response):
os.remove(complete_path)
return response
return send_from_directory(app.config['UPLOAD_FOLDER'], filepath)
return send_from_directory(app.config['UPLOAD_FOLDER'], filepath, as_attachment=True)
@app.route('/', methods=['GET', 'POST'])
def upload_file():
@ -180,7 +180,7 @@ def create_app(test_config=None):
os.remove(complete_path)
return response
return send_from_directory(app.config['UPLOAD_FOLDER'], filepath)
return send_from_directory(app.config['UPLOAD_FOLDER'], filepath, as_attachment=True)
class APIBulkDownloadCreator(Resource):
schema = {

2
test/test_api.py
View file

@ -157,6 +157,7 @@ class Mat2APITestCase(unittest.TestCase):
request = self.app.get(data['download_link'])
self.assertEqual(request.status_code, 200)
self.assertIn('attachment; filename=test_name.cleaned.jpg', request.headers['Content-Disposition'])
request = self.app.get(data['download_link'])
self.assertEqual(request.status_code, 404)
@ -219,6 +220,7 @@ class Mat2APITestCase(unittest.TestCase):
self.assertEqual(request.headers['Content-Length'], '1596')
request = self.app.get(response['download_link'])
self.assertIn('attachment; filename=files.', request.headers['Content-Disposition'])
zip_response = zipfile.ZipFile(BytesIO(request.data))
self.assertEquals(2, len(zip_response.namelist()))
for name in zip_response.namelist():