From 859acb48a27d8c4a4e479f873ccb012cc77e4670 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Thu, 9 Nov 2017 19:29:55 +0100 Subject: Manage fix of associated items (after imports or database dump) --- archaeological_operations/tests.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) (limited to 'archaeological_operations/tests.py') 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') -- cgit v1.2.3