Fix label generation filter + tests

1 files changed, 52 insertions, 0 deletions

diff --git a/ishtar_common/tests.py b/ishtar_common/tests.py
index 72836fa45..b5de13fe1 100644
--- a/ishtar_common/tests.py
+++ b/ishtar_common/tests.py
@@ -2909,3 +2909,55 @@ class JinjaFilterTest(TestCase):
             "sAINT GEORGES D'OLÉRON")
         self.assertEqual(utils_secretary.lowerfirst_filter("S"), "s")
         self.assertEqual(utils_secretary.lowerfirst_filter(""), "")
+
+
+class TemplateGenerationTest(TestCase):
+    def test_filters(self):
+        model, __ = models.ImporterModel.objects.get_or_create(
+            klass='archaeological_finds.models.Find'
+        )
+        q = models.DocumentTemplate.objects.filter(slug="test")
+        if q.count():
+            q.all()[0].delete()
+        doc = models.DocumentTemplate.objects.create(
+            name="Test",
+            slug="test",
+            associated_model=model,
+            available=True)
+
+        LABEL_EXPECTED_KEYS = [
+            'container_index',
+            'context_record_operation_common_name',
+            'context_record_town_name',
+            'container_location_name',
+            'context_record_operation_operation_type',
+            'container_reference',
+            'my_complete_id',
+            'complete_idx',
+            'complete_idxy',
+        ]
+        tpl_label = settings.ROOT_PATH + \
+            '../ishtar_common/tests/test-filters-label.odt'
+        BASE_EXPECTED_KEYS = [
+            'container_index',
+            'context_record_operation_common_name',
+            'context_record_town_name',
+            'container_location_name',
+            'context_record_operation_operation_type',
+            'container_reference',
+            'my_complete_id',
+            'complete_idx',
+            'complete_idxy',
+        ]
+        tpl_base = settings.ROOT_PATH + \
+              '../ishtar_common/tests/test-filters-base.odt'
+        tests = (
+            (LABEL_EXPECTED_KEYS, tpl_label, models.DocumentTemplate.LABEL_RE),
+            (BASE_EXPECTED_KEYS, tpl_base, models.DocumentTemplate.BASE_RE),
+        )
+        for expected_keys, tpl, filter_re in tests:
+            with open(tpl, 'rb') as tpl:
+                template = SimpleUploadedFile("template.odt", tpl.read())
+                filtr = doc.get_filter(template, filter_re)
+                for key in expected_keys:
+                    self.assertIn(key, filtr)