Operation - Parcels - Modify: first attempt to testing associated parcel with a context record #5227

1 files changed, 29 insertions, 3 deletions

diff --git a/archaeological_operations/tests.py b/archaeological_operations/tests.py
index 768c11d16..448ee4d2e 100644
--- a/archaeological_operations/tests.py
+++ b/archaeological_operations/tests.py
@@ -1660,12 +1660,25 @@ class ParcelTest(ImportTest, TestCase):
         town_2 = Town.objects.create(name="Minas Tirith", numero_insee="10901")
         parcel_1 = models.Parcel.objects.create(town=town_1, year=2640, section="M", parcel_number="1", public_domain=False)
         parcel_2 = models.Parcel.objects.create(town=town_2, year=2620, section="M", parcel_number="2", public_domain=True)
+        parcel_associated = models.Parcel.objects.create(town=town_1, year=2000, section="ASSO", parcel_number="10", public_domain=False)
 
         operation = create_operation(user)
         operation.towns.add(town_1)
         operation.towns.add(town_2)
         operation.parcels.add(parcel_1)
         operation.parcels.add(parcel_2)
+        operation.parcels.add(parcel_associated)
+
+        cr_data = {
+            "label": "Context record",
+            "operation": operation,
+            "parcel": parcel_associated,
+            "history_modifier": user,
+        }
+        ContextRecord = apps.get_model(
+            "archaeological_context_records", "ContextRecord"
+        )
+        cr = ContextRecord.objects.create(**cr_data)
 
         response = c.get(reverse("operation-parcels-modify", kwargs={"pk": operation.pk}))
 
@@ -1674,10 +1687,13 @@ class ParcelTest(ImportTest, TestCase):
         self.assertContains(response, town_2.name)
         self.assertContains(response, parcel_1.year)
         self.assertContains(response, parcel_2.year)
+        self.assertContains(response, parcel_associated.year)
         self.assertContains(response, parcel_1.section)
         self.assertContains(response, parcel_2.section)
+        self.assertContains(response, parcel_associated.section)
         self.assertContains(response, parcel_1.parcel_number)
         self.assertContains(response, parcel_2.parcel_number)
+        self.assertContains(response, parcel_associated.parcel_number)
 
         data = {}
         for idx in range(2 + views.PARCEL_FORMSET_EXTRA_FORM):
@@ -1701,10 +1717,15 @@ class ParcelTest(ImportTest, TestCase):
             "form-1-section": "SCT2",
             "form-1-parcel_number": "002",
             "form-1-town": town_2.pk,
-            "form-2-year": 2011,
-            "form-2-section": "XXX",
-            "form-2-parcel_number": "003",
+            "form-2-year": 2000,
+            "form-2-section": "ASSO",
+            "form-2-parcel_number": "10",
             "form-2-town": town_2.pk,
+            "form-2-DELETE": "on",
+            "form-3-year": 2011,
+            "form-3-section": "XXX",
+            "form-3-parcel_number": "003",
+            "form-3-town": town_2.pk,
         })
         post_response = c.post("/operation-parcels-modify/" + str(operation.pk) + "/", data)
 
@@ -1719,6 +1740,8 @@ class ParcelTest(ImportTest, TestCase):
         self.assertContains(response, "002")
         self.assertContains(response, "003")
         self.assertContains(response, "XXX")
+        self.assertContains(response, "This parcel is associated with a context "
+                                          "record. It can't be deleted.")
 
         parcels = models.Parcel.objects.all()
         self.assertEqual(parcels.count(), 2)
@@ -1726,6 +1749,9 @@ class ParcelTest(ImportTest, TestCase):
         deleted_parcel = models.Parcel.objects.filter(pk=parcel_1.pk)
         self.assertEqual(deleted_parcel.count(), 0)
 
+        associated_parcel = models.Parcel.objects.filter(pk=parcel_associated.pk)
+        self.assertEqual(associated_parcel.count(), 1)
+
         data.update({
             "_town": town_2.pk,
             "_parcel_selection": "2013:XD:10 to 20, YD:24",