@typespec/http-client-python 0.15.2 → 0.16.1-dev.0

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

@@ -82,3 +82,75 @@ def test_lro_begin_delete(client):
         resource_group_name=RESOURCE_GROUP_NAME,
         order_name="order1",
     ).result()
+
+
+def test_optional_body_get(client):
+    result = client.optional_body.get(
+        resource_group_name=RESOURCE_GROUP_NAME,
+        widget_name="widget1",
+    )
+    assert result.name == "widget1"
+    assert (
+        result.id
+        == "/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/test-rg/providers/Azure.ResourceManager.OperationTemplates/widgets/widget1"
+    )
+    assert result.type == "Azure.ResourceManager.OperationTemplates/widgets"
+    assert result.location == "eastus"
+    assert result.properties.name == "widget1"
+    assert result.properties.description == "A test widget"
+    assert result.properties.provisioning_state == "Succeeded"
+
+
+def test_optional_body_patch_without_body(client):
+    result = client.optional_body.patch(
+        resource_group_name=RESOURCE_GROUP_NAME,
+        widget_name="widget1",
+    )
+    assert result.name == "widget1"
+    assert result.properties.name == "widget1"
+    assert result.properties.description == "A test widget"
+
+
+def test_optional_body_patch_with_body(client):
+    result = client.optional_body.patch(
+        resource_group_name=RESOURCE_GROUP_NAME,
+        widget_name="widget1",
+        properties=models.Widget(
+            location="eastus",
+            properties=models.WidgetProperties(name="updated-widget", description="Updated description"),
+        ),
+    )
+    assert result.name == "widget1"
+    assert result.properties.name == "updated-widget"
+    assert result.properties.description == "Updated description"
+
+
+def test_optional_body_post_without_body(client):
+    result = client.optional_body.post(
+        resource_group_name=RESOURCE_GROUP_NAME,
+        widget_name="widget1",
+    )
+    assert result.result == "Action completed successfully"
+
+
+def test_optional_body_post_with_body(client):
+    result = client.optional_body.post(
+        resource_group_name=RESOURCE_GROUP_NAME,
+        widget_name="widget1",
+        body=models.ActionRequest(action_type="perform", parameters="test-parameters"),
+    )
+    assert result.result == "Action completed successfully with parameters"
+
+
+def test_optional_body_provider_post_without_body(client):
+    result = client.optional_body.provider_post()
+    assert result.total_allowed == 50
+    assert result.status == "Changed to default allowance"
+
+
+def test_optional_body_provider_post_with_body(client):
+    result = client.optional_body.provider_post(
+        body=models.ChangeAllowanceRequest(total_allowed=100, reason="Increased demand"),
+    )
+    assert result.total_allowed == 100
+    assert result.status == "Changed to requested allowance"

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

@@ -0,0 +1,45 @@
+# -------------------------------------------------------------------------
+# 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.hierarchybuilding import HierarchyBuildingClient
+from specs.azure.clientgenerator.core.hierarchybuilding.models import (
+    Pet,
+    Dog,
+)
+
+
+@pytest.fixture
+def client():
+    with HierarchyBuildingClient() as client:
+        yield client
+
+
+# ========== test for spector ==========
+
+
+def test_update_pet_as_pet(client: HierarchyBuildingClient):
+    resp = Pet(name="Buddy", trained=True)
+    assert client.pet_operations.update_pet_as_pet(Pet(name="Buddy", trained=True)) == resp
+
+
+def test_update_dog_as_pet(client: HierarchyBuildingClient):
+    resp = Dog(name="Rex", trained=True, breed="German Shepherd")
+    assert client.pet_operations.update_dog_as_pet(Dog(name="Rex", trained=True, breed="German Shepherd")) == resp
+
+
+def test_update_pet_as_animal(client: HierarchyBuildingClient):
+    resp = Pet(name="Buddy", trained=True)
+    assert client.animal_operations.update_pet_as_animal(Pet(name="Buddy", trained=True)) == resp
+
+
+def test_update_dog_as_animal(client: HierarchyBuildingClient):
+    resp = Dog(name="Rex", trained=True, breed="German Shepherd")
+    assert client.animal_operations.update_dog_as_animal(Dog(name="Rex", trained=True, breed="German Shepherd")) == resp
+
+
+def test_update_dog_as_dog(client: HierarchyBuildingClient):
+    resp = Dog(name="Rex", trained=True, breed="German Shepherd")
+    assert client.dog_operations.update_dog_as_dog(Dog(name="Rex", trained=True, breed="German Shepherd")) == resp

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

@@ -4,7 +4,7 @@
 # license information.
 # --------------------------------------------------------------------------
 import pytest
-from client.naming import NamingClient, models
+from client.naming.main import NamingClient, models
 
 
 @pytest.fixture
@@ -42,11 +42,11 @@ def test_header_response(client: NamingClient):
 
 
 def test_model_client(client: NamingClient):
-    client.client_model.client(models.ClientModel(default_name=True))
+    client.model_client.client(models.ClientModel(default_name=True))
 
 
 def test_model_language(client: NamingClient):
-    client.client_model.language(models.PythonModel(default_name=True))
+    client.model_client.language(models.PythonModel(default_name=True))
 
 
 def test_union_enum_member_name(client: NamingClient):

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

@@ -0,0 +1,35 @@
+# -------------------------------------------------------------------------
+# 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 client.naming.enumconflict import EnumConflictClient
+from client.naming.enumconflict.firstnamespace import models as first_models
+from client.naming.enumconflict.secondnamespace import models as second_models
+
+
+@pytest.fixture
+def client():
+    with EnumConflictClient() as client:
+        yield client
+
+
+def test_first_operations_first(client: EnumConflictClient):
+    """Test enum with same name in different namespace - first namespace."""
+    body = first_models.FirstModel(status=first_models.Status.ACTIVE, name="test")
+
+    response = client.first_operations.first(body=body)
+
+    assert response.status == "active"
+    assert response.name == "test"
+
+
+def test_second_operations_second(client: EnumConflictClient):
+    """Test enum with same name in different namespace - second namespace."""
+    body = second_models.SecondModel(status=second_models.SecondStatus.RUNNING, description="test description")
+
+    response = client.second_operations.second(body=body)
+
+    assert response.status == "running"
+    assert response.description == "test description"

package/generator/test/azure/requirements.txt

@@ -10,6 +10,7 @@ azure-mgmt-core==1.6.0
 -e ./generated/azure-client-generator-core-client-initialization
 -e ./generated/azure-client-generator-core-deserialize-empty-string-as-null
 -e ./generated/azure-client-generator-core-flatten-property
+-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
@@ -31,6 +32,7 @@ azure-mgmt-core==1.6.0
 -e ./generated/client-namespace
 -e ./generated/azure-payload-pageable
 -e ./generated/client-naming
+-e ./generated/client-naming-enum-conflict
 -e ./generated/client-overload
 -e ./generated/client-structure-default
 -e ./generated/client-structure-multiclient

package/generator/test/unbranded/mock_api_tests/conftest.py

@@ -9,7 +9,6 @@ import signal
 import pytest
 import re
 from pathlib import Path
-from typing import List
 
 FILE_FOLDER = Path(__file__).parent
 
@@ -76,7 +75,7 @@ SPECIAL_WORDS = [
 
 
 @pytest.fixture
-def special_words() -> List[str]:
+def special_words() -> list[str]:
     return SPECIAL_WORDS