Operations: fix deletion of sym-linked operations

1 files changed, 10 insertions, 2 deletions

diff --git a/archaeological_operations/tests.py b/archaeological_operations/tests.py
index d8399eedb..86eb2dfe7 100644
--- a/archaeological_operations/tests.py
+++ b/archaeological_operations/tests.py
@@ -578,8 +578,7 @@ class OperationTest(TestCase, OperationInitTest):
                          {'operator': self.orgas[0].pk})
         self.assertTrue(json.loads(response.content)['total'] == 2)
 
-    def testRelatedSearch(self):
-        c = Client()
+    def create_relations(self):
         rel1 = models.RelationType.objects.create(
             symmetrical=True, label='Include', txt_idx='include')
         rel2 = models.RelationType.objects.create(
@@ -589,6 +588,11 @@ class OperationTest(TestCase, OperationInitTest):
             left_record=self.operations[0],
             right_record=self.operations[1],
             relation_type=rel1)
+        return rel1, rel2
+
+    def testRelatedSearch(self):
+        c = Client()
+        rel1, rel2 = self.create_relations()
         self.operations[1].year = 2011
         self.operations[1].save()
         search = {'year': '2010', 'relation_types_0': rel2.pk}
@@ -599,6 +603,10 @@ class OperationTest(TestCase, OperationInitTest):
         response = c.get(reverse('get-operation'), search)
         self.assertTrue(json.loads(response.content)['total'] == 2)
 
+    def testRelations(self):
+        self.create_relations()
+        self.operations[0].delete()
+
     def testOwnSearch(self):
         c = Client()
         response = c.get(reverse('get-operation'), {'year': '2010'})