From 6af3244ceadede502b9ba4c8938e58a75f688aa6 Mon Sep 17 00:00:00 2001 From: Pratiksha Badheka Date: Sat, 8 Aug 2026 17:16:16 +0530 Subject: [PATCH] [airflow-ctl/v0-1-test] add test_create_default and test_update in test_operations.py (#71301) (cherry picked from commit 7790df7e2d4617865187b2f07632d00699a91f1e) Co-authored-by: Pratiksha Badheka --- .../tests/airflow_ctl/api/test_operations.py | 22 +++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/airflow-ctl/tests/airflow_ctl/api/test_operations.py b/airflow-ctl/tests/airflow_ctl/api/test_operations.py index bcef9acd74aff..d6a7975b7fc4a 100644 --- a/airflow-ctl/tests/airflow_ctl/api/test_operations.py +++ b/airflow-ctl/tests/airflow_ctl/api/test_operations.py @@ -85,6 +85,7 @@ PluginResponse, PoolBody, PoolCollectionResponse, + PoolPatchBody, PoolResponse, ProviderCollectionResponse, ProviderResponse, @@ -778,6 +779,16 @@ def handle_request(request: httpx.Request) -> httpx.Response: response = client.connections.bulk(connections=connection_bulk_body) assert response == self.connection_bulk_response + def test_create_defaults(self): + def handle_request(request: httpx.Request) -> httpx.Response: + assert request.url.path == "/api/v2/connections/defaults" + assert request.method == "POST" + return httpx.Response(200, json={}) + + client = make_api_client(transport=httpx.MockTransport(handle_request)) + response = client.connections.create_defaults() + assert response is None + def test_delete(self): def handle_request(request: httpx.Request) -> httpx.Response: assert request.url.path == f"/api/v2/connections/{self.connection_id}" @@ -1467,6 +1478,7 @@ class TestPoolsOperations: pools=[pool_response], total_entries=1, ) + pool_patch_body = PoolPatchBody(pool=pool_name, description="description") pool_bulk_response = BulkResponse( create=BulkActionResponse(success=[pool_name], errors=[]), update=None, @@ -1518,6 +1530,16 @@ def handle_request(request: httpx.Request) -> httpx.Response: response = client.pools.delete(self.pool_name) assert response == self.pool_name + def test_update(self): + def handle_request(request: httpx.Request) -> httpx.Response: + assert request.url.path == f"/api/v2/pools/{self.pool_name}" + assert request.method == "PATCH" + return httpx.Response(200, json=json.loads(self.pool_response.model_dump_json())) + + client = make_api_client(transport=httpx.MockTransport(handle_request)) + response = client.pools.update(pool_body=self.pool_patch_body) + assert response == self.pool_response + class TestProvidersOperations: provider_response = ProviderResponse(