Operations: test bulk update of finds and base finds (refs #3472)

1 files changed, 40 insertions, 0 deletions

diff --git a/archaeological_operations/tests.py b/archaeological_operations/tests.py
index 25f278880..23c32434b 100644
--- a/archaeological_operations/tests.py
+++ b/archaeological_operations/tests.py
@@ -782,6 +782,21 @@ class OperationTest(TestCase, OperationInitTest):
                    'history_modifier': self.get_default_user()}
         cr = ContextRecord.objects.create(**cr_data)
 
+        from archaeological_finds.models import BaseFind, Find, MaterialType
+        bf_data = {
+            'label': "Base find", 'history_modifier': self.get_default_user(),
+            'context_record': cr
+        }
+        base_find = BaseFind.objects.create(**bf_data)
+        find = Find.objects.create(
+            history_modifier=self.get_default_user(),
+            label='Find me'
+        )
+        find.base_finds.add(base_find)
+        mat = MaterialType.objects.create(
+            label='Adamentium', txt_idx='admentium', code='ADA')
+        find.material_types.add(mat)
+
         class TestObj(object):
             def __init__(self):
                 self.context_record_reached = []
@@ -806,6 +821,21 @@ class OperationTest(TestCase, OperationInitTest):
         self.assertEqual(ope_id, '{}2011-1'.format(
             settings.ISHTAR_DEF_OPE_PREFIX))
 
+        base_find = BaseFind.objects.get(pk=base_find.pk)
+        op_code, idx = base_find.cache_short_id.split(' | ')
+        self.assertEqual(op_code, 'OP2011-1')
+        self.assertEqual(idx, '00001')
+        op_code, mat_code, lbl, idx = base_find.cache_complete_id.split(' | ')
+        self.assertEqual(op_code, 'OP2011-1')
+        self.assertEqual(mat_code, 'ADA')
+        self.assertEqual(lbl, 'Context record')
+        self.assertEqual(idx, '00001')
+
+        find = Find.objects.get(pk=find.pk)
+        op_code_idx, lbl = find.cached_label.split(' | ')
+        self.assertEqual(op_code_idx, 'OP2011-1-00001')
+        self.assertEqual(lbl, 'Find me')
+
         operation = models.Operation.objects.get(pk=operation.pk)
         operation.code_patriarche = '666'
         operation.save()
@@ -813,6 +843,16 @@ class OperationTest(TestCase, OperationInitTest):
         ope_id, parcel_sec, parcel_nb, cr_label = cr.cached_label.split(' | ')
         self.assertEqual(ope_id, '{}666'.format(settings.ISHTAR_OPE_PREFIX))
 
+        base_find = BaseFind.objects.get(pk=base_find.pk)
+        op_code, idx = base_find.cache_short_id.split(' | ')
+        self.assertEqual(op_code, 'OA666')
+        op_code, mat_code, lbl, idx = base_find.cache_complete_id.split(' | ')
+        self.assertEqual(op_code, 'OA666')
+
+        find = Find.objects.get(pk=find.pk)
+        op_code_idx, lbl = find.cached_label.split(' | ')
+        self.assertEqual(op_code_idx, 'OA666-00001')
+
 
 class OperationSearchTest(TestCase, OperationInitTest):
     fixtures = [settings.ROOT_PATH +