test: Ensure that database connection is torn down in copydb tests

pmisik · Nov 4, 2024 · fa4b310 · fa4b310
1 parent d394b2e
commit fa4b310
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/master/buildbot/test/unit/scripts/test_copydb.py b/master/buildbot/test/unit/scripts/test_copydb.py
@@ -161,8 +161,8 @@ def resolve_db_url(self):
 
     def drop_database_tables(self, db_url):
         engine = enginestrategy.create_engine(db_url, basedir='basedir')
-        conn = engine.connect()
-        db.thd_clean_database(conn)
+        with engine.connect() as conn:
+            db.thd_clean_database(conn)
 
     @async_to_deferred
     async def test_full(self):