Manage fix of associated items (after imports or database dump)

1 files changed, 17 insertions, 0 deletions

diff --git a/archaeological_operations/tests.py b/archaeological_operations/tests.py
index b75c02cae..ec7ae44c5 100644
--- a/archaeological_operations/tests.py
+++ b/archaeological_operations/tests.py
@@ -857,6 +857,23 @@ class OperationTest(TestCase, OperationInitTest):
         parcel.save()
         self.assertEqual(parcel.external_id, 'blabla')
 
+    def test_associated(self):
+        scientist = Person.objects.create(name="C-3PO")
+        self.item.scientist = scientist
+        self.item.save()
+        scientist = Person.objects.get(pk=scientist.pk)
+        self.assertIn(PersonType.objects.get(txt_idx='head_scientist'),
+                      scientist.person_types.all())
+
+        # do not change if in the list
+        sra = Person.objects.create(name="R2D2")
+        sra.person_types.add(PersonType.objects.get(txt_idx='sra_agent'))
+        self.item.scientist = sra
+        self.item.save()
+        self.assertNotIn(PersonType.objects.get(txt_idx='head_scientist'),
+                         sra.person_types.all())
+
+
     def create_relations(self):
         rel1 = models.RelationType.objects.create(
             symmetrical=True, label='Include', txt_idx='include')