[Redis] az redis create/update: Add --zonal-allocation-policy to support the way of selecting zones for cache instance (#30705)

* update redis module with 2024-11-01 api version

* resolve linter error

* update redis module with 2024-11-01 api version

* resolve linter error

* remove unused import

* remove unused import

* resolve integration test failures

* update redis module with 2024-11-01 api version

* resolve linter error

* remove unused import

* update redis module with 2024-11-01 api version

* resolve linter error

* remove unused import

* resolve integration test failures

Author: aandukuria

src/azure-cli/azure/cli/command_modules/network/tests/latest/recordings/test_private_endpoint_connection_acfr.yaml

@@ -58,6 +58,8 @@
     crafted: true
   - name: Configure the multiple zones for new Premium Azure Cache for Redis
     text: az redis create --location westus2 --name MyRedisCache --resource-group MyResourceGroup --sku Premium --vm-size p1 --zones 1 2
+  - name: Deploying a Premium Azure Cache for Redis with zones automatically allocated
+    text: az redis create --location westus2 --name MyRedisCache --resource-group MyResourceGroup --sku Premium --vm-size p1 --zonal-allocation-policy Automatic
     crafted: true
   - name: Configure the memory policies for the cache.
     text: az redis create --resource-group resourceGroupName --name cacheName --location westus2 --sku Standard --vm-size c0 --redis-configuration @"config_max-memory.json"
@@ -179,6 +181,8 @@
     crafted: true
   - name: Disable access keys authentication for Redis.
     text: az redis update --name MyRedisCache --resource-group MyResourceGroup --set "disableAccessKeyAuthentication=true"
+  - name: Updating a non zone redundant cache to zone redundant cache using automatic zonal allocation policy.
+    text: az redis update --name MyRedisCache --resource-group MyResourceGroup --set "zonalAllocationPolicy=Automatic"
 """
 
 helps['redis force-reboot'] = """

src/azure-cli/azure/cli/command_modules/network/tests/latest/recordings/test_private_link_resource_acfr.yaml

@@ -10,7 +10,7 @@
 
 
 def load_arguments(self, _):  # pylint: disable=too-many-statements
-    from azure.mgmt.redis.models import RebootType, RedisKeyType, SkuName, TlsVersion, ReplicationRole, UpdateChannel
+    from azure.mgmt.redis.models import RebootType, RedisKeyType, SkuName, TlsVersion, ReplicationRole, UpdateChannel, ZonalAllocationPolicy
     from azure.cli.command_modules.redis._validators import JsonString, ScheduleEntryList
     from azure.cli.command_modules.redis.custom import allowed_c_family_sizes, allowed_p_family_sizes, allowed_auth_methods
     from azure.cli.core.commands.parameters import get_enum_type, tags_type, zones_type
@@ -54,6 +54,7 @@ def load_arguments(self, _):  # pylint: disable=too-many-statements
         c.argument('update_channel', arg_type=get_enum_type(UpdateChannel), help='Specifies the update channel for the monthly Redis updates your Redis Cache will receive. Caches using "Preview" update channel get latest Redis updates at least 4 weeks ahead of "Stable" channel caches. Default value is "Stable".')
         c.argument('storage_subscription_id', options_list=['--storage-subscription-id', '--storage-sub-id'], help='SubscriptionId of the storage account')
         c.argument('disable_access_key_authentication', arg_type=get_three_state_flag(), options_list=['--disable-access-keys'], help='Authentication to Redis through access keys is disabled when set as true')
+        c.argument('zonal_allocation_policy', options_list=['--zonal-allocation-policy', '--zonal-allocation'], arg_type=get_enum_type(ZonalAllocationPolicy), help='Specifies how availability zones are allocated to the Redis cache. "Automatic" enables zone redundancy and Azure will automatically select zones based on regional availability and capacity. "UserDefined" will select availability zones passed in by you using the "zones" parameter. "NoZones" will produce a non-zonal cache. If "zonal-allocation-policy" is not passed, it will be set to "UserDefined" when zones are passed in, otherwise, it will be set to "Automatic in regions where zones are supported and "NoZones" in regions where zones are not supported.')
 
     with self.argument_context('redis firewall-rules list') as c:
         c.argument('cache_name', arg_type=cache_name, id_part=None)

src/azure-cli/azure/cli/command_modules/redis/_help.py

@@ -76,7 +76,8 @@ def cli_redis_update(cmd, instance, sku=None, vm_size=None):
         sku=instance.sku,
         tags=instance.tags,
         update_channel=instance.update_channel,
-        disable_access_key_authentication=instance.disable_access_key_authentication
+        disable_access_key_authentication=instance.disable_access_key_authentication,
+        zonal_allocation_policy=instance.zonal_allocation_policy
     )
     return update_params
 
@@ -95,7 +96,8 @@ def cli_redis_create(cmd, client,
                      redis_configuration=None, enable_non_ssl_port=None, tenant_settings=None,
                      shard_count=None, minimum_tls_version=None, subnet_id=None, static_ip=None,
                      zones=None, replicas_per_master=None, redis_version=None, mi_system_assigned=None,
-                     mi_user_assigned=None, update_channel=None, disable_access_key_authentication=None):
+                     mi_user_assigned=None, update_channel=None, disable_access_key_authentication=None,
+                     zonal_allocation_policy=None):
     # pylint:disable=line-too-long
     if ((sku.lower() in ['standard', 'basic'] and vm_size.lower() not in allowed_c_family_sizes) or (
             sku.lower() in ['premium'] and vm_size.lower() not in allowed_p_family_sizes)):
@@ -126,7 +128,8 @@ def cli_redis_create(cmd, client,
         public_network_access=None,
         tags=tags,
         update_channel=update_channel,
-        disable_access_key_authentication=disable_access_key_authentication
+        disable_access_key_authentication=disable_access_key_authentication,
+        zonal_allocation_policy=zonal_allocation_policy
     )
     return client.begin_create(resource_group_name, name, params)