https://github.com/Azure/azure-cli/pull/27322/files

src/cosmosdb-preview/azext_cosmosdb_preview/_params.py

@@ -29,7 +29,7 @@
     CreatePhysicalPartitionIdListAction)
 
 from azext_cosmosdb_preview.vendored_sdks.azure_mgmt_cosmosdb.models import (
-    ContinuousTier, DefaultPriorityLevel
+    ContinuousTier, DefaultPriorityLevel, MinimalTlsVersion
 )
 
 from azure.cli.core.util import shell_safe_json_parse
@@ -307,6 +307,7 @@ def load_arguments(self, _):
         c.argument('databases_to_restore', nargs='+', action=CreateDatabaseRestoreResource, is_preview=True, arg_group='Restore')
         c.argument('gremlin_databases_to_restore', nargs='+', action=CreateGremlinDatabaseRestoreResource, is_preview=True, arg_group='Restore')
         c.argument('tables_to_restore', nargs='+', action=CreateTableRestoreResource, is_preview=True, arg_group='Restore')
+        c.argument('enable_burst_capacity', arg_type=get_three_state_flag(), help="Flag to enable burst capacity on the account.")
 
     for scope in ['cosmosdb create', 'cosmosdb update']:
         with self.argument_context(scope) as c:
@@ -337,10 +338,11 @@ def load_arguments(self, _):
             c.argument('backup_policy_type', arg_type=get_enum_type(BackupPolicyType), help="The type of backup policy of the account to create", arg_group='Backup Policy')
             c.argument('continuous_tier', arg_type=get_enum_type(ContinuousTier), help="The tier of Continuous backup", arg_group='Backup Policy')
             c.argument('enable_materialized_views', options_list=['--enable-materialized-views', '--enable-mv'], arg_type=get_three_state_flag(), help="Flag to enable MaterializedViews on the account.", is_preview=True)
-            c.argument('enable_burst_capacity', arg_type=get_three_state_flag(), help="Flag to enable burst capacity on the account.", is_preview=True)
+            c.argument('enable_burst_capacity', arg_type=get_three_state_flag(), help="Flag to enable burst capacity on the account.")
             c.argument('enable_priority_based_execution', options_list=['--enable-priority-based-execution', '--enable-pbe'], arg_type=get_three_state_flag(), help="Flag to enable priority based execution on the account.", is_preview=True)
             c.argument('default_priority_level', arg_type=get_enum_type(DefaultPriorityLevel), help="Default Priority Level of Request if not specified.", is_preview=True)
             c.argument('enable_per_region_per_partition_autoscale', arg_type=get_three_state_flag(), help="Enable or disable PerRegionPerPartitionAutoscale.", is_preview=True)
+            c.argument('minimal_tls_version', arg_type=get_enum_type(MinimalTlsVersion), help="Indicates the minimum allowed TLS version")
 
     with self.argument_context('cosmosdb update') as c:
         c.argument('key_uri', help="The URI of the key vault", is_preview=True)

src/cosmosdb-preview/azext_cosmosdb_preview/custom.py

@@ -780,7 +780,8 @@ def cli_cosmosdb_create(cmd,
                         enable_burst_capacity=None,
                         enable_priority_based_execution=None,
                         default_priority_level=None,
-                        enable_per_region_per_partition_autoscale=None):
+                        enable_per_region_per_partition_autoscale=None,
+                        minimal_tls_version=None):
     """Create a new Azure Cosmos DB database account."""
 
     from azure.cli.core.commands.client_factory import get_mgmt_service_client
@@ -837,7 +838,8 @@ def cli_cosmosdb_create(cmd,
                                     enable_burst_capacity=enable_burst_capacity,
                                     enable_priority_based_execution=enable_priority_based_execution,
                                     default_priority_level=default_priority_level,
-                                    enable_per_region_per_partition_autoscale=enable_per_region_per_partition_autoscale)
+                                    enable_per_region_per_partition_autoscale=enable_per_region_per_partition_autoscale,
+                                    minimal_tls_version=minimal_tls_version)
 
 
 # pylint: disable=too-many-branches
@@ -872,7 +874,8 @@ def cli_cosmosdb_update(client,
                         enable_burst_capacity=None,
                         enable_priority_based_execution=None,
                         default_priority_level=None,
-                        enable_per_region_per_partition_autoscale=None):
+                        enable_per_region_per_partition_autoscale=None,
+                        minimal_tls_version=None):
     """Update an existing Azure Cosmos DB database account. """
     existing = client.get(resource_group_name, account_name)
 
@@ -962,7 +965,8 @@ def cli_cosmosdb_update(client,
         enable_burst_capacity=enable_burst_capacity,
         enable_priority_based_execution=enable_priority_based_execution,
         default_priority_level=default_priority_level,
-        enable_per_region_per_partition_autoscale=enable_per_region_per_partition_autoscale)
+        enable_per_region_per_partition_autoscale=enable_per_region_per_partition_autoscale,
+        minimal_tls_version=minimal_tls_version)
 
     async_docdb_update = client.begin_update(resource_group_name, account_name, params)
     docdb_account = async_docdb_update.result()
@@ -1171,7 +1175,8 @@ def _create_database_account(client,
                              enable_priority_based_execution=None,
                              default_priority_level=None,
                              enable_per_region_per_partition_autoscale=None,
-                             disable_ttl=None):
+                             disable_ttl=None,
+                             minimal_tls_version=None):
     consistency_policy = None
     if default_consistency_level is not None:
         consistency_policy = ConsistencyPolicy(default_consistency_level=default_consistency_level,
@@ -1309,7 +1314,8 @@ def _create_database_account(client,
         enable_burst_capacity=enable_burst_capacity,
         enable_priority_based_execution=enable_priority_based_execution,
         default_priority_level=default_priority_level,
-        enable_per_region_per_partition_autoscale=enable_per_region_per_partition_autoscale
+        enable_per_region_per_partition_autoscale=enable_per_region_per_partition_autoscale,
+        minimal_tls_version=minimal_tls_version
     )
 
     async_docdb_create = client.begin_create_or_update(resource_group_name, account_name, params)