diff options
Diffstat (limited to 'ishtar_common/tests.py')
| -rw-r--r-- | ishtar_common/tests.py | 14 |
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) |