Merge branch 'main' of https://github.com/NiveditJain/azure-sdk-for-python

Author: Nivedit Jain

sdk/loadtestservice/azure-developer-loadtesting/azure/__init__.py

@@ -1 +1 @@
-__path__ = __import__('pkgutil').extend_path(__path__, __name__)  # type: ignore
+__path__ = __import__("pkgutil").extend_path(__path__, __name__)  # type: ignore

sdk/loadtestservice/azure-developer-loadtesting/azure/developer/__init__.py

@@ -1 +1 @@
-__path__ = __import__('pkgutil').extend_path(__path__, __name__)  # type: ignore
+__path__ = __import__("pkgutil").extend_path(__path__, __name__)  # type: ignore

sdk/loadtestservice/azure-developer-loadtesting/azure/developer/loadtesting/__init__.py

@@ -17,7 +17,8 @@
 except ImportError:
     _patch_all = []
 from ._patch import patch_sdk as _patch_sdk
-__all__ = ['LoadTestingClient']
+
+__all__ = ["LoadTestingClient"]
 __all__.extend([p for p in _patch_all if p not in __all__])
 
 _patch_sdk()

sdk/loadtestservice/azure-developer-loadtesting/azure/developer/loadtesting/_client.py

@@ -22,6 +22,7 @@
 
     from azure.core.credentials import TokenCredential
 
+
 class LoadTestingClient:  # pylint: disable=client-accepts-api-version-keyword
     """These APIs allow end users to create, view and run load tests using Azure Load Test Service.
 
@@ -42,38 +43,22 @@ class LoadTestingClient:  # pylint: disable=client-accepts-api-version-keyword
     :paramtype api_version: str
     """
 
-    def __init__(
-        self,
-        endpoint: str,
-        credential: "TokenCredential",
-        **kwargs: Any
-    ) -> None:
-        _endpoint = 'https://{Endpoint}'
+    def __init__(self, endpoint: str, credential: "TokenCredential", **kwargs: Any) -> None:
+        _endpoint = "https://{Endpoint}"
         self._config = LoadTestingClientConfiguration(endpoint=endpoint, credential=credential, **kwargs)
         self._client = PipelineClient(base_url=_endpoint, config=self._config, **kwargs)
 
         self._serialize = Serializer()
         self._deserialize = Deserializer()
         self._serialize.client_side_validation = False
-        self.app_component = AppComponentOperations(
-            self._client, self._config, self._serialize, self._deserialize
-        )
-        self.server_metrics = ServerMetricsOperations(
-            self._client, self._config, self._serialize, self._deserialize
-        )
+        self.app_component = AppComponentOperations(self._client, self._config, self._serialize, self._deserialize)
+        self.server_metrics = ServerMetricsOperations(self._client, self._config, self._serialize, self._deserialize)
         self.test = TestOperations(  # type: ignore  # pylint: disable=abstract-class-instantiated
             self._client, self._config, self._serialize, self._deserialize
         )
-        self.test_run = TestRunOperations(
-            self._client, self._config, self._serialize, self._deserialize
-        )
-
+        self.test_run = TestRunOperations(self._client, self._config, self._serialize, self._deserialize)
 
-    def send_request(
-        self,
-        request: HttpRequest,
-        **kwargs: Any
-    ) -> HttpResponse:
+    def send_request(self, request: HttpRequest, **kwargs: Any) -> HttpResponse:
         """Runs the network request through the client's chained policies.
 
         >>> from azure.core.rest import HttpRequest
@@ -93,7 +78,7 @@ def send_request(
 
         request_copy = deepcopy(request)
         path_format_arguments = {
-            "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, 'str', skip_quote=True),
+            "Endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, "str", skip_quote=True),
         }
 
         request_copy.url = self._client.format_url(request_copy.url, **path_format_arguments)

sdk/loadtestservice/azure-developer-loadtesting/azure/developer/loadtesting/_configuration.py

@@ -33,14 +33,9 @@ class LoadTestingClientConfiguration(Configuration):  # pylint: disable=too-many
     :paramtype api_version: str
     """
 
-    def __init__(
-        self,
-        endpoint: str,
-        credential: "TokenCredential",
-        **kwargs: Any
-    ) -> None:
+    def __init__(self, endpoint: str, credential: "TokenCredential", **kwargs: Any) -> None:
         super(LoadTestingClientConfiguration, self).__init__(**kwargs)
-        api_version = kwargs.pop('api_version', "2022-06-01-preview")  # type: str
+        api_version = kwargs.pop("api_version", "2022-06-01-preview")  # type: str
 
         if endpoint is None:
             raise ValueError("Parameter 'endpoint' must not be None.")
@@ -50,23 +45,24 @@ def __init__(
         self.endpoint = endpoint
         self.credential = credential
         self.api_version = api_version
-        self.credential_scopes = kwargs.pop('credential_scopes', ['https://loadtest.azure-dev.com/.default'])
-        kwargs.setdefault('sdk_moniker', 'developer-loadtesting/{}'.format(VERSION))
+        self.credential_scopes = kwargs.pop("credential_scopes", ["https://loadtest.azure-dev.com/.default"])
+        kwargs.setdefault("sdk_moniker", "developer-loadtesting/{}".format(VERSION))
         self._configure(**kwargs)
 
     def _configure(
-        self,
-        **kwargs  # type: Any
+        self, **kwargs  # type: Any
     ):
         # type: (...) -> None
-        self.user_agent_policy = kwargs.get('user_agent_policy') or policies.UserAgentPolicy(**kwargs)
-        self.headers_policy = kwargs.get('headers_policy') or policies.HeadersPolicy(**kwargs)
-        self.proxy_policy = kwargs.get('proxy_policy') or policies.ProxyPolicy(**kwargs)
-        self.logging_policy = kwargs.get('logging_policy') or policies.NetworkTraceLoggingPolicy(**kwargs)
-        self.http_logging_policy = kwargs.get('http_logging_policy') or policies.HttpLoggingPolicy(**kwargs)
-        self.retry_policy = kwargs.get('retry_policy') or policies.RetryPolicy(**kwargs)
-        self.custom_hook_policy = kwargs.get('custom_hook_policy') or policies.CustomHookPolicy(**kwargs)
-        self.redirect_policy = kwargs.get('redirect_policy') or policies.RedirectPolicy(**kwargs)
-        self.authentication_policy = kwargs.get('authentication_policy')
+        self.user_agent_policy = kwargs.get("user_agent_policy") or policies.UserAgentPolicy(**kwargs)
+        self.headers_policy = kwargs.get("headers_policy") or policies.HeadersPolicy(**kwargs)
+        self.proxy_policy = kwargs.get("proxy_policy") or policies.ProxyPolicy(**kwargs)
+        self.logging_policy = kwargs.get("logging_policy") or policies.NetworkTraceLoggingPolicy(**kwargs)
+        self.http_logging_policy = kwargs.get("http_logging_policy") or policies.HttpLoggingPolicy(**kwargs)
+        self.retry_policy = kwargs.get("retry_policy") or policies.RetryPolicy(**kwargs)
+        self.custom_hook_policy = kwargs.get("custom_hook_policy") or policies.CustomHookPolicy(**kwargs)
+        self.redirect_policy = kwargs.get("redirect_policy") or policies.RedirectPolicy(**kwargs)
+        self.authentication_policy = kwargs.get("authentication_policy")
         if self.credential and not self.authentication_policy:
-            self.authentication_policy = policies.BearerTokenCredentialPolicy(self.credential, *self.credential_scopes, **kwargs)
+            self.authentication_policy = policies.BearerTokenCredentialPolicy(
+                self.credential, *self.credential_scopes, **kwargs
+            )

sdk/loadtestservice/azure-developer-loadtesting/azure/developer/loadtesting/_patch.py

@@ -63,8 +63,9 @@ def __init__(self, endpoint: str, credential: "TokenCredential", **kwargs: Any)
         self._serialize.client_side_validation = False
 
         self.load_test_runs = TestRunOperations(self._client, self._config, self._serialize, self._deserialize)
-        self.load_test_administration = LoadTestingAdministration(self._client, self._config, self._serialize,
-                                                                  self._deserialize)
+        self.load_test_administration = LoadTestingAdministration(
+            self._client, self._config, self._serialize, self._deserialize
+        )
 
 
 __all__: List[str] = ["LoadTestingClient"]  # Add all objects you want publicly available to users at this package level