✅ django 3.2: fix tests

1 files changed, 14 insertions, 0 deletions

diff --git a/ishtar_common/tests.py b/ishtar_common/tests.py
index 8fd5ac9e1..e92b98442 100644
--- a/ishtar_common/tests.py
+++ b/ishtar_common/tests.py
@@ -4190,8 +4190,21 @@ class GeoVectorTest(TestCase):
         self.assertEqual(self.base_find.main_geodata, geo_vector_find)  # no change
 
     def test_cascade_remove(self):
+        BaseFind = apps.get_model("archaeological_finds", "BaseFind")
+        ContextRecord = apps.get_model("archaeological_context_records",
+                                       "ContextRecord")
+
         geo_vector = self._create_geodata()
         self.operation.geodata.add(geo_vector)
+
+        self.assertEqual(self.context_record.geodata.count(), 1)
+        self.context_record = ContextRecord.objects.get(pk=self.context_record.pk)
+
+        self.assertEqual(self.context_record.main_geodata, geo_vector)
+        self.assertEqual(self.base_find.geodata.count(), 1)
+        self.base_find = BaseFind.objects.get(pk=self.base_find.pk)
+        self.assertEqual(self.base_find.main_geodata, geo_vector)
+
         geo_vector2 = self._create_geodata()
         self.operation.geodata.add(geo_vector2)
         geo_vector_find = self._create_geodata(alt=True)
@@ -4204,6 +4217,7 @@ class GeoVectorTest(TestCase):
         self.assertEqual(self.operation.geodata.count(), 1)
         self.assertEqual(self.context_record.main_geodata, geo_vector2)
         self.assertEqual(self.context_record.geodata.count(), 1)
+        self.base_find = BaseFind.objects.get(pk=self.base_find.pk)
         self.assertEqual(self.base_find.main_geodata, geo_vector2)
         self.assertEqual(self.base_find.geodata.count(), 2)