diff --git a/tests/util/db_connection.py b/tests/util/db_connection.py index 55c9d53f72..4b36a811f2 100644 --- a/tests/util/db_connection.py +++ b/tests/util/db_connection.py @@ -10,12 +10,10 @@ from chia.util.db_wrapper import DBWrapper2 @asynccontextmanager async def DBConnection(db_version: int) -> AsyncIterator[DBWrapper2]: - db_path = Path(tempfile.NamedTemporaryFile().name) - if db_path.exists(): - db_path.unlink() - _db_wrapper = await DBWrapper2.create(database=db_path, reader_count=4, db_version=db_version) - try: - yield _db_wrapper - finally: - await _db_wrapper.close() - db_path.unlink() + with tempfile.TemporaryDirectory() as directory: + db_path = Path(directory).joinpath("db.sqlite") + _db_wrapper = await DBWrapper2.create(database=db_path, reader_count=4, db_version=db_version) + try: + yield _db_wrapper + finally: + await _db_wrapper.close()