From 48f4dd13e3c70f17a550a1e00d35ad8561e927e5 Mon Sep 17 00:00:00 2001 From: LIU ZHE YOU <68415893+jason810496@users.noreply.github.com> Date: Tue, 20 May 2025 00:25:36 +0800 Subject: [PATCH] [v3-0-test] Fix SQLite Lock error in TestTaskInstancesLog setup (#50763) (cherry picked from commit 6f40f99543a33ca6bf6573fd98172354364113df) Co-authored-by: LIU ZHE YOU <68415893+jason810496@users.noreply.github.com> --- .../unit/api_fastapi/core_api/routes/public/test_log.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_log.py b/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_log.py index c2d51d03b0421..a906edc8b2d51 100644 --- a/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_log.py +++ b/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_log.py @@ -82,7 +82,8 @@ def add_one(x: int): ti.id = str(uuid7()) ti.hostname = "localhost" session.merge(ti) - session.flush() + # Commit changes to avoid locks + session.commit() # Add dummy dag for checking picking correct log with same task_id and different dag_id case. with dag_maker( @@ -106,8 +107,9 @@ def add_one(x: int): ti.id = str(uuid7()) ti.hostname = "localhost" session.merge(ti) - session.flush() - session.flush() + + # Final commit to ensure all changes are persisted + session.commit() dagbag = create_dag_bag() dagbag.bag_dag(dag)