[POSTGRESQL] az postgresql flexible-server replica create: Select storage type PremiumV2_LRS for replica server (#33134)

Author: nasc17

src/azure-cli/azure/cli/command_modules/postgresql/_params.py

@@ -180,10 +180,16 @@ def _flexible_server_params(command_group):
                  'Must set iops and throughput if using PremiumV2_LRS.'
         )
 
+        storage_type_replica_arg_type = CLIArgumentType(
+            arg_type=get_enum_type(['PremiumV2_LRS']),
+            options_list=['--storage-type'],
+            help='Storage type for the read replica. Allowed value is PremiumV2_LRS. Default is for the read replica to match storage type of the primary server.'
+        )
+
         storage_type_restore_arg_type = CLIArgumentType(
             arg_type=get_enum_type(['PremiumV2_LRS']),
             options_list=['--storage-type'],
-            help='Storage type for the new server. Allowed value is PremiumV2_LRS. Default value is none.'
+            help='Storage type for the new server. Allowed value is PremiumV2_LRS. Default is for the new server to match storage type of the source server.'
         )
 
         performance_tier_arg_type = CLIArgumentType(
@@ -629,6 +635,7 @@ def _flexible_server_params(command_group):
             c.argument('performance_tier', default=None, arg_type=performance_tier_arg_type)
             c.argument('yes', arg_type=yes_arg_type)
             c.argument('tags', arg_type=tags_type)
+            c.argument('storage_type', default=None, arg_type=storage_type_replica_arg_type)
 
         with self.argument_context('{} flexible-server replica promote'.format(command_group)) as c:
             c.argument('replica_name', arg_type=replica_name_arg_type)

src/azure-cli/azure/cli/command_modules/postgresql/commands/replica_commands.py

@@ -33,7 +33,7 @@ def flexible_replica_create(cmd, client, resource_group_name, source_server, rep
                             location=None, vnet=None, vnet_address_prefix=None, subnet=None,
                             subnet_address_prefix=None, private_dns_zone_arguments=None, no_wait=False,
                             byok_identity=None, byok_key=None,
-                            sku_name=None, tier=None,
+                            sku_name=None, tier=None, storage_type=None,
                             storage_gb=None, performance_tier=None, yes=False, tags=None):
     validate_resource_group(resource_group_name)
 
@@ -119,7 +119,7 @@ def flexible_replica_create(cmd, client, resource_group_name, source_server, rep
 
     parameters.sku = postgresql_flexibleservers.models.Sku(name=sku_name, tier=tier)
 
-    parameters.storage = postgresql_flexibleservers.models.Storage(storage_size_gb=storage_gb, auto_grow=source_server_object.storage.auto_grow, tier=performance_tier)
+    parameters.storage = postgresql_flexibleservers.models.Storage(storage_size_gb=storage_gb, auto_grow=source_server_object.storage.auto_grow, tier=performance_tier, type=storage_type)
 
     return sdk_no_wait(no_wait, client.begin_create_or_update, resource_group_name, replica_name, parameters)