Skip to content

Commit 7790df7

Browse files
authored
add test_create_default and test_update in test_operations.py (#71301)
1 parent ef54cd6 commit 7790df7

1 file changed

Lines changed: 22 additions & 0 deletions

File tree

airflow-ctl/tests/airflow_ctl/api/test_operations.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,7 @@
8686
PluginResponse,
8787
PoolBody,
8888
PoolCollectionResponse,
89+
PoolPatchBody,
8990
PoolResponse,
9091
ProviderCollectionResponse,
9192
ProviderResponse,
@@ -831,6 +832,16 @@ def handle_request(request: httpx.Request) -> httpx.Response:
831832
response = client.connections.bulk(connections=connection_bulk_body)
832833
assert response == self.connection_bulk_response
833834

835+
def test_create_defaults(self):
836+
def handle_request(request: httpx.Request) -> httpx.Response:
837+
assert request.url.path == "/api/v2/connections/defaults"
838+
assert request.method == "POST"
839+
return httpx.Response(200, json={})
840+
841+
client = make_api_client(transport=httpx.MockTransport(handle_request))
842+
response = client.connections.create_defaults()
843+
assert response is None
844+
834845
def test_delete(self):
835846
def handle_request(request: httpx.Request) -> httpx.Response:
836847
assert request.url.path == f"/api/v2/connections/{self.connection_id}"
@@ -1636,6 +1647,7 @@ class TestPoolsOperations:
16361647
pools=[pool_response],
16371648
total_entries=1,
16381649
)
1650+
pool_patch_body = PoolPatchBody(pool=pool_name, description="description")
16391651
pool_bulk_response = BulkResponse(
16401652
create=BulkActionResponse(success=[pool_name], errors=[]),
16411653
update=None,
@@ -1687,6 +1699,16 @@ def handle_request(request: httpx.Request) -> httpx.Response:
16871699
response = client.pools.delete(self.pool_name)
16881700
assert response == self.pool_name
16891701

1702+
def test_update(self):
1703+
def handle_request(request: httpx.Request) -> httpx.Response:
1704+
assert request.url.path == f"/api/v2/pools/{self.pool_name}"
1705+
assert request.method == "PATCH"
1706+
return httpx.Response(200, json=json.loads(self.pool_response.model_dump_json()))
1707+
1708+
client = make_api_client(transport=httpx.MockTransport(handle_request))
1709+
response = client.pools.update(pool_body=self.pool_patch_body)
1710+
assert response == self.pool_response
1711+
16901712

16911713
class TestProvidersOperations:
16921714
provider_response = ProviderResponse(

0 commit comments

Comments
 (0)