Skip to content

Commit df811c7

Browse files
[v3-1-test] Add number of queries guard in public import jobs endpoints (#57559) (#57611)
(cherry picked from commit 96ea6a6) Co-authored-by: Pierre Jeambrun <[email protected]>
1 parent 87fd560 commit df811c7

File tree

1 file changed

+3
-1
lines changed
  • airflow-core/tests/unit/api_fastapi/core_api/routes/public

1 file changed

+3
-1
lines changed

airflow-core/tests/unit/api_fastapi/core_api/routes/public/test_job.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
from airflow.utils.session import provide_session
2626
from airflow.utils.state import State
2727

28+
from tests_common.test_utils.asserts import assert_queries_count
2829
from tests_common.test_utils.db import clear_db_jobs
2930
from tests_common.test_utils.format_datetime import from_datetime_to_zulu
3031

@@ -142,7 +143,8 @@ def test_get_jobs(
142143
):
143144
# setup testcase at runtime based on the `testcase` parameter
144145
self.setup(testcase)
145-
response = test_client.get("/jobs", params=query_params)
146+
with assert_queries_count(2):
147+
response = test_client.get("/jobs", params=query_params)
146148
assert response.status_code == expected_status_code
147149
if expected_status_code != 200:
148150
return

0 commit comments

Comments
 (0)