fix test

Author: msyyc

packages/http-client-python/generator/pygen/codegen/models/client.py

@@ -89,7 +89,15 @@ def __init__(
 
     @property
     def need_cloud_setting(self) -> bool:
-        return bool(self.code_model.options["azure_arm"] and self.credential_scopes is not None)
+        return bool(
+            self.code_model.options["azure_arm"]
+            and self.credential_scopes is not None
+            and self.endpoint_parameter is not None
+        )
+
+    @property
+    def endpoint_parameter(self) -> Optional[Parameter]:
+        return next((p for p in self.parameters.parameters if p.location == ParameterLocation.ENDPOINT_PATH), None)
 
     def _build_request_builders(
         self,

packages/http-client-python/generator/pygen/codegen/serializers/client_serializer.py

@@ -3,10 +3,10 @@
 # Licensed under the MIT License. See License.txt in the project root for
 # license information.
 # --------------------------------------------------------------------------
-from typing import List
+from typing import List, cast
 
 from . import utils
-from ..models import Client, ParameterMethodLocation
+from ..models import Client, ParameterMethodLocation, Parameter
 from .parameter_serializer import ParameterSerializer, PopKwargType
 from ...utils import build_policies
 
@@ -82,12 +82,13 @@ def initialize_config(self) -> List[str]:
         additional_signatures = []
         if self.client.need_cloud_setting:
             additional_signatures.append("credential_scopes=credential_scopes")
+            endpoint_parameter = cast(Parameter, self.client.endpoint_parameter)
             retval.extend(
                 [
                     '_cloud = kwargs.pop("cloud_setting", None) or settings.current.azure_cloud  # type: ignore',
                     "_endpoints = get_arm_endpoints(_cloud)",
-                    "if not base_url:",
-                    '    base_url = _endpoints["resource_manager"]',
+                    f"if not {endpoint_parameter.client_name}:",
+                    f'    {endpoint_parameter.client_name} = _endpoints["resource_manager"]',
                     'credential_scopes = kwargs.pop("credential_scopes", _endpoints["credential_scopes"])',
                 ]
             )