cli fixes

Author: dsapaliga

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

@@ -441,7 +441,7 @@ def load_arguments(self, _):
         c.argument('default_identity', help="The primary identity to access key vault in CMK related features. e.g. 'FirstPartyIdentity', 'SystemAssignedIdentity' and more.")
         c.argument('public_network_access', options_list=['--public-network-access', '-p'], arg_type=get_enum_type(['ENABLED', 'DISABLED']), help="Sets public network access in server to either Enabled or Disabled.")
         c.argument('disable_ttl', options_list=['--disable-ttl', '-d'], arg_type=get_three_state_flag(), help="Enable or disable restoring with ttl disabled.")
-        c.argument('source_backup_location', help="This is the location of the source account where backups are located. Provide this value if the source and target are in different locations.", is_preview=True)
+        c.argument('source_backup_location', help="This is the location of the source account where backups are located. Provide this value if the source and target are in different locations.")
 
     # Mongo role definition
     with self.argument_context('cosmosdb mongodb role definition') as c:

src/azure-cli/azure/cli/command_modules/cosmosdb/custom.py

@@ -367,6 +367,9 @@ def _create_database_account(client,
         if disable_ttl is not None:
             restore_parameters.restore_with_ttl_disabled = disable_ttl
 
+        if source_backup_location is not None:
+            restore_parameters.source_backup_location = source_backup_location
+
     params = DatabaseAccountCreateUpdateParameters(
         location=arm_location,
         locations=locations,
@@ -398,8 +401,7 @@ def _create_database_account(client,
         enable_per_region_per_partition_autoscale=enable_prpp_autoscale,
         minimal_tls_version=minimal_tls_version,
         enable_pbe=enable_pbe,
-        default_priority_level=default_priority_level,
-        source_backup_location=source_backup_location
+        default_priority_level=default_priority_level
     )
 
     async_docdb_create = client.begin_create_or_update(resource_group_name, account_name, params)