[Batch] Data plane SDK v14.2.0 (#34898)

* Ran autorest for 2024-02-01 updates

* Add autoosupgrade tests
- Remove application package test
- add sleep for timeout error

* Update changelog

Author: cRui861

sdk/batch/azure-batch/CHANGELOG.md

@@ -1,5 +1,13 @@
 # Release History
 
+## 14.2.0 (2024-02-01)
+
+### Features Added
+
+- Added `UpgradePolicy` to `CloudPool` definition for pool creation
+  - Added `AutomaticOSUpgradePolicy` to include configuration parameters for automatic OS upgrades
+  - Added `RollingUpgradePolicy` to include configuration parameters for rolling upgrades
+
 ## 14.1.0 (2023-11-01)
 
 ### Features Added

sdk/batch/azure-batch/azure/batch/_batch_service_client.py

@@ -67,7 +67,7 @@ def __init__(
         super(BatchServiceClient, self).__init__(self.config.credentials, self.config)
 
         client_models = {k: v for k, v in models.__dict__.items() if isinstance(v, type)}
-        self.api_version = '2023-11-01.18.0'
+        self.api_version = '2024-02-01.19.0'
         self._serialize = Serializer(client_models)
         self._deserialize = Deserializer(client_models)
 

sdk/batch/azure-batch/azure/batch/_version.py

@@ -9,5 +9,4 @@
 # regenerated.
 # --------------------------------------------------------------------------
 
-VERSION = "14.1.0"
-
+VERSION = "14.2.0"

sdk/batch/azure-batch/azure/batch/models/__init__.py

@@ -18,6 +18,7 @@
     from ._models_py3 import ApplicationPackageReference
     from ._models_py3 import ApplicationSummary
     from ._models_py3 import AuthenticationTokenSettings
+    from ._models_py3 import AutomaticOSUpgradePolicy
     from ._models_py3 import AutoPoolSpecification
     from ._models_py3 import AutoScaleRun
     from ._models_py3 import AutoScaleRunError
@@ -189,6 +190,7 @@
     from ._models_py3 import ResizeError
     from ._models_py3 import ResourceFile
     from ._models_py3 import ResourceStatistics
+    from ._models_py3 import RollingUpgradePolicy
     from ._models_py3 import Schedule
     from ._models_py3 import SecurityProfile
     from ._models_py3 import ServiceArtifactReference
@@ -223,6 +225,7 @@
     from ._models_py3 import TaskUpdateOptions
     from ._models_py3 import TaskUpdateParameter
     from ._models_py3 import UefiSettings
+    from ._models_py3 import UpgradePolicy
     from ._models_py3 import UploadBatchServiceLogsConfiguration
     from ._models_py3 import UploadBatchServiceLogsResult
     from ._models_py3 import UsageStatistics
@@ -244,6 +247,7 @@
     from ._models import ApplicationPackageReference
     from ._models import ApplicationSummary
     from ._models import AuthenticationTokenSettings
+    from ._models import AutomaticOSUpgradePolicy
     from ._models import AutoPoolSpecification
     from ._models import AutoScaleRun
     from ._models import AutoScaleRunError
@@ -415,6 +419,7 @@
     from ._models import ResizeError
     from ._models import ResourceFile
     from ._models import ResourceStatistics
+    from ._models import RollingUpgradePolicy
     from ._models import Schedule
     from ._models import SecurityProfile
     from ._models import ServiceArtifactReference
@@ -449,6 +454,7 @@
     from ._models import TaskUpdateOptions
     from ._models import TaskUpdateParameter
     from ._models import UefiSettings
+    from ._models import UpgradePolicy
     from ._models import UploadBatchServiceLogsConfiguration
     from ._models import UploadBatchServiceLogsResult
     from ._models import UsageStatistics
@@ -502,6 +508,7 @@
     NetworkSecurityGroupRuleAccess,
     IPAddressProvisioningType,
     NodeCommunicationMode,
+    UpgradeMode,
     PoolLifetimeOption,
     OnAllTasksComplete,
     OnTaskFailure,
@@ -537,6 +544,7 @@
     'ApplicationPackageReference',
     'ApplicationSummary',
     'AuthenticationTokenSettings',
+    'AutomaticOSUpgradePolicy',
     'AutoPoolSpecification',
     'AutoScaleRun',
     'AutoScaleRunError',
@@ -708,6 +716,7 @@
     'ResizeError',
     'ResourceFile',
     'ResourceStatistics',
+    'RollingUpgradePolicy',
     'Schedule',
     'SecurityProfile',
     'ServiceArtifactReference',
@@ -742,6 +751,7 @@
     'TaskUpdateOptions',
     'TaskUpdateParameter',
     'UefiSettings',
+    'UpgradePolicy',
     'UploadBatchServiceLogsConfiguration',
     'UploadBatchServiceLogsResult',
     'UsageStatistics',
@@ -794,6 +804,7 @@
     'NetworkSecurityGroupRuleAccess',
     'IPAddressProvisioningType',
     'NodeCommunicationMode',
+    'UpgradeMode',
     'PoolLifetimeOption',
     'OnAllTasksComplete',
     'OnTaskFailure',

sdk/batch/azure-batch/azure/batch/models/_batch_service_client_enums.py

@@ -179,6 +179,13 @@ class NodeCommunicationMode(str, Enum):
     simplified = "simplified"  #: Nodes using the simplified communication mode require outbound TCP communication on port 443 to the "BatchNodeManagement.{region}" service tag. No open inbound ports are required.
 
 
+class UpgradeMode(str, Enum):
+
+    automatic = "automatic"  #: All virtual machines in the scale set are automatically updated at the same time.
+    manual = "manual"  #: You control the application of updates to virtual machines in the scale set. You do this by using the manualUpgrade action.
+    rolling = "rolling"  #: The existing instances in a scale set are brought down in batches to be upgraded. Once the upgraded batch is complete, the instances will begin taking traffic again and the next batch will begin. This continues until all instances brought up-to-date.
+
+
 class PoolLifetimeOption(str, Enum):
 
     job_schedule = "jobschedule"  #: The Pool exists for the lifetime of the Job Schedule. The Batch Service creates the Pool when it creates the first Job on the schedule. You may apply this option only to Job Schedules, not to Jobs.
@@ -310,6 +317,7 @@ class ComputeNodeState(str, Enum):
     leaving_pool = "leavingpool"  #: The Compute Node is leaving the Pool, either because the user explicitly removed it or because the Pool is resizing or autoscaling down.
     offline = "offline"  #: The Compute Node is not currently running a Task, and scheduling of new Tasks to the Compute Node is disabled.
     preempted = "preempted"  #: The Spot/Low-priority Compute Node has been preempted. Tasks which were running on the Compute Node when it was preempted will be rescheduled when another Compute Node becomes available.
+    upgrading_os = "upgradingos"  #: The Compute Node is undergoing an OS upgrade operation.
 
 
 class SchedulingState(str, Enum):