diff options
Diffstat (limited to 'ishtar_common/tests.py')
| -rw-r--r-- | ishtar_common/tests.py | 82 |
1 files changed, 60 insertions, 22 deletions
diff --git a/ishtar_common/tests.py b/ishtar_common/tests.py index 71e599d59..72836fa45 100644 --- a/ishtar_common/tests.py +++ b/ishtar_common/tests.py @@ -2757,7 +2757,7 @@ class PersonQATest(TestCase): class DocumentTest(TestCase): - def test_custom_index(self): + def setUp(self): Operation = apps.get_model("archaeological_operations", "Operation") ContextRecord = apps.get_model("archaeological_context_records", "ContextRecord") @@ -2766,25 +2766,27 @@ class DocumentTest(TestCase): Find = apps.get_model("archaeological_finds", "Find") operation_type, __ = models.OperationType.objects.get_or_create( txt_idx="arch_diagnostic", label="Diagnostic") - ope1 = Operation.objects.create( + self.ope1 = Operation.objects.create( code_patriarche="001", operation_type_id=operation_type.pk) - ope2 = Operation.objects.create( + self.ope2 = Operation.objects.create( code_patriarche="002", operation_type_id=operation_type.pk) su, __ = Unit.objects.get_or_create( txt_idx='stratigraphic-unit', label="Stratigraphic unit", order=1) - cr1 = ContextRecord.objects.create(operation=ope1, unit=su) - cr2 = ContextRecord.objects.create(operation=ope2, unit=su) - bf1 = BaseFind.objects.create(context_record=cr1) - bf2 = BaseFind.objects.create(context_record=cr2) - find1 = Find.objects.create() - find1.base_finds.add(bf1) - find2 = Find.objects.create() - find2.base_finds.add(bf2) - st1 = models.SourceType.objects.create(label="Report", code="REP") - st2 = models.SourceType.objects.create(label="Illustration", code="ILL") + self.cr1 = ContextRecord.objects.create(operation=self.ope1, unit=su) + self.cr2 = ContextRecord.objects.create(operation=self.ope2, unit=su) + bf1 = BaseFind.objects.create(context_record=self.cr1) + bf2 = BaseFind.objects.create(context_record=self.cr2) + self.find1 = Find.objects.create() + self.find1.base_finds.add(bf1) + self.find2 = Find.objects.create() + self.find2.base_finds.add(bf2) + self.st1 = models.SourceType.objects.create(label="Report", code="REP") + self.st2 = models.SourceType.objects.create(label="Illustration", + code="ILL") + def test_custom_index(self): profile, created = models.IshtarSiteProfile.objects.get_or_create( slug='default', active=True) profile.document_complete_identifier = \ @@ -2792,25 +2794,25 @@ class DocumentTest(TestCase): profile.document_custom_index = "operation" profile.save() - doc = models.Document.objects.create(source_type=st1, + doc = models.Document.objects.create(source_type=self.st1, title="Operation report") - doc.operations.add(ope1) + doc.operations.add(self.ope1) doc = models.Document.objects.get(pk=doc.pk) self.assertEqual(doc.complete_identifier, "001-REP-1") - doc2 = models.Document.objects.create(source_type=st2, + doc2 = models.Document.objects.create(source_type=self.st2, title="Illustration CR") - doc2.context_records.add(cr1) + doc2.context_records.add(self.cr1) doc2 = models.Document.objects.get(pk=doc2.pk) self.assertEqual(doc2.complete_identifier, "001-ILL-2") - doc3 = models.Document.objects.create(source_type=st1, + doc3 = models.Document.objects.create(source_type=self.st1, title="Operation report 2") - doc3.operations.add(ope2) + doc3.operations.add(self.ope2) doc3 = models.Document.objects.get(pk=doc3.pk) self.assertEqual(doc3.complete_identifier, "002-REP-1") - doc3.operations.add(ope1) + doc3.operations.add(self.ope1) doc3.custom_index = None doc3.save() doc3 = models.Document.objects.get(pk=doc3.pk) @@ -2823,20 +2825,56 @@ class DocumentTest(TestCase): "{{ \"%03d\" % (custom_index|int)}}{% else %}no-code{% endif %}" profile.save() - doc3.operations.remove(ope1) + doc3.operations.remove(self.ope1) doc3.custom_index = None doc3.complete_identifier = "" doc3.save() doc3 = models.Document.objects.get(pk=doc3.pk) self.assertEqual(doc3.complete_identifier, '002-REP-001') - doc3.operations.remove(ope2) + doc3.operations.remove(self.ope2) doc3.custom_index = None doc3.complete_identifier = "" doc3.save() doc3 = models.Document.objects.get(pk=doc3.pk) self.assertEqual(doc3.complete_identifier, 'no-code') + def test_clean_duplicate_association(self): + doc = models.Document.objects.create(source_type=self.st1, + title="Operation report") + doc.operations.add(self.ope1) + doc.context_records.add(self.cr1) + doc = models.Document.objects.get(pk=doc.pk) + self.assertEqual(doc.operations.count(), 1) + + doc.context_records.remove(self.cr1) + profile = models.get_current_profile() + profile.clean_redundant_document_association = True + profile.save() + + doc = models.Document.objects.get(pk=doc.pk) + doc.context_records.add(self.cr1) + + doc = models.Document.objects.get(pk=doc.pk) + self.assertEqual(doc.context_records.count(), 1) + self.assertEqual(doc.operations.count(), 0) + + doc.context_records.remove(self.cr1) + doc.operations.add(self.ope1) + doc.context_records.add(self.cr2) + self.assertEqual(doc.operations.count(), 1) # different ope keep it + self.assertEqual(doc.context_records.count(), 1) + + # adding a find - remove the operation + doc.finds.add(self.find1) + self.assertEqual(doc.finds.count(), 1) + self.assertEqual(doc.context_records.count(), 1) + self.assertEqual(doc.operations.count(), 0) + + # adding an operation is undo + doc.operations.add(self.ope1) + self.assertEqual(doc.operations.count(), 0) + class JinjaFilterTest(TestCase): def test_splitpart(self): |