From 1c015214f859bff65ba1d87981f266fd4840ffa4 Mon Sep 17 00:00:00 2001 From: pierrejeambrun Date: Tue, 28 Oct 2025 16:42:56 +0100 Subject: [PATCH] Add number of queries guard in public backfill list endpoints --- .../api_fastapi/core_api/routes/public/test_backfills.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_backfills.py b/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_backfills.py index 069208cd683cb..cf064e9bcdcf8 100644 --- a/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_backfills.py +++ b/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_backfills.py @@ -35,6 +35,7 @@ from airflow.utils.session import provide_session from airflow.utils.state import DagRunState +from tests_common.test_utils.asserts import assert_queries_count from tests_common.test_utils.db import ( clear_db_backfills, clear_db_dag_bundles, @@ -124,7 +125,10 @@ def test_list_backfill(self, test_client, session): b = Backfill(dag_id=dag.dag_id, from_date=from_date, to_date=to_date) session.add(b) session.commit() - response = test_client.get(f"/backfills?dag_id={dag.dag_id}") + + with assert_queries_count(2): + response = test_client.get(f"/backfills?dag_id={dag.dag_id}") + assert response.status_code == 200 assert response.json() == { "backfills": [