diff options
Diffstat (limited to 'archaeological_operations/tests.py')
| -rw-r--r-- | archaeological_operations/tests.py | 65 |
1 files changed, 54 insertions, 11 deletions
diff --git a/archaeological_operations/tests.py b/archaeological_operations/tests.py index b8db7e2c7..385337213 100644 --- a/archaeological_operations/tests.py +++ b/archaeological_operations/tests.py @@ -91,7 +91,8 @@ from ishtar_common.models import ( Document, ValueFormater, Regexp, - MediaExporter + MediaExporter, + SourceType ) from ishtar_common.models_rest import ApiUser, ApiSearchModel, ApiSheetFilter from archaeological_files.models import File, FileType @@ -4101,7 +4102,8 @@ class OperationExportMediaTest(TestCase, TestPermissionQuery, OperationInitTest) response = c.get(self.export_url) self.assertEqual(response.status_code, 200) - def _test_files(self, response, filenames): + def _test_files(self, response, filenames, exclude=None): + exclude = exclude or [] with tempfile.TemporaryDirectory() as tmpdir: f, z = None, None try: @@ -4117,6 +4119,8 @@ class OperationExportMediaTest(TestCase, TestPermissionQuery, OperationInitTest) image = f"{tmpdir}/{filename}" with Image.open(image) as im: self.assertIsNone(im.verify()) + for excluded in exclude: + self.assertNotIn(exclude, name_list) finally: if z: z.close() @@ -4124,27 +4128,37 @@ class OperationExportMediaTest(TestCase, TestPermissionQuery, OperationInitTest) f.close() def test_content(self): + document = self.documents[0] + document.pk = None + document.save() + self.operation.documents.add(document) + c = Client() c.login(username=self.username, password=self.password) response = c.get(self.export_url) - self._test_files(response, [('operation_image_00001.png', True)]) + self._test_files(response, [ + ('operation_image_00001.png', True), + ('operation_image_00002.png', True), + ]) self.exporter.files_to_export = 'A' self.exporter.save() response = c.get(self.export_url) - self._test_files( - response, - [('operation_image_00001.png', True), ('operation_file_00001.txt', False), ] - ) + self._test_files(response, [ + ('operation_image_00001.png', True), + ('operation_image_00002.png', True), + ('operation_file_00001.txt', False), + ]) self.exporter.thumbnail_for_images = True self.exporter.save() response = c.get(self.export_url) - self._test_files( - response, - [('operation_image_00001.jpg', True), ('operation_file_00001.txt', False), ] - ) + self._test_files(response, [ + ('operation_image_00001.jpg', True), + ('operation_image_00002.jpg', True), + ('operation_file_00001.txt', False), + ]) def _add_find(self): cr_data = { @@ -4189,6 +4203,35 @@ class OperationExportMediaTest(TestCase, TestPermissionQuery, OperationInitTest) ('find_image_00001.png', True),] ) + def test_query(self): + document_type = SourceType.objects.all()[0] + document = self.documents[0] + document.pk = None + document.source_type = document_type + document.save() + self.operation.documents.add(document) + + c = Client() + c.login(username=self.username, password=self.password) + + # classic export + response = c.get(self.export_url) + self._test_files( + response, + [('operation_image_00001.png', True), + ('operation_image_00002.png', True)] + ) + + # filter: only one document is relevant + self.exporter.query = f"type={document_type.label}" + self.exporter.save() + response = c.get(self.export_url) + self._test_files( + response, + [('operation_image_00001.png', True)], + exclude=[('operation_image_00002.png', True)] + ) + class LabelTest(TestCase, OperationInitTest): fixtures = FILE_FIXTURES |