Move signal disconnect into finally block

1 files changed, 6 insertions, 4 deletions

diff --git a/tests/modeltests/delete/tests.py b/tests/modeltests/delete/tests.py
index 5d7b7a0b33..99981df018 100644
--- a/tests/modeltests/delete/tests.py
+++ b/tests/modeltests/delete/tests.py
@@ -233,13 +233,15 @@ class DeletionTests(TestCase):
         def log_post_delete(instance, **kwargs):
             self.assertTrue(R.objects.filter(pk=instance.r_id))
 
-        models.signals.post_delete.connect(log_post_delete, sender=S)
-
         r = R.objects.create(pk=1)
         S.objects.create(pk=1, r=r)
-        r.delete()
 
-        models.signals.post_delete.disconnect(log_post_delete)
+        models.signals.post_delete.connect(log_post_delete, sender=S)
+
+        try:
+            r.delete()
+        finally:
+            models.signals.post_delete.disconnect(log_post_delete)
 
     @skipUnlessDBFeature("can_defer_constraint_checks")
     def test_can_defer_constraint_checks(self):