diff options
Diffstat (limited to 'ishtar_common/tests.py')
| -rw-r--r-- | ishtar_common/tests.py | 88 |
1 files changed, 82 insertions, 6 deletions
diff --git a/ishtar_common/tests.py b/ishtar_common/tests.py index 94a636b7b..05799a0aa 100644 --- a/ishtar_common/tests.py +++ b/ishtar_common/tests.py @@ -1253,7 +1253,7 @@ class UserProfileTest(TestCase): base_profile = models.UserProfile.objects.get(pk=base_profile.pk) self.assertEqual( base_profile.name, - u"New name" + "New name" ) self.client.post( @@ -2217,7 +2217,6 @@ class ShortMenuTest(TestCase): class ImportTest(TestCase): - def create_import(self): create_user() imp_model = models.ImporterModel.objects.create( @@ -2494,8 +2493,6 @@ class NewItems(TestCase): self.assertEqual(person.author.count(), 1) - - class AccountWizardTest(WizardTest, TestCase): fixtures = [settings.ROOT_PATH + '../fixtures/initial_data-auth-fr.json', @@ -2538,7 +2535,6 @@ class AccountWizardTest(WizardTest, TestCase): class DashboardTest(TestCase): - def setUp(self): self.username, self.password, self.user = create_superuser() profile, created = models.IshtarSiteProfile.objects.get_or_create( @@ -2568,7 +2564,6 @@ class DashboardTest(TestCase): class CleanMedia(TestCase): - def test_rename(self): test_names = [ ("éofficier2-12-02-04.93_gvK3hAr-1_2m7zZPn-1_nKhh2S2-1_"\ @@ -2646,3 +2641,84 @@ class PersonQATest(TestCase): self.title_2 ) + +class DocumentTest(TestCase): + def test_custom_index(self): + Operation = apps.get_model("archaeological_operations", "Operation") + ContextRecord = apps.get_model("archaeological_context_records", + "ContextRecord") + Unit = apps.get_model("archaeological_context_records", "Unit") + BaseFind = apps.get_model("archaeological_finds", "BaseFind") + 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( + code_patriarche="001", + operation_type_id=operation_type.pk) + 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") + + profile, created = models.IshtarSiteProfile.objects.get_or_create( + slug='default', active=True) + profile.document_complete_identifier = \ + "{operation_codes}-{source_type__code}-{custom_index}" + profile.document_custom_index = "operation" + profile.save() + + doc = models.Document.objects.create(source_type=st1, + title="Operation report") + doc.operations.add(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, + title="Illustration CR") + doc2.context_records.add(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, + title="Operation report 2") + doc3.operations.add(ope2) + doc3 = models.Document.objects.get(pk=doc3.pk) + self.assertEqual(doc3.complete_identifier, "002-REP-1") + + doc3.operations.add(ope1) + doc3.custom_index = None + doc3.save() + doc3 = models.Document.objects.get(pk=doc3.pk) + self.assertEqual(doc3.custom_index, None) # 2 operations - no index + self.assertEqual(doc3.complete_identifier, '001|002-REP-') + + # complex jinja identifier + profile.document_complete_identifier = \ + "{% if custom_index %}{{operation_codes}}-{{source_type__code}}-" \ + "{{ \"%03d\" % (custom_index|int)}}{% else %}no-code{% endif %}" + profile.save() + + doc3.operations.remove(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.custom_index = None + doc3.complete_identifier = "" + doc3.save() + doc3 = models.Document.objects.get(pk=doc3.pk) + self.assertEqual(doc3.complete_identifier, 'no-code') |