@typespec/http-client-python 0.23.0 → 0.24.0

package/generator/test/azure/mock_api_tests/asynctests/test_service_multi_service_async.py

@@ -0,0 +1,31 @@
+# -------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+# --------------------------------------------------------------------------
+import pytest
+from azure.core.exceptions import HttpResponseError
+from service.multiservice.aio import CombinedClient
+from service.multiservice.models import VersionsA, VersionsB
+
+
+@pytest.fixture
+def client():
+    """Fixture that creates a CombinedClient for testing."""
+    return CombinedClient(endpoint="http://localhost:3000")
+
+
+@pytest.mark.asyncio
+async def test_service_multi_service_foo(client):
+    with pytest.raises(HttpResponseError):
+        await client.foo.test(api_version=VersionsA.AV1)
+
+    await client.foo.test()
+
+
+@pytest.mark.asyncio
+async def test_service_multi_service_bar(client):
+    with pytest.raises(HttpResponseError):
+        await client.bar.test(api_version=VersionsB.BV1)
+
+    await client.bar.test()

package/generator/test/azure/mock_api_tests/asynctests/test_special_words_async.py

@@ -40,3 +40,21 @@ async def test_model(client: SpecialWordsClient, special_words):
 @pytest.mark.asyncio
 async def test_model_properties(client: SpecialWordsClient):
     await client.model_properties.same_as_model(models.SameAsModel(same_as_model="ok"))
+
+
+@pytest.mark.asyncio
+async def test_model_properties_dict_methods(client: SpecialWordsClient):
+    await client.model_properties.dict_methods(
+        body=models.DictMethods(
+            keys_property="ok",
+            items_property="ok",
+            values_property="ok",
+            popitem_property="ok",
+            clear_property="ok",
+            update_property="ok",
+            setdefault_property="ok",
+            pop_property="ok",
+            get_property="ok",
+            copy_property="ok",
+        )
+    )

package/generator/test/azure/mock_api_tests/test_azure_arm_operationtemplates.py

@@ -154,3 +154,19 @@ def test_optional_body_provider_post_with_body(client):
     )
     assert result.total_allowed == 100
     assert result.status == "Changed to requested allowance"
+
+
+# enable the test case after https://github.com/Azure/typespec-azure/pull/3695 merged and released
+# def test_lro_paging_begin_post_paging_lro(client):
+#     result = client.lro_paging.begin_post_paging_lro(
+#         resource_group_name=RESOURCE_GROUP_NAME,
+#         product_name="default",
+#     ).result()
+#     items = list(result)
+#     assert len(items) == 2
+#     assert items[0].name == "product1"
+#     assert items[0].properties.product_id == "product1"
+#     assert items[0].properties.provisioning_state == "Succeeded"
+#     assert items[1].name == "product2"
+#     assert items[1].properties.product_id == "product2"
+#     assert items[1].properties.provisioning_state == "Succeeded"

package/generator/test/azure/mock_api_tests/test_azure_client_generator_core_client_default_value.py

@@ -0,0 +1,42 @@
+# -------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+# --------------------------------------------------------------------------
+import pytest
+from specs.azure.clientgenerator.core.clientdefaultvalue import ClientDefaultValueClient
+from specs.azure.clientgenerator.core.clientdefaultvalue.models import ModelWithDefaultValues
+
+
+@pytest.fixture
+def client():
+    with ClientDefaultValueClient() as client:
+        yield client
+
+
+def test_put_model_property(client: ClientDefaultValueClient):
+    """Test case 1: @clientDefaultValue for model property."""
+    body = ModelWithDefaultValues(name="test")
+    result = client.put_model_property(body=body)
+    assert result.name == "test"
+    assert result.timeout == 30
+    assert result.tier == "standard"
+    assert result.retry is True
+
+
+def test_get_operation_parameter(client: ClientDefaultValueClient):
+    """Test case 2: @clientDefaultValue for operation parameter."""
+    # Test with only required parameter (name), defaults should be applied
+    client.get_operation_parameter(name="test")
+
+
+def test_get_path_parameter(client: ClientDefaultValueClient):
+    """Test case 3: @clientDefaultValue for first path segment."""
+    # Test with only required segment2, segment1 should use default
+    client.get_path_parameter(segment2="segment2")
+
+
+def test_get_header_parameter(client: ClientDefaultValueClient):
+    """Test case 4: @clientDefaultValue for header parameters."""
+    # Test with default header values
+    client.get_header_parameter()

package/generator/test/azure/mock_api_tests/test_azure_client_generator_core_client_location.py

@@ -4,42 +4,67 @@
 # license information.
 # --------------------------------------------------------------------------
 import pytest
-from specs.azure.clientgenerator.core.clientlocation import ClientLocationClient
+from specs.azure.clientgenerator.core.clientlocation.parameter import MoveMethodParameterToClient
+from specs.azure.clientgenerator.core.clientlocation.subclient import MoveToExistingSubClient
+from specs.azure.clientgenerator.core.clientlocation.newsubclient import MoveToNewSubClient
+from specs.azure.clientgenerator.core.clientlocation.rootclient import MoveToRootClient
 
 
 @pytest.fixture
-def client():
-    with ClientLocationClient(storage_account="testaccount") as client:
+def move_method_parameter_to_client():
+    with MoveMethodParameterToClient(storage_account="testaccount") as client:
         yield client
 
 
-def test_get_health_status(client: ClientLocationClient):
-    client.get_health_status()
+@pytest.fixture
+def move_to_existing_sub_client():
+    with MoveToExistingSubClient() as client:
+        yield client
+
+
+@pytest.fixture
+def move_to_new_sub_client():
+    with MoveToNewSubClient() as client:
+        yield client
+
+
+@pytest.fixture
+def move_to_root_client():
+    with MoveToRootClient() as client:
+        yield client
+
+
+def test_move_method_parameter_to_client_blob_operations_get_blob(
+    move_method_parameter_to_client: MoveMethodParameterToClient,
+):
+    move_method_parameter_to_client.blob_operations.get_blob(container="testcontainer", blob="testblob.txt")
 
 
-def test_archive_operations_archive_product(client: ClientLocationClient):
-    client.archive_operations.archive_product()
+def test_move_to_existing_sub_client_user_operations_get_user(move_to_existing_sub_client: MoveToExistingSubClient):
+    move_to_existing_sub_client.user_operations.get_user()
 
 
-def test_move_to_existing_sub_client_admin_operations_get_admin_info(client: ClientLocationClient):
-    client.move_to_existing_sub_client.admin_operations.get_admin_info()
+def test_move_to_existing_sub_client_admin_operations_delete_user(move_to_existing_sub_client: MoveToExistingSubClient):
+    move_to_existing_sub_client.admin_operations.delete_user()
 
 
-def test_move_to_existing_sub_client_admin_operations_delete_user(client: ClientLocationClient):
-    client.move_to_existing_sub_client.admin_operations.delete_user()
+def test_move_to_existing_sub_client_admin_operations_get_admin_info(
+    move_to_existing_sub_client: MoveToExistingSubClient,
+):
+    move_to_existing_sub_client.admin_operations.get_admin_info()
 
 
-def test_move_to_existing_sub_client_user_operations_get_user(client: ClientLocationClient):
-    client.move_to_existing_sub_client.user_operations.get_user()
+def test_move_to_new_sub_client_product_operations_list_products(move_to_new_sub_client: MoveToNewSubClient):
+    move_to_new_sub_client.product_operations.list_products()
 
 
-def test_move_to_new_sub_client_product_operations_list_products(client: ClientLocationClient):
-    client.move_to_new_sub_client.product_operations.list_products()
+def test_move_to_new_sub_client_archive_operations_archive_product(move_to_new_sub_client: MoveToNewSubClient):
+    move_to_new_sub_client.archive_operations.archive_product()
 
 
-def test_move_to_root_client_resource_operations_get_resource(client: ClientLocationClient):
-    client.move_to_root_client.resource_operations.get_resource()
+def test_move_to_root_client_resource_operations_get_resource(move_to_root_client: MoveToRootClient):
+    move_to_root_client.resource_operations.get_resource()
 
 
-def test_move_method_parameter_to_client_blob_operations_get_blob(client: ClientLocationClient):
-    client.move_method_parameter_to_client.blob_operations.get_blob(container="testcontainer", blob="testblob.txt")
+def test_move_to_root_client_get_health_status(move_to_root_client: MoveToRootClient):
+    move_to_root_client.get_health_status()

package/generator/test/azure/mock_api_tests/test_azure_resource_manager_multi_service.py

@@ -0,0 +1,104 @@
+# -------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+# --------------------------------------------------------------------------
+import pytest
+from azure.core.exceptions import HttpResponseError
+from azure.resourcemanager.multiservice.combined import CombinedClient
+from azure.resourcemanager.multiservice.combined.models import VirtualMachine, Disk
+
+
+@pytest.fixture
+def client(credential, authentication_policy):
+    """Create a Combined client for testing."""
+    return CombinedClient(
+        credential=credential,  # Will use mock server, no real auth needed
+        subscription_id="00000000-0000-0000-0000-000000000000",
+        base_url="http://localhost:3000",
+        authentication_policy=authentication_policy,
+        polling_interval=0.1,  # Speed up tests by reducing polling interval
+    )
+
+
+def test_virtual_machines_get(client):
+    resource_group_name = "test-rg"
+    vm_name = "vm1"
+
+    with pytest.raises(HttpResponseError):
+        client.virtual_machines.get(
+            resource_group_name=resource_group_name,
+            vm_name=vm_name,
+            api_version="av1",  # invalid api version shall raise error
+        )
+
+    result = client.virtual_machines.get(resource_group_name=resource_group_name, vm_name=vm_name)
+
+    assert result is not None
+    assert isinstance(result, VirtualMachine)
+    assert result.name == vm_name
+
+
+def test_virtual_machines_create_or_update(client):
+    resource_group_name = "test-rg"
+    vm_name = "vm1"
+
+    vm_resource = VirtualMachine(location="eastus", properties={})
+
+    with pytest.raises(HttpResponseError):
+        client.virtual_machines.begin_create_or_update(
+            resource_group_name=resource_group_name,
+            vm_name=vm_name,
+            resource=vm_resource,
+            api_version="av1",  # invalid api version shall raise error
+        ).result()
+
+    poller = client.virtual_machines.begin_create_or_update(
+        resource_group_name=resource_group_name, vm_name=vm_name, resource=vm_resource
+    )
+
+    result = poller.result()
+    assert result is not None
+    assert isinstance(result, VirtualMachine)
+    assert result.location == "eastus"
+
+
+def test_disks_get(client):
+    resource_group_name = "test-rg"
+    disk_name = "disk1"
+    with pytest.raises(HttpResponseError):
+        client.disks.get(
+            resource_group_name=resource_group_name,
+            disk_name=disk_name,
+            api_version="av1",  # invalid api version shall raise error
+        )
+
+    result = client.disks.get(resource_group_name=resource_group_name, disk_name=disk_name)
+
+    assert result is not None
+    assert isinstance(result, Disk)
+    assert result.name == disk_name
+
+
+def test_disks_create_or_update(client):
+    resource_group_name = "test-rg"
+    disk_name = "disk1"
+
+    disk_resource = Disk(location="eastus", properties={})
+
+    with pytest.raises(HttpResponseError):
+        client.disks.begin_create_or_update(
+            resource_group_name=resource_group_name,
+            disk_name=disk_name,
+            resource=disk_resource,
+            api_version="av1",  # invalid api version shall raise error
+        ).result()
+
+    poller = client.disks.begin_create_or_update(
+        resource_group_name=resource_group_name, disk_name=disk_name, resource=disk_resource
+    )
+
+    result = poller.result()
+    assert result is not None
+    assert isinstance(result, Disk)
+    assert result.location == "eastus"

package/generator/test/azure/mock_api_tests/test_service_multi_service.py

@@ -0,0 +1,29 @@
+# -------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+# --------------------------------------------------------------------------
+import pytest
+from azure.core.exceptions import HttpResponseError
+from service.multiservice import CombinedClient
+from service.multiservice.models import VersionsA, VersionsB
+
+
+@pytest.fixture
+def client():
+    """Fixture that creates a CombinedClient for testing."""
+    return CombinedClient(endpoint="http://localhost:3000")
+
+
+def test_service_multi_service_foo(client):
+    with pytest.raises(HttpResponseError):
+        client.foo.test(api_version=VersionsA.AV1)
+
+    client.foo.test()
+
+
+def test_service_multi_service_bar(client):
+    with pytest.raises(HttpResponseError):
+        client.bar.test(api_version=VersionsB.BV1)
+
+    client.bar.test()

package/generator/test/azure/mock_api_tests/test_special_words.py

@@ -35,3 +35,20 @@ def test_model(client: SpecialWordsClient, special_words):
 
 def test_model_properties(client: SpecialWordsClient):
     client.model_properties.same_as_model(models.SameAsModel(same_as_model="ok"))
+
+
+def test_model_properties_dict_methods(client: SpecialWordsClient):
+    client.model_properties.dict_methods(
+        body=models.DictMethods(
+            keys_property="ok",
+            items_property="ok",
+            values_property="ok",
+            popitem_property="ok",
+            clear_property="ok",
+            update_property="ok",
+            setdefault_property="ok",
+            pop_property="ok",
+            get_property="ok",
+            copy_property="ok",
+        )
+    )

package/generator/test/azure/requirements.txt

@@ -13,7 +13,11 @@ azure-mgmt-core==1.6.0
 -e ./generated/azure-client-generator-core-hierarchy-building
 -e ./generated/azure-client-generator-core-usage
 -e ./generated/azure-client-generator-core-override
--e ./generated/azure-client-generator-core-client-location
+-e ./generated/azure-client-generator-core-client-location-move-method-parameter-to-client
+-e ./generated/azure-client-generator-core-client-location-move-to-existing-sub-client
+-e ./generated/azure-client-generator-core-client-location-move-to-new-sub-client
+-e ./generated/azure-client-generator-core-client-location-move-to-root-client
+-e ./generated/azure-client-generator-core-client-default-value
 -e ./generated/azure-client-generator-core-alternate-type
 -e ./generated/azure-client-generator-core-next-link-verb
 -e ./generated/azure-core-basic
@@ -32,6 +36,7 @@ azure-mgmt-core==1.6.0
 -e ./generated/azure-resource-manager-operation-templates
 -e ./generated/azure-resource-manager-resources
 -e ./generated/azure-resource-manager-method-subscription-id
+-e ./generated/azure-resource-manager-multi-service
 -e ./generated/azure-versioning-previewversion
 -e ./generated/client-namespace
 -e ./generated/azure-payload-pageable
@@ -45,6 +50,7 @@ azure-mgmt-core==1.6.0
 -e ./generated/client-structure-twooperationgroup
 -e ./generated/resiliency-srv-driven1
 -e ./generated/resiliency-srv-driven2
+-e ./generated/service-multi-service
 
 # common test case
 -e ./generated/authentication-api-key
@@ -59,6 +65,7 @@ azure-mgmt-core==1.6.0
 -e ./generated/parameters-basic
 -e ./generated/parameters-collection-format
 -e ./generated/parameters-path
+-e ./generated/parameters-query
 -e ./generated/parameters-spread
 -e ./generated/serialization-encoded-name-json
 -e ./generated/server-endpoint-not-defined
@@ -85,6 +92,7 @@ azure-mgmt-core==1.6.0
 -e ./generated/typetest-property-additionalproperties
 -e ./generated/typetest-scalar
 -e ./generated/typetest-union
+-e ./generated/typetest-discriminatedunion
 -e ./generated/typetest-model-empty
 -e ./generated/headasbooleantrue
 -e ./generated/headasbooleanfalse

package/generator/test/generic_mock_api_tests/asynctests/test_parameters_query_async.py

@@ -0,0 +1,18 @@
+# -------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+# --------------------------------------------------------------------------
+import pytest
+from parameters.query.aio import QueryClient
+
+
+@pytest.fixture
+async def client():
+    async with QueryClient() as client:
+        yield client
+
+
+@pytest.mark.asyncio
+async def test_constant(client: QueryClient):
+    await client.constant.post()

package/generator/test/generic_mock_api_tests/test_parameters_query.py

@@ -0,0 +1,17 @@
+# -------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+# --------------------------------------------------------------------------
+import pytest
+from parameters.query import QueryClient
+
+
+@pytest.fixture
+def client():
+    with QueryClient() as client:
+        yield client
+
+
+def test_constant(client: QueryClient):
+    client.constant.post()

package/generator/test/generic_mock_api_tests/test_typetest_scalar.py

@@ -51,8 +51,3 @@ def test_decimal128_verify(client: ScalarClient):
 def test_decimal_verify(client: ScalarClient):
     prepare = client.decimal_verify.prepare_verify()
     client.decimal_verify.verify(reduce(lambda x, y: x + y, prepare))
-
-
-# open this test case after adopt new version of http-specs for https://github.com/microsoft/typespec/pull/8807
-# def test_constant_query(client: ScalarClient):
-#     client.constant_query.post()