@typespec/http-client-python 0.28.2 → 0.29.0

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

@@ -1,207 +0,0 @@
-# -------------------------------------------------------------------------
-# Copyright (c) Microsoft Corporation. All rights reserved.
-# Licensed under the MIT License. See License.txt in the project root for
-# license information.
-# --------------------------------------------------------------------------
-import os
-import subprocess
-import signal
-import pytest
-import re
-from typing import Literal
-from pathlib import Path
-
-FILE_FOLDER = Path(__file__).parent
-
-
-def start_server_process():
-    azure_http_path = Path(os.path.dirname(__file__)) / Path("../../../../node_modules/@azure-tools/azure-http-specs")
-    http_path = Path(os.path.dirname(__file__)) / Path("../../../../node_modules/@typespec/http-specs")
-    os.chdir(azure_http_path.resolve())
-    cmd = f"npx tsp-spector serve ./specs  {(http_path / 'specs').resolve()}"
-    if os.name == "nt":
-        return subprocess.Popen(cmd, shell=True)
-    return subprocess.Popen(cmd, shell=True, preexec_fn=os.setsid)
-
-
-def terminate_server_process(process):
-    if os.name == "nt":
-        process.kill()
-    else:
-        os.killpg(os.getpgid(process.pid), signal.SIGTERM)  # Send the signal to all the process groups
-
-
-@pytest.fixture(scope="session", autouse=True)
-def testserver():
-    """Start spector ranch mock api tests"""
-    server = start_server_process()
-    yield
-    terminate_server_process(server)
-
-
-_VALID_UUID = re.compile(r"^[0-9a-f]{8}-([0-9a-f]{4}-){3}[0-9a-f]{12}$")
-_VALID_RFC7231 = re.compile(
-    r"^(Mon|Tue|Wed|Thu|Fri|Sat|Sun),\s\d{2}\s"
-    r"(Jan|Feb|Mar|Apr|May|Jun|Jul|Aug|Sep|Oct|Nov|Dec)\s\d{4}\s\d{2}:\d{2}:\d{2}\sGMT$"
-)
-
-
-def validate_format(value: str, format: Literal["uuid", "rfc7231"]):
-    if format == "uuid":
-        assert _VALID_UUID.match(value)
-    elif format == "rfc7231":
-        assert _VALID_RFC7231.match(value)
-    else:
-        raise ValueError("Unknown format")
-
-
-@pytest.fixture
-def check_repeatability_header():
-    def func(request):
-        validate_format(request.http_request.headers["Repeatability-Request-ID"], "uuid")
-        validate_format(request.http_request.headers["Repeatability-First-Sent"], "rfc7231")
-
-    return func
-
-
-@pytest.fixture
-def check_client_request_id_header():
-    def func(request, header: str, checked: dict):
-        validate_format(request.http_request.headers[header], "uuid")
-        checked[header] = request.http_request.headers[header]
-
-    return func
-
-
-# ================== after azure-core fix, the following code can be removed (begin) ==================
-import urllib.parse
-from azure.core.rest import HttpRequest
-
-
-def update_api_version_of_status_link(status_link: str):
-    request_params = {}
-    parsed_status_link = urllib.parse.urlparse(status_link)
-    request_params = {
-        key.lower(): [urllib.parse.quote(v) for v in value]
-        for key, value in urllib.parse.parse_qs(parsed_status_link.query).items()
-    }
-    request_params["api-version"] = "2022-12-01-preview"
-    status_link = urllib.parse.urljoin(status_link, parsed_status_link.path)
-    return status_link, request_params
-
-
-@pytest.fixture
-def polling_method():
-    from azure.core.polling.base_polling import LROBasePolling
-
-    class TempLroBasePolling(LROBasePolling):
-
-        def request_status(self, status_link: str):
-            if self._path_format_arguments:
-                status_link = self._client.format_url(status_link, **self._path_format_arguments)
-            status_link, request_params = update_api_version_of_status_link(status_link)
-            if "request_id" not in self._operation_config:
-                self._operation_config["request_id"] = self._get_request_id()
-
-            rest_request = HttpRequest("GET", status_link, params=request_params)
-            return self._client.send_request(rest_request, _return_pipeline_response=True, **self._operation_config)
-
-    return TempLroBasePolling(0)
-
-
-@pytest.fixture
-def async_polling_method():
-    from azure.core.polling.async_base_polling import AsyncLROBasePolling
-
-    class AsyncTempLroBasePolling(AsyncLROBasePolling):
-
-        async def request_status(self, status_link: str):
-            if self._path_format_arguments:
-                status_link = self._client.format_url(status_link, **self._path_format_arguments)
-            status_link, request_params = update_api_version_of_status_link(status_link)
-            # Re-inject 'x-ms-client-request-id' while polling
-            if "request_id" not in self._operation_config:
-                self._operation_config["request_id"] = self._get_request_id()
-
-            rest_request = HttpRequest("GET", status_link, params=request_params)
-            return await self._client.send_request(
-                rest_request, _return_pipeline_response=True, **self._operation_config
-            )
-
-    return AsyncTempLroBasePolling(0)
-
-
-# ================== after azure-core fix, the up code can be removed (end) ==================
-
-
-@pytest.fixture()
-def credential():
-    """I actually don't need anything, since the authentication policy
-    will bypass it.
-    """
-
-    class FakeCredential:
-        pass
-
-    return FakeCredential()
-
-
-@pytest.fixture()
-def authentication_policy():
-    from azure.core.pipeline.policies import SansIOHTTPPolicy
-
-    return SansIOHTTPPolicy()
-
-
-SPECIAL_WORDS = [
-    "and",
-    "as",
-    "assert",
-    "async",
-    "await",
-    "break",
-    "class",
-    "constructor",
-    "continue",
-    "def",
-    "del",
-    "elif",
-    "else",
-    "except",
-    "exec",
-    "finally",
-    "for",
-    "from",
-    "global",
-    "if",
-    "import",
-    "in",
-    "is",
-    "lambda",
-    "not",
-    "or",
-    "pass",
-    "raise",
-    "return",
-    "try",
-    "while",
-    "with",
-    "yield",
-]
-
-
-@pytest.fixture
-def special_words() -> list[str]:
-    return SPECIAL_WORDS
-
-
-@pytest.fixture
-def png_data() -> bytes:
-    with open(str(FILE_FOLDER / "data/image.png"), "rb") as file_in:
-        return file_in.read()
-
-
-@pytest.fixture
-def jpg_data() -> bytes:
-    with open(str(FILE_FOLDER / "data/image.jpg"), "rb") as file_in:
-        return file_in.read()

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

@@ -1,89 +0,0 @@
-# -------------------------------------------------------------------------
-# 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.resourcemanager.commonproperties import CommonPropertiesClient
-from azure.resourcemanager.commonproperties import models
-from azure.core import exceptions
-
-SUBSCRIPTION_ID = "00000000-0000-0000-0000-000000000000"
-RESOURCE_GROUP_NAME = "test-rg"
-
-
-@pytest.fixture
-def client(credential, authentication_policy):
-    with CommonPropertiesClient(
-        credential, SUBSCRIPTION_ID, "http://localhost:3000", authentication_policy=authentication_policy
-    ) as client:
-        yield client
-
-
-def test_managed_identity_get(client):
-    result = client.managed_identity.get(
-        resource_group_name=RESOURCE_GROUP_NAME, managed_identity_tracked_resource_name="identity"
-    )
-    assert result.location == "eastus"
-    assert result.identity.type == "SystemAssigned"
-    assert result.properties.provisioning_state == "Succeeded"
-
-
-def test_managed_identity_create_with_system_assigned(client):
-    result = client.managed_identity.create_with_system_assigned(
-        resource_group_name=RESOURCE_GROUP_NAME,
-        managed_identity_tracked_resource_name="identity",
-        resource=models.ManagedIdentityTrackedResource(
-            location="eastus", identity=models.ManagedServiceIdentity(type="SystemAssigned")
-        ),
-    )
-    assert result.location == "eastus"
-    assert result.identity.type == "SystemAssigned"
-    assert result.properties.provisioning_state == "Succeeded"
-
-
-def test_managed_identity_update_with_user_assigned_and_system_assigned(client):
-    result = client.managed_identity.update_with_user_assigned_and_system_assigned(
-        resource_group_name=RESOURCE_GROUP_NAME,
-        managed_identity_tracked_resource_name="identity",
-        properties=models.ManagedIdentityTrackedResource(
-            location="eastus",
-            identity=models.ManagedServiceIdentity(
-                type="SystemAssigned,UserAssigned",
-                user_assigned_identities={
-                    "/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/test-rg/providers/Microsoft.ManagedIdentity/userAssignedIdentities/id1": models.UserAssignedIdentity()
-                },
-            ),
-        ),
-    )
-    assert result.location == "eastus"
-    assert result.identity.type == "SystemAssigned,UserAssigned"
-    assert result.properties.provisioning_state == "Succeeded"
-
-
-def test_error_get_for_predefined_error(client):
-    try:
-        client.error.get_for_predefined_error(
-            resource_group_name=RESOURCE_GROUP_NAME,
-            confidential_resource_name="confidential",
-        )
-    except exceptions.ResourceNotFoundError as e:
-        assert e.status_code == 404
-        assert (
-            e.error.message
-            == "The Resource 'Azure.ResourceManager.CommonProperties/confidentialResources/confidential' under resource group 'test-rg' was not found."
-        )
-
-
-def test_error_create_for_user_defined_error(client):
-    try:
-        client.error.create_for_user_defined_error(
-            resource_group_name=RESOURCE_GROUP_NAME,
-            confidential_resource_name="confidential",
-            resource=models.ConfidentialResource(
-                location="eastus", properties=models.ConfidentialResourceProperties(username="00")
-            ),
-        )
-    except exceptions.HttpResponseError as e:
-        assert e.status_code == 400
-        assert e.error.message == "Username should not contain only numbers."

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

@@ -1,27 +0,0 @@
-# -------------------------------------------------------------------------
-# 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.resourcemanager.largeheader import LargeHeaderClient
-from azure.resourcemanager.largeheader import models
-
-SUBSCRIPTION_ID = "00000000-0000-0000-0000-000000000000"
-RESOURCE_GROUP_NAME = "test-rg"
-
-
-@pytest.fixture
-def client(credential, authentication_policy):
-    with LargeHeaderClient(
-        credential, SUBSCRIPTION_ID, "http://localhost:3000", authentication_policy=authentication_policy
-    ) as client:
-        yield client
-
-
-def test_large_headers_begin_two6_k(client: LargeHeaderClient):
-    result = client.large_headers.begin_two6_k(
-        resource_group_name=RESOURCE_GROUP_NAME,
-        large_header_name="header1",
-    ).result()
-    assert result == models.CancelResult(succeeded=True)

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

@@ -1,34 +0,0 @@
-# -------------------------------------------------------------------------
-# 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.resourcemanager.nonresource import NonResourceClient
-from azure.resourcemanager.nonresource import models
-
-SUBSCRIPTION_ID = "00000000-0000-0000-0000-000000000000"
-RESOURCE_GROUP_NAME = "test-rg"
-
-
-@pytest.fixture
-def client(credential, authentication_policy):
-    with NonResourceClient(
-        credential, SUBSCRIPTION_ID, "http://localhost:3000", authentication_policy=authentication_policy
-    ) as client:
-        yield client
-
-
-def test_non_resource_create(client: NonResourceClient):
-    result = client.non_resource_operations.create(
-        location="eastus", parameter="hello", body=models.NonResource(id="id", name="hello", type="nonResource")
-    )
-    assert result == models.NonResource(id="id", name="hello", type="nonResource")
-
-
-def test_non_resource_get(client: NonResourceClient):
-    result = client.non_resource_operations.get(
-        location="eastus",
-        parameter="hello",
-    )
-    assert result == models.NonResource(id="id", name="hello", type="nonResource")

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

@@ -1,180 +0,0 @@
-# -------------------------------------------------------------------------
-# 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.resourcemanager.operationtemplates import OperationTemplatesClient
-from azure.resourcemanager.operationtemplates import models
-
-SUBSCRIPTION_ID = "00000000-0000-0000-0000-000000000000"
-RESOURCE_GROUP_NAME = "test-rg"
-
-
-@pytest.fixture
-def client(credential, authentication_policy):
-    with OperationTemplatesClient(
-        credential,
-        SUBSCRIPTION_ID,
-        "http://localhost:3000",
-        authentication_policy=authentication_policy,
-        polling_interval=0,
-    ) as client:
-        yield client
-
-
-def test_check_name_availability_check_global(client):
-    result = client.check_name_availability.check_global(
-        body=models.CheckNameAvailabilityRequest(name="checkName", type="Microsoft.Web/site")
-    )
-    assert result.name_available == False
-    assert result.reason == models.CheckNameAvailabilityReason.ALREADY_EXISTS
-    assert result.message == "Hostname 'checkName' already exists. Please select a different name."
-
-
-def test_check_name_availability_check_local(client):
-    result = client.check_name_availability.check_local(
-        location="westus",
-        body=models.CheckNameAvailabilityRequest(name="checkName", type="Microsoft.Web/site"),
-    )
-    assert result.name_available == False
-    assert result.reason == models.CheckNameAvailabilityReason.ALREADY_EXISTS
-    assert result.message == "Hostname 'checkName' already exists. Please select a different name."
-
-
-def test_operations_list(client):
-    result = client.operations.list().next()
-    assert result.name == "Microsoft.Compute/virtualMachines/write"
-    assert result.display.operation == "Create or Update Virtual Machine."
-    assert result.origin == "user,system"
-    assert result.action_type == "Internal"
-
-
-def test_lro_begin_create_or_replace(client):
-    result = client.lro.begin_create_or_replace(
-        resource_group_name=RESOURCE_GROUP_NAME,
-        order_name="order1",
-        resource=models.Order(
-            location="eastus",
-            properties=models.OrderProperties(product_id="product1", amount=1),
-        ),
-    ).result()
-    assert result.name == "order1"
-    assert (
-        result.id
-        == "/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/test-rg/providers/Azure.ResourceManager.OperationTemplates/orders/order1"
-    )
-    assert result.type == "Azure.ResourceManager.Resources/orders"
-    assert result.location == "eastus"
-    assert result.system_data.created_by == "AzureSDK"
-
-
-def test_lro_begin_export(client):
-    client.lro.begin_export(
-        resource_group_name=RESOURCE_GROUP_NAME,
-        order_name="order1",
-        body=models.ExportRequest(format="csv"),
-    ).result()
-
-
-def test_lro_begin_delete(client):
-    client.lro.begin_delete(
-        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"
-
-
-def test_lro_begin_export_array(client):
-    result = client.lro.begin_export_array(
-        body=models.ExportRequest(format="csv"),
-    ).result()
-    assert len(result) == 2
-    assert result[0].content == "order1,product1,1"
-    assert result[1].content == "order2,product2,2"
-
-
-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"