Resolve "Delete Files After Adding To Zip"

1 files changed, 13 insertions, 0 deletions

diff --git a/test/test_api.py b/test/test_api.py
index 4b67478..532ceb9 100644
--- a/test/test_api.py
+++ b/test/test_api.py
@@ -3,6 +3,9 @@ import tempfile
 import json
 import os
 import shutil
+import zipfile
+
+from six import BytesIO
 
 import main
 
@@ -208,11 +211,21 @@ class Mat2APITestCase(unittest.TestCase):
         self.assertEqual(response['meta_after'], {})
 
         request = self.app.get(response['download_link'])
+        zip_response = zipfile.ZipFile(BytesIO(request.data))
+        self.assertEquals(2, len(zip_response.namelist()))
+        for name in zip_response.namelist():
+            self.assertIn('.cleaned.jpg', name)
         self.assertEqual(request.status_code, 200)
 
         request = self.app.get(response['download_link'])
         self.assertEqual(request.status_code, 404)
 
+        request = self.app.get(upload_one['download_link'])
+        self.assertEqual(request.status_code, 404)
+
+        request = self.app.get(upload_two['download_link'])
+        self.assertEqual(request.status_code, 404)
+
     def test_api_bulk_download_validation(self):
         post_body = {
             u'download_list': [