diff --git a/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_variables.py b/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_variables.py index 399c8a6976f69..ffd093f3798fa 100644 --- a/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_variables.py +++ b/airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_variables.py @@ -25,6 +25,7 @@ from airflow.models.variable import Variable from airflow.utils.session import provide_session +from tests_common.test_utils.asserts import assert_queries_count from tests_common.test_utils.db import clear_db_variables from tests_common.test_utils.logs import check_last_log @@ -293,7 +294,8 @@ def test_should_respond_200( self, session, test_client, query_params, expected_total_entries, expected_keys ): self.create_variables() - response = test_client.get("/variables", params=query_params) + with assert_queries_count(3): + response = test_client.get("/variables", params=query_params) assert response.status_code == 200 body = response.json()