1
0
mirror of https://github.com/django/django.git synced 2025-10-31 09:41:08 +00:00

Used addCleanup() in tests where appropriate.

This commit is contained in:
Mariusz Felisiak
2023-12-31 10:01:31 +01:00
committed by GitHub
parent 81ccf92f15
commit d88ec42bd0
37 changed files with 121 additions and 240 deletions

View File

@@ -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)