From f544d94a8c903dcdc9f576af32958334b4a2a71c Mon Sep 17 00:00:00 2001
From: jfriedli
Date: Thu, 3 Oct 2019 04:29:42 -0700
Subject: Content disposition header

---
 test/test_api.py | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'test/test_api.py')

diff --git a/test/test_api.py b/test/test_api.py
index 2029820..de297c4 100644
--- a/test/test_api.py
+++ b/test/test_api.py
@@ -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():
-- 
cgit v1.3