Update CosmosDB tests and include new recording for test_locations_database_accounts_offline

Author: Sheril Maria Lawrence

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

@@ -309,6 +309,15 @@
     crafted: true
 """
 
+helps['cosmosdb offline-region'] = """
+type: command
+short-summary: Offline the specified region for the specified Azure Cosmos DB database account.
+examples:
+  - name: Offlines North Europe regional account for the Azure Cosmos DB database account MyCosmosDBDatabaseAccount.
+    text: az cosmosdb offline-region --name MyCosmosDBDatabaseAccount --resource-group MyResourceGroup --region NorthEurope
+    crafted: true
+"""
+
 helps['cosmosdb gremlin'] = """
 type: group
 short-summary: Manage Gremlin resources of Azure Cosmos DB account.

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

@@ -142,6 +142,9 @@ def load_arguments(self, _):
         c.argument('failover_parameters', options_list=['--failover-policies'], validator=validate_failover_policies,
                    help="space-separated failover policies in 'regionName=failoverPriority' format. Number of policies must match the number of regions the account is currently replicated. All regionName values must match those of the regions the account is currently replicated. All failoverPriority values must be unique. There must be one failoverPriority value zero (0) specified. All remaining failoverPriority values can be any positive integer and they don't have to be contiguos, neither written in any specific order. E.g eastus=0 westus=1", nargs='+')
 
+    with self.argument_context('cosmosdb offline-region') as c:
+        c.argument('region', help='The region to offline for the CosmosDB account.')
+
     with self.argument_context('cosmosdb network-rule list') as c:
         c.argument('account_name', id_part=None)
 

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

@@ -167,6 +167,7 @@ def load_command_table(self, _):
         g.command('check-name-exists', 'check_name_exists')
         g.custom_command('delete', 'cli_cosmosdb_delete', confirmation=True, supports_no_wait=True)
         g.command('failover-priority-change', 'begin_failover_priority_change')
+        g.custom_command('offline-region', 'cli_offline_region')
         g.custom_command('create', 'cli_cosmosdb_create', transform=transform_db_account_json_output)
         g.custom_command('update', 'cli_cosmosdb_update', transform=transform_db_account_json_output)
         g.custom_command('list', 'cli_cosmosdb_list', transform=transform_db_account_list_output)

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

@@ -72,7 +72,8 @@
     DedicatedGatewayType,
     ServiceType,
     MongoRoleDefinitionCreateUpdateParameters,
-    MongoUserDefinitionCreateUpdateParameters
+    MongoUserDefinitionCreateUpdateParameters,
+    RegionForOnlineOffline
 )
 
 logger = get_logger(__name__)
@@ -3523,3 +3524,16 @@ def cli_cosmosdb_service_update(client,
     )
 
     return client.begin_create(resource_group_name, account_name, service_name, create_update_parameters=params)
+
+
+def cli_offline_region(client,
+                       account_name,
+                       resource_group_name,
+                       region):
+
+    region_parameter_for_offline = RegionForOnlineOffline(region=region)
+    return client.begin_offline_region(
+        resource_group_name=resource_group_name,
+        account_name=account_name,
+        region_parameter_for_offline=region_parameter_for_offline
+    )