azure-developer-loadtesting 1.0.0__zip

azure-developer-loadtesting-1.0.0/azure/developer/loadtesting/_generated/aio/operations/_patch.py

@@ -0,0 +1,187 @@
+# ------------------------------------
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+# ------------------------------------
+"""Customize generated code here.
+
+Follow our quickstart for examples: https://aka.ms/azsdk/python/dpcodegen/python/customize
+"""
+import asyncio
+import logging
+from functools import partial
+from typing import List, Optional, Any, IO, Union, Callable, overload, Generic, TypeVar
+
+from azure.core.polling import AsyncPollingMethod, AsyncLROPoller
+from azure.core.tracing.decorator import distributed_trace
+
+from ._operations import AdministrationOperations as AdministrationOperationsGenerated, JSON
+from ._operations import TestRunOperations as TestRunOperationsGenerated
+
+PollingReturnType = TypeVar("PollingReturnType")
+logger = logging.getLogger(__name__)
+
+
+class AsyncLoadTestingPollingMethod(AsyncPollingMethod):
+    """Base class for custom async polling methods."""
+
+    def _update_status(self) -> None:
+        raise NotImplementedError("This method needs to be implemented")
+
+    async def _update_resource(self) -> None:
+        self._resource = await self._command()
+
+    def initialize(self, client, initial_response, deserialization_callback) -> None:
+        self._command = client
+        self._initial_response = initial_response
+        self._resource = initial_response
+
+    def status(self) -> str:
+        return self._status
+
+    def finished(self) -> bool:
+        return self._status in self._termination_statuses
+
+    def resource(self) -> JSON:
+        return self._resource
+
+    async def run(self) -> None:
+        try:
+            while not self.finished():
+                await self._update_resource()
+                self._update_status()
+
+                if not self.finished():
+                    await asyncio.sleep(self._polling_interval)
+        except Exception as e:
+            logger.error(e)
+            raise e
+
+
+class AsyncValidationCheckPoller(AsyncLoadTestingPollingMethod):
+    def __init__(self, interval=5) -> None:
+        self._resource = None
+        self._command = None
+        self._initial_response = None
+        self._polling_interval = interval
+        self._status = None
+        self._termination_statuses = ["VALIDATION_SUCCESS", "VALIDATION_FAILED", "VALIDATION_NOT_REQUIRED"]
+
+    def _update_status(self) -> None:
+        self._status = self._resource["validationStatus"]
+
+
+class AsyncTestRunStatusPoller(AsyncLoadTestingPollingMethod):
+    def __init__(self, interval=5) -> None:
+        self._resource = None
+        self._command = None
+        self._initial_response = None
+        self._polling_interval = interval
+        self._status = None
+        self._termination_statuses = ["DONE", "FAILED", "CANCELLED"]
+
+    def _update_status(self) -> None:
+        self._status = self._resource["status"]
+
+
+class AdministrationOperations(AdministrationOperationsGenerated):
+    """
+    for performing the operations on test
+    """
+
+    def __init__(self, *args, **kwargs):
+        super(AdministrationOperations, self).__init__(*args, **kwargs)
+
+    @distributed_trace
+    async def begin_upload_test_file(
+        self, test_id: str, file_name: str, body: IO, *, file_type: Optional[str] = None, **kwargs: Any
+    ) -> AsyncLROPoller[JSON]:
+        """Upload file to the test
+
+        :param test_id: Unique id for the test
+        :type test_id: str
+        :param file_name: Name of the file to be uploaded
+        :type file_name: str
+        :param body: File content to be uploaded
+        :type body: IO
+        :param file_type: Type of the file to be uploaded
+        :type file_type: str
+        :return: An instance of LROPoller object to check the validation status of file
+        :rtype: ~azure.core.polling.LROPoller
+        :raises ~azure.core.exceptions.HttpResponseError:
+        :raises ~azure.core.exceptions.ResourceNotFoundError:
+        """
+
+        polling_interval = kwargs.pop("_polling_interval", None)
+        if polling_interval is None:
+            polling_interval = 5
+        upload_test_file_operation = await super().begin_upload_test_file(
+            test_id=test_id, file_name=file_name, body=body, file_type=file_type, **kwargs
+        )
+
+        command = partial(self.get_test_file, test_id=test_id, file_name=file_name)
+
+        create_validation_status_polling = AsyncValidationCheckPoller(interval=polling_interval)
+        return AsyncLROPoller(command, upload_test_file_operation, lambda *_: None, create_validation_status_polling)
+
+
+class TestRunOperations(TestRunOperationsGenerated):
+    """
+    class to perform operations on TestRun
+    """
+
+    def __init__(self, *args, **kwargs):
+        super(TestRunOperations, self).__init__(*args, **kwargs)
+
+    @distributed_trace
+    async def begin_test_run(
+        self, test_run_id: str, body: Union[JSON, IO], *, old_test_run_id: Optional[str] = None, **kwargs: Any
+    ) -> AsyncLROPoller[JSON]:
+        """Create and start a new test run with the given name.
+
+        Create and start a new test run with the given name.
+
+        :param test_run_id: Unique name for the load test run, must contain only lower-case alphabetic,
+         numeric, underscore or hyphen characters. Required.
+        :type test_run_id: str
+        :param body: Load test run model. Is either a model type or a IO type. Required.
+        :type body: JSON or IO
+        :keyword old_test_run_id: Existing test run identifier that should be rerun, if this is
+         provided, the test will run with the JMX file, configuration and app components from the
+         existing test run. You can override the configuration values for new test run in the request
+         body. Default value is None.
+        :paramtype old_test_run_id: str
+        :keyword content_type: Body Parameter content-type. Known values are:
+         'application/merge-patch+json'. Default value is None.
+        :paramtype content_type: str
+
+        :rtype: ~azure.developer.loadtesting._polling.LoadTestingLROPoller
+        :raises ~azure.core.exceptions.HttpResponseError:
+        :raises ~azure.core.exceptions.ResourceNotFoundError:
+        """
+
+        polling_interval = kwargs.pop("_polling_interval", None)
+        if polling_interval is None:
+            polling_interval = 5
+        create_or_update_test_run_operation = await super()._test_run_initial(
+            test_run_id, body, old_test_run_id=old_test_run_id, **kwargs
+        )
+
+        command = partial(self.get_test_run, test_run_id=test_run_id)
+
+        create_test_run_polling = AsyncTestRunStatusPoller(interval=polling_interval)
+        return AsyncLROPoller(command, create_or_update_test_run_operation, lambda *_: None, create_test_run_polling)
+
+
+__all__: List[str] = ["AdministrationOperations", "TestRunOperations"]
+
+
+# Add all objects you want publicly available to users at this package level
+
+
+def patch_sdk():
+    """Do not remove from this file.
+
+    `patch_sdk` is a last resort escape hatch that allows you to do customizations
+    you can't accomplish using the techniques described in
+    https://aka.ms/azsdk/python/dpcodegen/python/customize
+    """

azure-developer-loadtesting-1.0.0/azure/developer/loadtesting/_generated/operations/__init__.py

@@ -0,0 +1,21 @@
+# coding=utf-8
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is regenerated.
+# --------------------------------------------------------------------------
+
+from ._operations import AdministrationOperations
+from ._operations import TestRunOperations
+
+from ._patch import __all__ as _patch_all
+from ._patch import *  # pylint: disable=unused-wildcard-import
+from ._patch import patch_sdk as _patch_sdk
+
+__all__ = [
+    "AdministrationOperations",
+    "TestRunOperations",
+]
+__all__.extend([p for p in _patch_all if p not in __all__])
+_patch_sdk()