Used addCleanup() in tests where appropriate.

1 files changed, 6 insertions, 5 deletions

diff --git a/tests/defer_regress/tests.py b/tests/defer_regress/tests.py
index 3dfe96ddb3..10100e348d 100644
--- a/tests/defer_regress/tests.py
+++ b/tests/defer_regress/tests.py
@@ -322,12 +322,13 @@ class DeferDeletionSignalsTests(TestCase):
         self.post_delete_senders = []
         for sender in self.senders:
             models.signals.pre_delete.connect(self.pre_delete_receiver, sender)
+            self.addCleanup(
+                models.signals.pre_delete.disconnect, self.pre_delete_receiver, sender
+            )
             models.signals.post_delete.connect(self.post_delete_receiver, sender)
-
-    def tearDown(self):
-        for sender in self.senders:
-            models.signals.pre_delete.disconnect(self.pre_delete_receiver, sender)
-            models.signals.post_delete.disconnect(self.post_delete_receiver, sender)
+            self.addCleanup(
+                models.signals.post_delete.disconnect, self.post_delete_receiver, sender
+            )
 
     def pre_delete_receiver(self, sender, **kwargs):
         self.pre_delete_senders.append(sender)