summaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
Diffstat (limited to 'test')
-rw-r--r--test/test.py3
-rw-r--r--test/test_api.py5
-rw-r--r--test/test_file_removal_scheduler.py1
3 files changed, 5 insertions, 4 deletions
diff --git a/test/test.py b/test/test.py
index 6734dcf..97e7567 100644
--- a/test/test.py
+++ b/test/test.py
@@ -170,7 +170,7 @@ class Mat2WebTestCase(TestCase):
170 170
171 request = app.get(self.get_context_variable('download_uri')) 171 request = app.get(self.get_context_variable('download_uri'))
172 self.assertEqual(302, request.status_code) 172 self.assertEqual(302, request.status_code)
173 os.environ['MAT2_MAX_FILE_AGE_FOR_REMOVAL'] = '9999' 173 os.environ['MAT2_MAX_FILE_AGE_FOR_REMOVAL'] = str(15*60)
174 174
175 def test_info_page(self): 175 def test_info_page(self):
176 rv = self.client.get('/info') 176 rv = self.client.get('/info')
@@ -191,7 +191,6 @@ class Mat2WebTestCase(TestCase):
191 self.assertIn(b'Invalid Filename', rv.data) 191 self.assertIn(b'Invalid Filename', rv.data)
192 192
193 193
194
195if __name__ == '__main__': 194if __name__ == '__main__':
196 unittest.main() 195 unittest.main()
197 196
diff --git a/test/test_api.py b/test/test_api.py
index 63955d0..1a50aa6 100644
--- a/test/test_api.py
+++ b/test/test_api.py
@@ -42,7 +42,7 @@ class Mat2APITestCase(unittest.TestCase):
42 self.assertEqual(request.status_code, 200) 42 self.assertEqual(request.status_code, 200)
43 43
44 data = request.get_json() 44 data = request.get_json()
45 self.assertEqual(data['output_filename'], 'test_name.cleaned.jpg') 45 self.assertEqual(data['inactive_after_sec'], 15 * 60)
46 self.assertEqual(data['output_filename'], 'test_name.cleaned.jpg') 46 self.assertEqual(data['output_filename'], 'test_name.cleaned.jpg')
47 self.assertEqual(data['mime'], 'image/jpeg') 47 self.assertEqual(data['mime'], 'image/jpeg')
48 self.assertEqual(len(data['secret']), 64) 48 self.assertEqual(len(data['secret']), 64)
@@ -237,6 +237,7 @@ class Mat2APITestCase(unittest.TestCase):
237 237
238 self.assertIn('files.', response['output_filename']) 238 self.assertIn('files.', response['output_filename'])
239 self.assertIn('cleaned.zip', response['output_filename']) 239 self.assertIn('cleaned.zip', response['output_filename'])
240 self.assertEquals(15 * 60, response['inactive_after_sec'])
240 self.assertIn(response['mime'], 'application/zip') 241 self.assertIn(response['mime'], 'application/zip')
241 self.assertEqual(response['meta_after'], {}) 242 self.assertEqual(response['meta_after'], {})
242 243
@@ -387,7 +388,7 @@ class Mat2APITestCase(unittest.TestCase):
387 randint_mock.return_value = 0 388 randint_mock.return_value = 0
388 self.upload_download_test_jpg_and_assert_response_code(app, 404) 389 self.upload_download_test_jpg_and_assert_response_code(app, 404)
389 390
390 os.environ['MAT2_MAX_FILE_AGE_FOR_REMOVAL'] = '9999' 391 os.environ['MAT2_MAX_FILE_AGE_FOR_REMOVAL'] = str(15 * 60)
391 392
392 def upload_download_test_jpg_and_assert_response_code(self, app, code): 393 def upload_download_test_jpg_and_assert_response_code(self, app, code):
393 request = app.post('/api/upload', 394 request = app.post('/api/upload',
diff --git a/test/test_file_removal_scheduler.py b/test/test_file_removal_scheduler.py
index 1210cb2..563fc73 100644
--- a/test/test_file_removal_scheduler.py
+++ b/test/test_file_removal_scheduler.py
@@ -42,6 +42,7 @@ class Mat2WebTestCase(unittest.TestCase):
42 randint_mock.return_value = 0 42 randint_mock.return_value = 0
43 file_removal_scheduler.run_file_removal_job(self.app.config['UPLOAD_FOLDER']) 43 file_removal_scheduler.run_file_removal_job(self.app.config['UPLOAD_FOLDER'])
44 self.assertTrue(path.exists(path.join(self.upload_folder, filename))) 44 self.assertTrue(path.exists(path.join(self.upload_folder, filename)))
45 environ['MAT2_MAX_FILE_AGE_FOR_REMOVAL'] = str(15 * 60)
45 46
46 def tearDown(self): 47 def tearDown(self):
47 shutil.rmtree(self.upload_folder) 48 shutil.rmtree(self.upload_folder)