@typespec/http-client-python 0.23.1 → 0.24.1

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_payload_multipart.py

@@ -138,3 +138,34 @@ def test_complex_with_http_part(client: MultiPartClient):
 
 def test_http_parts_non_string_float(client: MultiPartClient):
     client.form_data.http_parts.non_string.float(models.FloatRequest(temperature=0.5))
+
+
+def test_with_wire_name(client: MultiPartClient):
+    client.form_data.with_wire_name(
+        models.MultiPartRequestWithWireName(
+            identifier="123",
+            image=open(str(JPG), "rb"),
+        )
+    )
+
+
+def test_optional_parts(client: MultiPartClient):
+    # First time with only id
+    client.form_data.optional_parts(
+        models.MultiPartOptionalRequest(
+            id="123",
+        )
+    )
+    # Second time with only profileImage
+    client.form_data.optional_parts(
+        models.MultiPartOptionalRequest(
+            profile_image=open(str(JPG), "rb"),
+        )
+    )
+    # Third time with both id and profileImage
+    client.form_data.optional_parts(
+        models.MultiPartOptionalRequest(
+            id="123",
+            profile_image=open(str(JPG), "rb"),
+        )
+    )

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,10 +50,12 @@ 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
 -e ./generated/authentication-http-custom
+-e ./generated/authentication-noauth-union
 -e ./generated/authentication-oauth2
 -e ./generated/authentication-union
 -e ./generated/setuppy-authentication-union
@@ -59,6 +66,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 +93,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()

package/generator/test/generic_mock_api_tests/test_typetest_union_discriminated.py

@@ -0,0 +1,290 @@
+# -------------------------------------------------------------------------
+# 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 typetest.discriminatedunion import DiscriminatedClient
+from typetest.discriminatedunion import models
+
+
+@pytest.fixture
+def client():
+    with DiscriminatedClient() as client:
+        yield client
+
+
+@pytest.fixture
+def cat_body():
+    """Cat model for testing."""
+    return models.Cat(name="Whiskers", meow=True)
+
+
+@pytest.fixture
+def dog_body():
+    """Dog model for testing."""
+    return models.Dog(name="Rex", bark=False)
+
+
+# Tests for No Envelope / Default (inline discriminator with "kind")
+@pytest.mark.skip(reason="After completely support discriminated unions, enable these tests")
+class TestNoEnvelopeDefault:
+    """Test discriminated union with inline discriminator (no envelope)."""
+
+    def test_get_default_cat(self, client: DiscriminatedClient, cat_body: models.Cat):
+        """Test getting cat with default (no query param or kind=cat).
+
+        Expected response:
+        {
+            "kind": "cat",
+            "name": "Whiskers",
+            "meow": true
+        }
+        """
+        result = client.no_envelope.default.get()
+        assert result == cat_body
+        assert isinstance(result, models.Cat)
+
+    def test_get_with_kind_cat(self, client: DiscriminatedClient, cat_body: models.Cat):
+        """Test getting cat with kind=cat query parameter.
+
+        Expected response:
+        {
+            "kind": "cat",
+            "name": "Whiskers",
+            "meow": true
+        }
+        """
+        result = client.no_envelope.default.get(kind="cat")
+        assert result == cat_body
+        assert isinstance(result, models.Cat)
+
+    def test_get_with_kind_dog(self, client: DiscriminatedClient, dog_body: models.Dog):
+        """Test getting dog with kind=dog query parameter.
+
+        Expected response:
+        {
+            "kind": "dog",
+            "name": "Rex",
+            "bark": false
+        }
+        """
+        result = client.no_envelope.default.get(kind="dog")
+        assert result == dog_body
+        assert isinstance(result, models.Dog)
+
+    def test_put_cat(self, client: DiscriminatedClient, cat_body: models.Cat):
+        """Test sending cat with inline discriminator.
+
+        Expected request:
+        {
+            "kind": "cat",
+            "name": "Whiskers",
+            "meow": true
+        }
+        """
+        result = client.no_envelope.default.put(cat_body)
+        assert result == cat_body
+        assert isinstance(result, models.Cat)
+
+
+# Tests for No Envelope / Custom Discriminator (inline with custom "type" property)
+@pytest.mark.skip(reason="After completely support discriminated unions, enable these tests")
+class TestNoEnvelopeCustomDiscriminator:
+    """Test discriminated union with inline discriminator and custom discriminator property name."""
+
+    def test_get_default_cat(self, client: DiscriminatedClient, cat_body: models.Cat):
+        """Test getting cat with default (no query param or type=cat).
+
+        Expected response:
+        {
+            "type": "cat",
+            "name": "Whiskers",
+            "meow": true
+        }
+        """
+        result = client.no_envelope.custom_discriminator.get()
+        assert result == cat_body
+        assert isinstance(result, models.Cat)
+
+    def test_get_with_type_cat(self, client: DiscriminatedClient, cat_body: models.Cat):
+        """Test getting cat with type=cat query parameter.
+
+        Expected response:
+        {
+            "type": "cat",
+            "name": "Whiskers",
+            "meow": true
+        }
+        """
+        result = client.no_envelope.custom_discriminator.get(type="cat")
+        assert result == cat_body
+        assert isinstance(result, models.Cat)
+
+    def test_get_with_type_dog(self, client: DiscriminatedClient, dog_body: models.Dog):
+        """Test getting dog with type=dog query parameter.
+
+        Expected response:
+        {
+            "type": "dog",
+            "name": "Rex",
+            "bark": false
+        }
+        """
+        result = client.no_envelope.custom_discriminator.get(type="dog")
+        assert result == dog_body
+        assert isinstance(result, models.Dog)
+
+    def test_put_cat(self, client: DiscriminatedClient, cat_body: models.Cat):
+        """Test sending cat with inline custom discriminator.
+
+        Expected request:
+        {
+            "type": "cat",
+            "name": "Whiskers",
+            "meow": true
+        }
+        """
+        result = client.no_envelope.custom_discriminator.put(cat_body)
+        assert result == cat_body
+        assert isinstance(result, models.Cat)
+
+
+# Tests for Envelope / Object / Default (envelope with "kind" and "value")
+@pytest.mark.skip(reason="After completely support discriminated unions, enable these tests")
+class TestEnvelopeObjectDefault:
+    """Test discriminated union with default envelope serialization."""
+
+    def test_get_default_cat(self, client: DiscriminatedClient, cat_body: models.Cat):
+        """Test getting cat with default (no query param or kind=cat).
+
+        Expected response:
+        {
+            "kind": "cat",
+            "value": {
+                "name": "Whiskers",
+                "meow": true
+            }
+        }
+        """
+        result = client.envelope.object.default.get()
+        assert result == cat_body
+        assert isinstance(result, models.Cat)
+
+    def test_get_with_kind_cat(self, client: DiscriminatedClient, cat_body: models.Cat):
+        """Test getting cat with kind=cat query parameter.
+
+        Expected response:
+        {
+            "kind": "cat",
+            "value": {
+                "name": "Whiskers",
+                "meow": true
+            }
+        }
+        """
+        result = client.envelope.object.default.get(kind="cat")
+        assert result == cat_body
+        assert isinstance(result, models.Cat)
+
+    def test_get_with_kind_dog(self, client: DiscriminatedClient, dog_body: models.Dog):
+        """Test getting dog with kind=dog query parameter.
+
+        Expected response:
+        {
+            "kind": "dog",
+            "value": {
+                "name": "Rex",
+                "bark": false
+            }
+        }
+        """
+        result = client.envelope.object.default.get(kind="dog")
+        assert result == dog_body
+        assert isinstance(result, models.Dog)
+
+    def test_put_cat(self, client: DiscriminatedClient, cat_body: models.Cat):
+        """Test sending cat with envelope serialization.
+
+        Expected request:
+        {
+            "kind": "cat",
+            "value": {
+                "name": "Whiskers",
+                "meow": true
+            }
+        }
+        """
+        result = client.envelope.object.default.put(cat_body)
+        assert result == cat_body
+        assert isinstance(result, models.Cat)
+
+
+# Tests for Envelope / Object / Custom Properties (envelope with custom "petType" and "petData")
+@pytest.mark.skip(reason="After completely support discriminated unions, enable these tests")
+class TestEnvelopeObjectCustomProperties:
+    """Test discriminated union with custom property names in envelope."""
+
+    def test_get_default_cat(self, client: DiscriminatedClient, cat_body: models.Cat):
+        """Test getting cat with default (no query param or petType=cat).
+
+        Expected response:
+        {
+            "petType": "cat",
+            "petData": {
+                "name": "Whiskers",
+                "meow": true
+            }
+        }
+        """
+        result = client.envelope.object.custom_properties.get()
+        assert result == cat_body
+        assert isinstance(result, models.Cat)
+
+    def test_get_with_pet_type_cat(self, client: DiscriminatedClient, cat_body: models.Cat):
+        """Test getting cat with petType=cat query parameter.
+
+        Expected response:
+        {
+            "petType": "cat",
+            "petData": {
+                "name": "Whiskers",
+                "meow": true
+            }
+        }
+        """
+        result = client.envelope.object.custom_properties.get(pet_type="cat")
+        assert result == cat_body
+        assert isinstance(result, models.Cat)
+
+    def test_get_with_pet_type_dog(self, client: DiscriminatedClient, dog_body: models.Dog):
+        """Test getting dog with petType=dog query parameter.
+
+        Expected response:
+        {
+            "petType": "dog",
+            "petData": {
+                "name": "Rex",
+                "bark": false
+            }
+        }
+        """
+        result = client.envelope.object.custom_properties.get(pet_type="dog")
+        assert result == dog_body
+        assert isinstance(result, models.Dog)
+
+    def test_put_cat(self, client: DiscriminatedClient, cat_body: models.Cat):
+        """Test sending cat with custom property names in envelope.
+
+        Expected request:
+        {
+            "petType": "cat",
+            "petData": {
+                "name": "Whiskers",
+                "meow": true
+            }
+        }
+        """
+        result = client.envelope.object.custom_properties.put(cat_body)
+        assert result == cat_body
+        assert isinstance(result, models.Cat)