|
86 | 86 | PluginResponse, |
87 | 87 | PoolBody, |
88 | 88 | PoolCollectionResponse, |
| 89 | + PoolPatchBody, |
89 | 90 | PoolResponse, |
90 | 91 | ProviderCollectionResponse, |
91 | 92 | ProviderResponse, |
@@ -831,6 +832,16 @@ def handle_request(request: httpx.Request) -> httpx.Response: |
831 | 832 | response = client.connections.bulk(connections=connection_bulk_body) |
832 | 833 | assert response == self.connection_bulk_response |
833 | 834 |
|
| 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 | + |
834 | 845 | def test_delete(self): |
835 | 846 | def handle_request(request: httpx.Request) -> httpx.Response: |
836 | 847 | assert request.url.path == f"/api/v2/connections/{self.connection_id}" |
@@ -1636,6 +1647,7 @@ class TestPoolsOperations: |
1636 | 1647 | pools=[pool_response], |
1637 | 1648 | total_entries=1, |
1638 | 1649 | ) |
| 1650 | + pool_patch_body = PoolPatchBody(pool=pool_name, description="description") |
1639 | 1651 | pool_bulk_response = BulkResponse( |
1640 | 1652 | create=BulkActionResponse(success=[pool_name], errors=[]), |
1641 | 1653 | update=None, |
@@ -1687,6 +1699,16 @@ def handle_request(request: httpx.Request) -> httpx.Response: |
1687 | 1699 | response = client.pools.delete(self.pool_name) |
1688 | 1700 | assert response == self.pool_name |
1689 | 1701 |
|
| 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 | + |
1690 | 1712 |
|
1691 | 1713 | class TestProvidersOperations: |
1692 | 1714 | provider_response = ProviderResponse( |
|
0 commit comments