From d9d4ebf3a24db79e6d193a27187c2cde067c89e2 Mon Sep 17 00:00:00 2001
From: jfriedli
Date: Tue, 24 Sep 2019 10:59:00 -0700
Subject: Resolve "Delete Files After Adding To Zip"

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

(limited to 'test/test_api.py')

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': [
-- 
cgit v1.3