File - Parcels - Modify: adapting conditions and tests for selection_form #5227

1 files changed, 18 insertions, 6 deletions

diff --git a/archaeological_files/tests.py b/archaeological_files/tests.py
index 2f8f1e158..1d54584da 100644
--- a/archaeological_files/tests.py
+++ b/archaeological_files/tests.py
@@ -542,8 +542,6 @@ class FileOperationTest(TestCase, OperationInitTest, FileInit):
                 f"form-{idx}-DELETE": '',
             })
         data.update({
-            "_town": town_2.pk,
-            "_parcel_selection": "2013:XD:10",
             "form-0-pk": parcel_1.pk,
             "form-0-year": 2021,
             "form-0-section": "SCT1",
@@ -575,16 +573,30 @@ class FileOperationTest(TestCase, OperationInitTest, FileInit):
         self.assertContains(response, "002")
         self.assertContains(response, "003")
         self.assertContains(response, "XXX")
-        self.assertContains(response, 2013)
-        self.assertContains(response, "XD")
-        self.assertContains(response, "10")
 
         parcels = Parcel.objects.all()
-        self.assertEqual(parcels.count(), 3)
+        self.assertEqual(parcels.count(), 2)
 
         deleted_parcel = Parcel.objects.filter(pk=parcel_1.pk)
         self.assertEqual(deleted_parcel.count(), 0)
 
+        data.update({
+            "_town": town_2.pk,
+            "_parcel_selection": "2013:XD:10 to 20, YD:24",
+        })
+
+        post_response = c.post("/file-parcels-modify/" + str(self.item.pk) + "/", data)
+        response = c.get(reverse("file-parcels-modify", kwargs={"pk": self.item.pk}))
+
+        self.assertContains(response, 2013)
+        self.assertContains(response, "XD")
+        self.assertContains(response, "YD")
+        self.assertContains(response, "24")
+        for i in range(10, 20):
+            self.assertContains(response, i)
+        parcels = Parcel.objects.all()
+        self.assertEqual(parcels.count(), 14)
+
 
     def testFileAssociation(self):
         # parcel association