Used addCleanup() in tests where appropriate.

1 files changed, 2 insertions, 4 deletions

diff --git a/tests/delete_regress/tests.py b/tests/delete_regress/tests.py
index 71e3bcb405..89f4d5ddd8 100644
--- a/tests/delete_regress/tests.py
+++ b/tests/delete_regress/tests.py
@@ -51,11 +51,9 @@ class DeleteLockingTest(TransactionTestCase):
         # Create a second connection to the default database
         self.conn2 = connection.copy()
         self.conn2.set_autocommit(False)
-
-    def tearDown(self):
         # Close down the second connection.
-        self.conn2.rollback()
-        self.conn2.close()
+        self.addCleanup(self.conn2.close)
+        self.addCleanup(self.conn2.rollback)
 
     def test_concurrent_delete(self):
         """Concurrent deletes don't collide and lock the database (#9479)."""