[AutoPR automation/resource-manager] Test code - DO NOT MERGE

azure-mgmt-automation/azure/mgmt/automation/automation_client.py

@@ -68,21 +68,18 @@ class AutomationClientConfiguration(AzureConfiguration):
      identify Microsoft Azure subscription. The subscription ID forms part of
      the URI for every service call.
     :type subscription_id: str
-    :param count_type1: The type of counts to retrieve. Possible values
-     include: 'status', 'nodeconfiguration'
-    :type count_type1: str or ~azure.mgmt.automation.models.CountType
+    :ivar count_type: The type of counts to retrieve
+    :type count_type: str
     :param str base_url: Service URL
     """
 
     def __init__(
-            self, credentials, subscription_id, count_type1, base_url=None):
+            self, credentials, subscription_id, base_url=None):
 
         if credentials is None:
             raise ValueError("Parameter 'credentials' must not be None.")
         if subscription_id is None:
             raise ValueError("Parameter 'subscription_id' must not be None.")
-        if count_type1 is None:
-            raise ValueError("Parameter 'count_type1' must not be None.")
         if not base_url:
             base_url = 'https://management.azure.com'
 
@@ -93,7 +90,7 @@ def __init__(
 
         self.credentials = credentials
         self.subscription_id = subscription_id
-        self.count_type1 = count_type1
+        self.count_type = "status"
 
 
 class AutomationClient(object):
@@ -190,16 +187,13 @@ class AutomationClient(object):
      identify Microsoft Azure subscription. The subscription ID forms part of
      the URI for every service call.
     :type subscription_id: str
-    :param count_type1: The type of counts to retrieve. Possible values
-     include: 'status', 'nodeconfiguration'
-    :type count_type1: str or ~azure.mgmt.automation.models.CountType
     :param str base_url: Service URL
     """
 
     def __init__(
-            self, credentials, subscription_id, count_type1, base_url=None):
+            self, credentials, subscription_id, base_url=None):
 
-        self.config = AutomationClientConfiguration(credentials, subscription_id, count_type1, base_url)
+        self.config = AutomationClientConfiguration(credentials, subscription_id, base_url)
         self._client = ServiceClient(self.config.credentials, self.config)
 
         client_models = {k: v for k, v in models.__dict__.items() if isinstance(v, type)}

azure-mgmt-automation/azure/mgmt/automation/models/__init__.py

@@ -9,10 +9,6 @@
 # regenerated.
 # --------------------------------------------------------------------------
 
-from .error_response import ErrorResponse, ErrorResponseException
-from .resource import Resource
-from .tracked_resource import TrackedResource
-from .proxy_resource import ProxyResource
 from .sku import Sku
 from .automation_account import AutomationAccount
 from .automation_account_create_or_update_parameters import AutomationAccountCreateOrUpdateParameters
@@ -24,6 +20,10 @@
 from .key import Key
 from .key_list_result import KeyListResult
 from .automation_account_update_parameters import AutomationAccountUpdateParameters
+from .proxy_resource import ProxyResource
+from .resource import Resource
+from .tracked_resource import TrackedResource
+from .error_response import ErrorResponse, ErrorResponseException
 from .certificate_create_or_update_parameters import CertificateCreateOrUpdateParameters
 from .certificate import Certificate
 from .certificate_update_parameters import CertificateUpdateParameters
@@ -63,8 +63,6 @@
 from .module_create_or_update_parameters import ModuleCreateOrUpdateParameters
 from .module_update_parameters import ModuleUpdateParameters
 from .type_field import TypeField
-from .job_stream import JobStream
-from .job_stream_list_result import JobStreamListResult
 from .runbook_parameter import RunbookParameter
 from .runbook_draft import RunbookDraft
 from .runbook import Runbook
@@ -75,6 +73,8 @@
 from .test_job import TestJob
 from .runbook_create_or_update_draft_properties import RunbookCreateOrUpdateDraftProperties
 from .runbook_create_or_update_draft_parameters import RunbookCreateOrUpdateDraftParameters
+from .job_stream import JobStream
+from .job_stream_list_result import JobStreamListResult
 from .advanced_schedule_monthly_occurrence import AdvancedScheduleMonthlyOccurrence
 from .advanced_schedule import AdvancedSchedule
 from .schedule_create_or_update_parameters import ScheduleCreateOrUpdateParameters
@@ -171,11 +171,11 @@
     DscConfigurationState,
     GroupTypeEnum,
     ModuleProvisioningState,
-    JobStreamType,
     RunbookTypeEnum,
     RunbookState,
     RunbookProvisioningState,
     HttpStatusCode,
+    JobStreamType,
     ScheduleDay,
     ScheduleFrequency,
     OperatingSystemType,
@@ -188,14 +188,9 @@
     JobStatus,
     JobProvisioningState,
     AgentRegistrationKeyName,
-    CountType,
 )
 
 __all__ = [
-    'ErrorResponse', 'ErrorResponseException',
-    'Resource',
-    'TrackedResource',
-    'ProxyResource',
     'Sku',
     'AutomationAccount',
     'AutomationAccountCreateOrUpdateParameters',
@@ -207,6 +202,10 @@
     'Key',
     'KeyListResult',
     'AutomationAccountUpdateParameters',
+    'ProxyResource',
+    'Resource',
+    'TrackedResource',
+    'ErrorResponse', 'ErrorResponseException',
     'CertificateCreateOrUpdateParameters',
     'Certificate',
     'CertificateUpdateParameters',
@@ -246,8 +245,6 @@
     'ModuleCreateOrUpdateParameters',
     'ModuleUpdateParameters',
     'TypeField',
-    'JobStream',
-    'JobStreamListResult',
     'RunbookParameter',
     'RunbookDraft',
     'Runbook',
@@ -258,6 +255,8 @@
     'TestJob',
     'RunbookCreateOrUpdateDraftProperties',
     'RunbookCreateOrUpdateDraftParameters',
+    'JobStream',
+    'JobStreamListResult',
     'AdvancedScheduleMonthlyOccurrence',
     'AdvancedSchedule',
     'ScheduleCreateOrUpdateParameters',
@@ -353,11 +352,11 @@
     'DscConfigurationState',
     'GroupTypeEnum',
     'ModuleProvisioningState',
-    'JobStreamType',
     'RunbookTypeEnum',
     'RunbookState',
     'RunbookProvisioningState',
     'HttpStatusCode',
+    'JobStreamType',
     'ScheduleDay',
     'ScheduleFrequency',
     'OperatingSystemType',
@@ -370,5 +369,4 @@
     'JobStatus',
     'JobProvisioningState',
     'AgentRegistrationKeyName',
-    'CountType',
 ]

azure-mgmt-automation/azure/mgmt/automation/models/automation_client_enums.py

@@ -81,17 +81,6 @@ class ModuleProvisioningState(Enum):
     updating = "Updating"
 
 
-class JobStreamType(Enum):
-
-    progress = "Progress"
-    output = "Output"
-    warning = "Warning"
-    error = "Error"
-    debug = "Debug"
-    verbose = "Verbose"
-    any = "Any"
-
-
 class RunbookTypeEnum(Enum):
 
     script = "Script"
@@ -165,6 +154,17 @@ class HttpStatusCode(Enum):
     http_version_not_supported = "HttpVersionNotSupported"
 
 
+class JobStreamType(Enum):
+
+    progress = "Progress"
+    output = "Output"
+    warning = "Warning"
+    error = "Error"
+    debug = "Debug"
+    verbose = "Verbose"
+    any = "Any"
+
+
 class ScheduleDay(Enum):
 
     monday = "Monday"
@@ -267,9 +267,3 @@ class AgentRegistrationKeyName(Enum):
 
     primary = "primary"
     secondary = "secondary"
-
-
-class CountType(Enum):
-
-    status = "status"
-    nodeconfiguration = "nodeconfiguration"

azure-mgmt-automation/azure/mgmt/automation/operations/runbook_draft_operations.py

@@ -170,7 +170,7 @@ def replace_content(
         :type automation_account_name: str
         :param runbook_name: The runbook name.
         :type runbook_name: str
-        :param runbook_content: The runbook draft content.
+        :param runbook_content: The runbook draft content.
         :type runbook_content: str
         :param dict custom_headers: headers that will be added to the request
         :param bool raw: returns the direct response alongside the