diff --git a/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_dags.py b/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_dags.py index 94482a3b0aa96..cb59ec3ec4353 100644 --- a/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_dags.py +++ b/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_dags.py @@ -31,7 +31,7 @@ from airflow.utils.state import DagRunState, TaskInstanceState from airflow.utils.types import DagRunTriggeredByType, DagRunType -from tests_common.test_utils.asserts import count_queries +from tests_common.test_utils.asserts import assert_queries_count, count_queries from tests_common.test_utils.db import ( clear_db_assets, clear_db_connections, @@ -441,7 +441,8 @@ def test_get_dags(self, test_client, query_params, expected_total_entries, expec if any(param in query_params for param in ["has_asset_schedule", "asset_dependency"]): self._create_asset_test_data(session) - response = test_client.get("/dags", params=query_params) + with assert_queries_count(4): + response = test_client.get("/dags", params=query_params) assert response.status_code == 200 body = response.json()