diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/test.py | 3 | ||||
| -rw-r--r-- | test/test_api.py | 5 | ||||
| -rw-r--r-- | test/test_file_removal_scheduler.py | 1 |
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 | |||
| 195 | if __name__ == '__main__': | 194 | if __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) |
