From 25db53e042daffdc346e3393a90a79abdc9e3c51 Mon Sep 17 00:00:00 2001
From: pratiksha badheka <pratikshabadheka27@gmail.com>
Date: Fri, 7 Aug 2026 09:45:45 -0700
Subject: [PATCH] add test_create_default and test_update in test_operations.py

---
 .../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 952068c86aa25..1bc183d4f4592 100644
--- a/airflow-ctl/tests/airflow_ctl/api/test_operations.py
+++ b/airflow-ctl/tests/airflow_ctl/api/test_operations.py
@@ -86,6 +86,7 @@
     PluginResponse,
     PoolBody,
     PoolCollectionResponse,
+    PoolPatchBody,
     PoolResponse,
     ProviderCollectionResponse,
     ProviderResponse,
@@ -831,6 +832,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}"
@@ -1636,6 +1647,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,
@@ -1687,6 +1699,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(