Context Record, Bulk update : draft of tests #5293

1 files changed, 50 insertions, 0 deletions

diff --git a/archaeological_context_records/tests.py b/archaeological_context_records/tests.py
index c9171f828..4e98ec941 100644
--- a/archaeological_context_records/tests.py
+++ b/archaeological_context_records/tests.py
@@ -569,6 +569,56 @@ class ContextRecordTest(ContextRecordInit, TestCase):
         self.assertEqual(cr3.custom_index, 1)
 
 
+class ContextRecordQATest(ContextRecordInit, TestCase):
+    fixtures = CONTEXT_RECORD_TOWNS_FIXTURES
+
+    def setUp(self):
+        self.username, self.password, self.user = create_superuser()
+
+        self.create_context_record(data={"label": "CR 1"})
+        self.create_context_record(data={"label": "CR 2"})
+        self.rel_types = models.RelationType.objects.all()
+        self.units = models.Unit.objects.all()
+
+    def test_bulk_update(self):
+        c = Client()
+        pks = "{}-{}".format(self.context_records[0].pk, self.context_records[1].pk)
+        print(models.Parcel.objects.all())
+        print(models.Unit.objects.all())
+        print(models.Operation.objects.all())
+        c.login(username=self.username, password=self.password)
+        print(pks)
+        response = c.get(reverse("contextrecord-qa-bulk-update", args=[pks]))
+        print(response)
+        self.assertEqual(response.status_code, 200)
+
+        cr_1 = self.context_records[0]
+        cr_2 = self.context_records[1]
+        unit = models.Unit.objects.all().first()
+        sym_rel_type = models.RelationType.objects.create(
+            symmetrical=True, txt_idx="sym"
+        )
+        rel_type_1 = models.RelationType.objects.create(
+            symmetrical=False, txt_idx="rel_1"
+        )
+
+        data = {
+            "qa_unit": unit.pk,
+            "qa_relation_type": sym_rel_type,
+        }
+
+        response = c.post(reverse(
+            "contextrecord-qa-bulk-update", args=[pks]), data
+        )
+
+        if response.status_code != 200:
+            self.assertRedirects(response, "/success/")
+
+        crs = models.ContextRecord.objects.all()
+        for cr in crs:
+            self.assertEqual(cr.unit, su)
+
+
 class ContextRecordSearchTest(ContextRecordInit, TestCase, SearchText):
     fixtures = CONTEXT_RECORD_TOWNS_FIXTURES
     SEARCH_URL = "get-contextrecord"