# coding: utf-8 # ------------------------------------------------------------------------- # Copyright (c) Microsoft Corporation. All rights reserved. # Licensed under the MIT License. See License.txt in the project root for # license information. # -------------------------------------------------------------------------- # TEST SCENARIO COVERAGE # ---------------------- # Methods Total : 132 # Methods Covered : 132 # Examples Total : 134 # Examples Tested : 134 # Coverage % : 100 # ---------------------- # Current Operation Coverage: # DatabaseAccounts: 18/18 # Operations: 1/1 # Database: 3/3 # Collection: 3/3 # CollectionRegion: 1/1 # DatabaseAccountRegion: 1/1 # PercentileSourceTarget: 1/1 # PercentileTarget: 1/1 # Percentile: 1/1 # CollectionPartitionRegion: 1/1 # CollectionPartition: 2/2 # PartitionKeyRangeId: 1/1 # PartitionKeyRangeIdRegion: 1/1 # SqlResources: 28/28 # MongoDBResources: 16/16 # TableResources: 8/8 # CassandraResources: 16/16 # GremlinResources: 16/16 # NotebookWorkspaces: 7/7 # PrivateLinkResources: 2/2 # PrivateEndpointConnections: 4/4 import unittest import azure.mgmt.cosmosdb from devtools_testutils import AzureMgmtTestCase, RandomNameResourceGroupPreparer AZURE_LOCATION = "eastus" class MgmtCosmosDBTest(AzureMgmtTestCase): def setUp(self): super(MgmtCosmosDBTest, self).setUp() self.mgmt_client = self.create_mgmt_client(azure.mgmt.cosmosdb.CosmosDBManagementClient) if self.is_live: from azure.mgmt.network import NetworkManagementClient self.network_client = self.create_mgmt_client(NetworkManagementClient) def create_virtual_network(self, group_name, location, network_name, subnet_name): azure_operation_poller = self.network_client.virtual_networks.create_or_update( group_name, network_name, {"location": location, "address_space": {"address_prefixes": ["10.0.0.0/16"]}}, ) result_create = azure_operation_poller.result() async_subnet_creation = self.network_client.subnets.create_or_update( group_name, network_name, subnet_name, {"address_prefix": "10.0.0.0/24"} ) subnet_info = async_subnet_creation.result() return subnet_info @unittest.skip("hard to test") @RandomNameResourceGroupPreparer(location=AZURE_LOCATION) def test_cosmosdb(self, resource_group): SUBSCRIPTION_ID = self.settings.SUBSCRIPTION_ID TENANT_ID = self.settings.TENANT_ID RESOURCE_GROUP = resource_group.name ACCOUNT_NAME = "myAccount" VIRTUAL_NETWORK_NAME = "myVirtualNetwork" SUBNET_NAME = "mySubnet" DATABASE_RID = "myDatabaseRid" COLLECTION_RID = "myCollectionRid" REGION = "myRegion" SOURCE_REGION = "mySourceRegion" TARGET_REGION = "myTargetRegion" PERCENTILE_NAME = "myPercentile" PARTITION_NAME = "myPartition" PARTITION_KEY_RANGE_ID = "myPartitionKeyRangeId" DATABASE_NAME = "myDatabase" THROUGHPUT_SETTING_NAME = "myThroughputSetting" CONTAINER_NAME = "myContainer" STORED_PROCEDURE_NAME = "myStoredProcedure" USER_DEFINED_FUNCTION_NAME = "myUserDefinedFunction" TRIGGER_NAME = "myTrigger" COLLECTION_NAME = "myCollection" TABLE_NAME = "myTable" KEYSPACE_NAME = "myKeyspace" GRAPH_NAME = "myGraph" NOTEBOOK_WORKSPACE_NAME = "myNotebookWorkspace" GROUP_NAME = "myGroup" PRIVATE_ENDPOINT_CONNECTION_NAME = "myPrivateEndpointConnection" # -------------------------------------------------------------------------- # /DatabaseAccounts/put/CosmosDBDatabaseAccountCreateMin[put] # -------------------------------------------------------------------------- BODY = { "location": AZURE_LOCATION, "database_account_offer_type": "Standard", "locations": [{"failover_priority": "0", "location_name": "southcentralus", "is_zone_redundant": False}], } result = self.mgmt_client.database_accounts.begin_create_or_update( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, create_update_parameters=BODY ) result = result.result() # -------------------------------------------------------------------------- # /DatabaseAccounts/put/CosmosDBDatabaseAccountCreateMax[put] # -------------------------------------------------------------------------- BODY = { "location": AZURE_LOCATION, "kind": "MongoDB", "database_account_offer_type": "Standard", "ip_rules": [{"ip_address_or_range": "23.43.230.120"}, {"ip_address_or_range": "110.12.240.0/12"}], "is_virtual_network_filter_enabled": True, "virtual_network_rules": [ { "id": "/subscriptions/" + SUBSCRIPTION_ID + "/resourceGroups/" + RESOURCE_GROUP + "/providers/Microsoft.Network/virtualNetworks/" + VIRTUAL_NETWORK_NAME + "/subnets/" + SUBNET_NAME, "ignore_missing_vnet_service_endpoint": False, } ], "locations": [ {"failover_priority": "0", "location_name": "southcentralus", "is_zone_redundant": False}, {"failover_priority": "1", "location_name": "eastus", "is_zone_redundant": False}, ], "consistency_policy": { "default_consistency_level": "BoundedStaleness", "max_interval_in_seconds": "10", "max_staleness_prefix": "200", }, "key_vault_key_uri": "https://myKeyVault.vault.azure.net", "enable_free_tier": False, "api_properties": {"server_version": "3.2"}, "enable_analytical_storage": True, "cors": [{"allowed_origins": "https://test"}], } # result = self.mgmt_client.database_accounts.begin_create_or_update(resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, create_update_parameters=BODY) # result = result.result() # -------------------------------------------------------------------------- # /TableResources/put/CosmosDBTableReplace[put] # -------------------------------------------------------------------------- BODY = {"location": AZURE_LOCATION, "resource": {"id": "tableName"}} result = self.mgmt_client.table_resources.begin_create_update_table( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, table_name=TABLE_NAME, create_update_table_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /SqlResources/put/CosmosDBSqlDatabaseCreateUpdate[put] # -------------------------------------------------------------------------- BODY = {"location": AZURE_LOCATION, "resource": {"id": "databaseName"}} result = self.mgmt_client.sql_resources.begin_create_update_sql_database( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, create_update_sql_database_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /MongoDBResources/put/CosmosDBMongoDBDatabaseCreateUpdate[put] # -------------------------------------------------------------------------- BODY = {"location": AZURE_LOCATION, "resource": {"id": "databaseName"}} result = self.mgmt_client.mongo_dbresources.begin_create_update_mongo_dbdatabase( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, create_update_mongo_dbdatabase_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /GremlinResources/put/CosmosDBGremlinDatabaseCreateUpdate[put] # -------------------------------------------------------------------------- BODY = {"location": AZURE_LOCATION, "resource": {"id": "databaseName"}} result = self.mgmt_client.gremlin_resources.begin_create_update_gremlin_database( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, create_update_gremlin_database_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /CassandraResources/put/CosmosDBCassandraKeyspaceCreateUpdate[put] # -------------------------------------------------------------------------- BODY = {"location": AZURE_LOCATION, "resource": {"id": "keyspaceName"}} result = self.mgmt_client.cassandra_resources.begin_create_update_cassandra_keyspace( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, keyspace_name=KEYSPACE_NAME, create_update_cassandra_keyspace_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /NotebookWorkspaces/put/CosmosDBNotebookWorkspaceCreate[put] # -------------------------------------------------------------------------- BODY = {} result = self.mgmt_client.notebook_workspaces.begin_create_or_update( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, notebook_workspace_name=NOTEBOOK_WORKSPACE_NAME, notebook_create_update_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /GremlinResources/put/CosmosDBGremlinGraphCreateUpdate[put] # -------------------------------------------------------------------------- BODY = { "location": AZURE_LOCATION, "resource": { "id": "graphName", "indexing_policy": { "indexing_mode": "Consistent", "automatic": True, "included_paths": [ { "path": "/*", "indexes": [ {"kind": "Range", "data_type": "String", "precision": "-1"}, {"kind": "Range", "data_type": "Number", "precision": "-1"}, ], } ], "excluded_paths": [], }, "partition_key": {"paths": ["/AccountNumber"], "kind": "Hash"}, "default_ttl": "100", "unique_key_policy": {"unique_keys": [{"paths": ["/testPath"]}]}, "conflict_resolution_policy": {"mode": "LastWriterWins", "conflict_resolution_path": "/path"}, }, } result = self.mgmt_client.gremlin_resources.begin_create_update_gremlin_graph( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, graph_name=GRAPH_NAME, create_update_gremlin_graph_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /CassandraResources/put/CosmosDBCassandraTableCreateUpdate[put] # -------------------------------------------------------------------------- BODY = { "location": AZURE_LOCATION, "resource": { "id": "tableName", "default_ttl": "100", "analytical_storage_ttl": "500", "schema": { "columns": [{"name": "columnA", "type": "Ascii"}], "partition_keys": [{"name": "columnA"}], "cluster_keys": [{"name": "columnA", "order_by": "Asc"}], }, }, } result = self.mgmt_client.cassandra_resources.begin_create_update_cassandra_table( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, keyspace_name=KEYSPACE_NAME, table_name=TABLE_NAME, create_update_cassandra_table_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /SqlResources/put/CosmosDBSqlContainerCreateUpdate[put] # -------------------------------------------------------------------------- BODY = { "location": AZURE_LOCATION, "resource": { "id": "containerName", "indexing_policy": { "indexing_mode": "Consistent", "automatic": True, "included_paths": [ { "path": "/*", "indexes": [ {"kind": "Range", "data_type": "String", "precision": "-1"}, {"kind": "Range", "data_type": "Number", "precision": "-1"}, ], } ], "excluded_paths": [], }, "partition_key": {"paths": ["/AccountNumber"], "kind": "Hash"}, "default_ttl": "100", "unique_key_policy": {"unique_keys": [{"paths": ["/testPath"]}]}, "conflict_resolution_policy": {"mode": "LastWriterWins", "conflict_resolution_path": "/path"}, }, } result = self.mgmt_client.sql_resources.begin_create_update_sql_container( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, container_name=CONTAINER_NAME, create_update_sql_container_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /PrivateEndpointConnections/put/Approve or reject a private endpoint connection with a given name.[put] # -------------------------------------------------------------------------- BODY = { "private_link_service_connection_state": { "status": "Approved", "description": "Approved by johndoe@contoso.com", } } result = self.mgmt_client.private_endpoint_connections.begin_create_or_update( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, private_endpoint_connection_name=PRIVATE_ENDPOINT_CONNECTION_NAME, parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /MongoDBResources/put/CosmosDBMongoDBCollectionCreateUpdate[put] # -------------------------------------------------------------------------- BODY = { "location": AZURE_LOCATION, "resource": { "id": "collectionName", "indexes": [{"key": {"keys": ["testKey"]}, "options": {"expire_after_seconds": "100", "unique": True}}], "shard_key": {"test_key": "Hash"}, "analytical_storage_ttl": "500", }, } result = self.mgmt_client.mongo_dbresources.begin_create_update_mongo_dbcollection( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, collection_name=COLLECTION_NAME, create_update_mongo_dbcollection_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /TableResources/put/CosmosDBTableThroughputUpdate[put] # -------------------------------------------------------------------------- BODY = {"location": AZURE_LOCATION, "resource": {"throughput": "400"}} result = self.mgmt_client.table_resources.begin_update_table_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, table_name=TABLE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, update_throughput_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /SqlResources/put/CosmosDBSqlDatabaseThroughputUpdate[put] # -------------------------------------------------------------------------- BODY = {"location": AZURE_LOCATION, "resource": {"throughput": "400"}} result = self.mgmt_client.sql_resources.begin_update_sql_database_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, update_throughput_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /MongoDBResources/put/CosmosDBMongoDBDatabaseThroughputUpdate[put] # -------------------------------------------------------------------------- BODY = {"location": AZURE_LOCATION, "resource": {"throughput": "400"}} result = self.mgmt_client.mongo_dbresources.begin_update_mongo_dbdatabase_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, update_throughput_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /GremlinResources/put/CosmosDBGremlinDatabaseThroughputUpdate[put] # -------------------------------------------------------------------------- BODY = {"location": AZURE_LOCATION, "resource": {"throughput": "400"}} result = self.mgmt_client.gremlin_resources.begin_update_gremlin_database_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, update_throughput_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /CassandraResources/put/CosmosDBCassandraKeyspaceThroughputUpdate[put] # -------------------------------------------------------------------------- BODY = {"location": AZURE_LOCATION, "resource": {"throughput": "400"}} result = self.mgmt_client.cassandra_resources.begin_update_cassandra_keyspace_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, keyspace_name=KEYSPACE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, update_throughput_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /SqlResources/put/CosmosDBSqlTriggerCreateUpdate[put] # -------------------------------------------------------------------------- BODY = { "resource": { "id": "triggerName", "body": "body", "trigger_type": "triggerType", "trigger_operation": "triggerOperation", } } result = self.mgmt_client.sql_resources.begin_create_update_sql_trigger( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, container_name=CONTAINER_NAME, trigger_name=TRIGGER_NAME, create_update_sql_trigger_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /GremlinResources/put/CosmosDBGremlinGraphThroughputUpdate[put] # -------------------------------------------------------------------------- BODY = {"location": AZURE_LOCATION, "resource": {"throughput": "400"}} result = self.mgmt_client.gremlin_resources.begin_update_gremlin_graph_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, graph_name=GRAPH_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, update_throughput_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /SqlResources/put/CosmosDBSqlStoredProcedureCreateUpdate[put] # -------------------------------------------------------------------------- BODY = {"resource": {"id": "storedProcedureName", "body": "body"}} result = self.mgmt_client.sql_resources.begin_create_update_sql_stored_procedure( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, container_name=CONTAINER_NAME, stored_procedure_name=STORED_PROCEDURE_NAME, create_update_sql_stored_procedure_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /CassandraResources/put/CosmosDBCassandraTableThroughputUpdate[put] # -------------------------------------------------------------------------- BODY = {"location": AZURE_LOCATION, "resource": {"throughput": "400"}} result = self.mgmt_client.cassandra_resources.begin_update_cassandra_table_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, keyspace_name=KEYSPACE_NAME, table_name=TABLE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, update_throughput_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /SqlResources/put/CosmosDBSqlContainerThroughputUpdate[put] # -------------------------------------------------------------------------- BODY = {"location": AZURE_LOCATION, "resource": {"throughput": "400"}} result = self.mgmt_client.sql_resources.begin_update_sql_container_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, container_name=CONTAINER_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, update_throughput_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /SqlResources/put/CosmosDBSqlUserDefinedFunctionCreateUpdate[put] # -------------------------------------------------------------------------- BODY = {"resource": {"id": "userDefinedFunctionName", "body": "body"}} result = self.mgmt_client.sql_resources.begin_create_update_sql_user_defined_function( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, container_name=CONTAINER_NAME, user_defined_function_name=USER_DEFINED_FUNCTION_NAME, create_update_sql_user_defined_function_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /MongoDBResources/put/CosmosDBMongoDBCollectionThroughputUpdate[put] # -------------------------------------------------------------------------- BODY = {"location": AZURE_LOCATION, "resource": {"throughput": "400"}} result = self.mgmt_client.mongo_dbresources.begin_update_mongo_dbcollection_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, collection_name=COLLECTION_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, update_throughput_parameters=BODY, ) result = result.result() # -------------------------------------------------------------------------- # /PartitionKeyRangeIdRegion/get/CosmosDBDatabaseAccountRegionGetMetrics[get] # -------------------------------------------------------------------------- result = self.mgmt_client.partition_key_range_id_region.list_metrics( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, region=REGION, database_rid=DATABASE_RID, collection_rid=COLLECTION_RID, partition_key_range_id=PARTITION_KEY_RANGE_ID, filter="$filter=(name.value eq 'Max RUs Per Second') and timeGrain eq duration'PT1M' and startTime eq '2017-11-19T23:53:55.2780000Z' and endTime eq '2017-11-20T23:58:55.2780000Z", ) # -------------------------------------------------------------------------- # /MongoDBResources/get/CosmosDBMongoDBCollectionThroughputGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.mongo_dbresources.get_mongo_dbcollection_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, collection_name=COLLECTION_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) # -------------------------------------------------------------------------- # /SqlResources/get/CosmosDBSqlUserDefinedFunctionGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.sql_resources.get_sql_user_defined_function( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, container_name=CONTAINER_NAME, user_defined_function_name=USER_DEFINED_FUNCTION_NAME, ) # -------------------------------------------------------------------------- # /PartitionKeyRangeId/get/CosmosDBDatabaseAccountRegionGetMetrics[get] # -------------------------------------------------------------------------- result = self.mgmt_client.partition_key_range_id.list_metrics( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_rid=DATABASE_RID, collection_rid=COLLECTION_RID, partition_key_range_id=PARTITION_KEY_RANGE_ID, filter="$filter=(name.value eq 'Max RUs Per Second') and timeGrain eq duration'PT1M' and startTime eq '2017-11-19T23:53:55.2780000Z' and endTime eq '2017-11-20T23:58:55.2780000Z", ) # -------------------------------------------------------------------------- # /CollectionPartitionRegion/get/CosmosDBDatabaseAccountRegionGetMetrics[get] # -------------------------------------------------------------------------- result = self.mgmt_client.collection_partition_region.list_metrics( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, region=REGION, database_rid=DATABASE_RID, collection_rid=COLLECTION_RID, partition_name=PARTITION_NAME, filter="$filter=(name.value eq 'Max RUs Per Second') and timeGrain eq duration'PT1M' and startTime eq '2017-11-19T23:53:55.2780000Z' and endTime eq '2017-11-20T23:58:55.2780000Z", ) # -------------------------------------------------------------------------- # /SqlResources/get/CosmosDBSqlContainerThroughputGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.sql_resources.get_sql_container_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, container_name=CONTAINER_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) # -------------------------------------------------------------------------- # /CassandraResources/get/CosmosDBCassandraTableThroughputGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.cassandra_resources.get_cassandra_table_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, keyspace_name=KEYSPACE_NAME, table_name=TABLE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) # -------------------------------------------------------------------------- # /SqlResources/get/CosmosDBSqlStoredProcedureGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.sql_resources.get_sql_stored_procedure( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, container_name=CONTAINER_NAME, stored_procedure_name=STORED_PROCEDURE_NAME, ) # -------------------------------------------------------------------------- # /GremlinResources/get/CosmosDBGremlinGraphThroughputGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.gremlin_resources.get_gremlin_graph_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, graph_name=GRAPH_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) # -------------------------------------------------------------------------- # /PercentileSourceTarget/get/CosmosDBDatabaseAccountRegionGetMetrics[get] # -------------------------------------------------------------------------- result = self.mgmt_client.percentile_source_target.list_metrics( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, source_region=SOURCE_REGION, target_region=TARGET_REGION, percentile_name=PERCENTILE_NAME, filter="$filter=(name.value eq 'Probabilistic Bounded Staleness') and timeGrain eq duration'PT5M' and startTime eq '2017-11-19T23:53:55.2780000Z' and endTime eq '2017-11-20T00:13:55.2780000Z", ) # -------------------------------------------------------------------------- # /CollectionPartition/get/CosmosDBDatabaseAccountRegionGetMetrics[get] # -------------------------------------------------------------------------- result = self.mgmt_client.collection_partition.list_metrics( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_rid=DATABASE_RID, collection_rid=COLLECTION_RID, partition_name=PARTITION_NAME, filter="$filter=(name.value eq 'Max RUs Per Second') and timeGrain eq duration'PT1M' and startTime eq '2017-11-19T23:53:55.2780000Z' and endTime eq '2017-11-20T23:58:55.2780000Z", ) # -------------------------------------------------------------------------- # /CollectionPartition/get/CosmosDBCollectionGetUsages[get] # -------------------------------------------------------------------------- result = self.mgmt_client.collection_partition.list_usages( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_rid=DATABASE_RID, collection_rid=COLLECTION_RID, partition_name=PARTITION_NAME, filter="$filter=name.value eq 'Partition Storage'", ) # -------------------------------------------------------------------------- # /SqlResources/get/CosmosDBSqlTriggerGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.sql_resources.get_sql_trigger( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, container_name=CONTAINER_NAME, trigger_name=TRIGGER_NAME, ) # -------------------------------------------------------------------------- # /CollectionRegion/get/CosmosDBRegionCollectionGetMetrics[get] # -------------------------------------------------------------------------- result = self.mgmt_client.collection_region.list_metrics( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, region=REGION, database_rid=DATABASE_RID, collection_rid=COLLECTION_RID, filter="$filter=(name.value eq 'Total Requests') and timeGrain eq duration'PT5M' and startTime eq '2017-11-19T23:53:55.2780000Z' and endTime eq '2017-11-20T00:13:55.2780000Z", ) # -------------------------------------------------------------------------- # /CassandraResources/get/CosmosDBCassandraKeyspaceThroughputGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.cassandra_resources.get_cassandra_keyspace_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, keyspace_name=KEYSPACE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) # -------------------------------------------------------------------------- # /GremlinResources/get/CosmosDBGremlinDatabaseThroughputGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.gremlin_resources.get_gremlin_database_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) # -------------------------------------------------------------------------- # /SqlResources/get/CosmosDBSqlUserDefinedFunctionList[get] # -------------------------------------------------------------------------- result = self.mgmt_client.sql_resources.list_sql_user_defined_functions( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, container_name=CONTAINER_NAME, ) # -------------------------------------------------------------------------- # /MongoDBResources/get/CosmosDBMongoDBDatabaseThroughputGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.mongo_dbresources.get_mongo_dbdatabase_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) # -------------------------------------------------------------------------- # /SqlResources/get/CosmosDBSqlStoredProcedureList[get] # -------------------------------------------------------------------------- result = self.mgmt_client.sql_resources.list_sql_stored_procedures( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, container_name=CONTAINER_NAME, ) # -------------------------------------------------------------------------- # /SqlResources/get/CosmosDBSqlDatabaseThroughputGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.sql_resources.get_sql_database_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) # -------------------------------------------------------------------------- # /Collection/get/CosmosDBCollectionGetMetricDefinitions[get] # -------------------------------------------------------------------------- result = self.mgmt_client.collection.list_metric_definitions( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_rid=DATABASE_RID, collection_rid=COLLECTION_RID, ) # -------------------------------------------------------------------------- # /SqlResources/get/CosmosDBSqlTriggerList[get] # -------------------------------------------------------------------------- result = self.mgmt_client.sql_resources.list_sql_triggers( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, container_name=CONTAINER_NAME, ) # -------------------------------------------------------------------------- # /TableResources/get/CosmosDBTableThroughputGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.table_resources.get_table_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, table_name=TABLE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) # -------------------------------------------------------------------------- # /MongoDBResources/get/CosmosDBMongoDBCollectionGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.mongo_dbresources.get_mongo_dbcollection( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, collection_name=COLLECTION_NAME, ) # -------------------------------------------------------------------------- # /Collection/get/CosmosDBCollectionGetMetrics[get] # -------------------------------------------------------------------------- result = self.mgmt_client.collection.list_metrics( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_rid=DATABASE_RID, collection_rid=COLLECTION_RID, filter="$filter=(name.value eq 'Total Requests') and timeGrain eq duration'PT5M' and startTime eq '2017-11-19T23:53:55.2780000Z' and endTime eq '2017-11-20T00:13:55.2780000Z", ) # -------------------------------------------------------------------------- # /Collection/get/CosmosDBCollectionGetUsages[get] # -------------------------------------------------------------------------- result = self.mgmt_client.collection.list_usages( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_rid=DATABASE_RID, collection_rid=COLLECTION_RID, filter="$filter=name.value eq 'Storage'", ) # -------------------------------------------------------------------------- # /PercentileTarget/get/CosmosDBDatabaseAccountRegionGetMetrics[get] # -------------------------------------------------------------------------- result = self.mgmt_client.percentile_target.list_metrics( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, target_region=TARGET_REGION, percentile_name=PERCENTILE_NAME, filter="$filter=(name.value eq 'Probabilistic Bounded Staleness') and timeGrain eq duration'PT5M' and startTime eq '2017-11-19T23:53:55.2780000Z' and endTime eq '2017-11-20T00:13:55.2780000Z", ) # -------------------------------------------------------------------------- # /PrivateEndpointConnections/get/Gets private endpoint connection.[get] # -------------------------------------------------------------------------- result = self.mgmt_client.private_endpoint_connections.list_by_database_account( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME ) # -------------------------------------------------------------------------- # /SqlResources/get/CosmosDBSqlContainerGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.sql_resources.get_sql_container( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, container_name=CONTAINER_NAME, ) # -------------------------------------------------------------------------- # /CassandraResources/get/CosmosDBCassandraTableGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.cassandra_resources.get_cassandra_table( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, keyspace_name=KEYSPACE_NAME, table_name=TABLE_NAME, ) # -------------------------------------------------------------------------- # /GremlinResources/get/CosmosDBGremlinGraphGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.gremlin_resources.get_gremlin_graph( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, graph_name=GRAPH_NAME, ) # -------------------------------------------------------------------------- # /MongoDBResources/get/CosmosDBMongoDBCollectionList[get] # -------------------------------------------------------------------------- result = self.mgmt_client.mongo_dbresources.list_mongo_dbcollections( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME ) # -------------------------------------------------------------------------- # /NotebookWorkspaces/get/CosmosDBNotebookWorkspaceGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.notebook_workspaces.get( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, notebook_workspace_name=NOTEBOOK_WORKSPACE_NAME, ) # -------------------------------------------------------------------------- # /Database/get/CosmosDBDatabaseGetMetricDefinitions[get] # -------------------------------------------------------------------------- result = self.mgmt_client.database.list_metric_definitions( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_rid=DATABASE_RID ) # -------------------------------------------------------------------------- # /CassandraResources/get/CosmosDBCassandraTableList[get] # -------------------------------------------------------------------------- result = self.mgmt_client.cassandra_resources.list_cassandra_tables( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, keyspace_name=KEYSPACE_NAME ) # -------------------------------------------------------------------------- # /GremlinResources/get/CosmosDBGremlinGraphList[get] # -------------------------------------------------------------------------- result = self.mgmt_client.gremlin_resources.list_gremlin_graphs( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME ) # -------------------------------------------------------------------------- # /SqlResources/get/CosmosDBSqlContainerList[get] # -------------------------------------------------------------------------- result = self.mgmt_client.sql_resources.list_sql_containers( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME ) # -------------------------------------------------------------------------- # /CassandraResources/get/CosmosDBCassandraKeyspaceGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.cassandra_resources.get_cassandra_keyspace( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, keyspace_name=KEYSPACE_NAME ) # -------------------------------------------------------------------------- # /PrivateLinkResources/get/Gets private endpoint connection.[get] # -------------------------------------------------------------------------- result = self.mgmt_client.private_link_resources.list_by_database_account( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME ) # -------------------------------------------------------------------------- # /Database/get/CosmosDBDatabaseGetMetrics[get] # -------------------------------------------------------------------------- result = self.mgmt_client.database.list_metrics( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_rid=DATABASE_RID, filter="$filter=(name.value eq 'Total Requests') and timeGrain eq duration'PT5M' and startTime eq '2017-11-19T23:53:55.2780000Z' and endTime eq '2017-11-20T00:13:55.2780000Z", ) # -------------------------------------------------------------------------- # /GremlinResources/get/CosmosDBGremlinDatabaseGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.gremlin_resources.get_gremlin_database( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME ) # -------------------------------------------------------------------------- # /MongoDBResources/get/CosmosDBMongoDBDatabaseGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.mongo_dbresources.get_mongo_dbdatabase( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME ) # -------------------------------------------------------------------------- # /Database/get/CosmosDBDatabaseGetUsages[get] # -------------------------------------------------------------------------- result = self.mgmt_client.database.list_usages( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_rid=DATABASE_RID, filter="$filter=name.value eq 'Storage'", ) # -------------------------------------------------------------------------- # /Percentile/get/CosmosDBDatabaseAccountRegionGetMetrics[get] # -------------------------------------------------------------------------- result = self.mgmt_client.percentile.list_metrics( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, percentile_name=PERCENTILE_NAME, filter="$filter=(name.value eq 'Probabilistic Bounded Staleness') and timeGrain eq duration'PT5M' and startTime eq '2017-11-19T23:53:55.2780000Z' and endTime eq '2017-11-20T00:13:55.2780000Z", ) # -------------------------------------------------------------------------- # /SqlResources/get/CosmosDBSqlDatabaseGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.sql_resources.get_sql_database( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME ) # -------------------------------------------------------------------------- # /DatabaseAccountRegion/get/CosmosDBDatabaseAccountRegionGetMetrics[get] # -------------------------------------------------------------------------- result = self.mgmt_client.database_account_region.list_metrics( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, region=REGION, filter="$filter=(name.value eq 'Total Requests') and timeGrain eq duration'PT5M' and startTime eq '2017-11-19T23:53:55.2780000Z' and endTime eq '2017-11-20T00:13:55.2780000Z", ) # -------------------------------------------------------------------------- # /PrivateEndpointConnections/get/Gets private endpoint connection.[get] # -------------------------------------------------------------------------- result = self.mgmt_client.private_endpoint_connections.list_by_database_account( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME ) # -------------------------------------------------------------------------- # /TableResources/get/CosmosDBTableGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.table_resources.get_table( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, table_name=TABLE_NAME ) # -------------------------------------------------------------------------- # /PrivateLinkResources/get/Gets private endpoint connection.[get] # -------------------------------------------------------------------------- result = self.mgmt_client.private_link_resources.list_by_database_account( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME ) # -------------------------------------------------------------------------- # /CassandraResources/get/CosmosDBCassandraKeyspaceList[get] # -------------------------------------------------------------------------- result = self.mgmt_client.cassandra_resources.list_cassandra_keyspaces( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME ) # -------------------------------------------------------------------------- # /NotebookWorkspaces/get/CosmosDBNotebookWorkspaceList[get] # -------------------------------------------------------------------------- result = self.mgmt_client.notebook_workspaces.list_by_database_account( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME ) # -------------------------------------------------------------------------- # /DatabaseAccounts/get/CosmosDBDatabaseAccountGetMetricDefinitions[get] # -------------------------------------------------------------------------- result = self.mgmt_client.database_accounts.list_metric_definitions( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME ) # -------------------------------------------------------------------------- # /MongoDBResources/get/CosmosDBMongoDBDatabaseList[get] # -------------------------------------------------------------------------- result = self.mgmt_client.mongo_dbresources.list_mongo_dbdatabases( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME ) # -------------------------------------------------------------------------- # /GremlinResources/get/CosmosDBGremlinDatabaseList[get] # -------------------------------------------------------------------------- result = self.mgmt_client.gremlin_resources.list_gremlin_databases( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME ) # -------------------------------------------------------------------------- # /SqlResources/get/CosmosDBSqlDatabaseList[get] # -------------------------------------------------------------------------- result = self.mgmt_client.sql_resources.list_sql_databases( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME ) # -------------------------------------------------------------------------- # /DatabaseAccounts/get/CosmosDBDatabaseAccountListReadOnlyKeys[get] # -------------------------------------------------------------------------- result = self.mgmt_client.database_accounts.get_read_only_keys( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME ) # -------------------------------------------------------------------------- # /DatabaseAccounts/get/CosmosDBDatabaseAccountGetMetrics[get] # -------------------------------------------------------------------------- result = self.mgmt_client.database_accounts.list_metrics( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, filter="$filter=(name.value eq 'Total Requests') and timeGrain eq duration'PT5M' and startTime eq '2017-11-19T23:53:55.2780000Z' and endTime eq '2017-11-20T00:13:55.2780000Z", ) # -------------------------------------------------------------------------- # /DatabaseAccounts/get/CosmosDBDatabaseAccountGetUsages[get] # -------------------------------------------------------------------------- result = self.mgmt_client.database_accounts.list_usages( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, filter="$filter=name.value eq 'Storage'" ) # -------------------------------------------------------------------------- # /TableResources/get/CosmosDBTableList[get] # -------------------------------------------------------------------------- result = self.mgmt_client.table_resources.list_tables( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME ) # -------------------------------------------------------------------------- # /DatabaseAccounts/get/CosmosDBDatabaseAccountGet[get] # -------------------------------------------------------------------------- result = self.mgmt_client.database_accounts.get(resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME) # -------------------------------------------------------------------------- # /DatabaseAccounts/get/CosmosDBDatabaseAccountListByResourceGroup[get] # -------------------------------------------------------------------------- result = self.mgmt_client.database_accounts.list_by_resource_group(resource_group_name=RESOURCE_GROUP) # -------------------------------------------------------------------------- # /DatabaseAccounts/get/CosmosDBDatabaseAccountList[get] # -------------------------------------------------------------------------- result = self.mgmt_client.database_accounts.list() # -------------------------------------------------------------------------- # /Operations/get/CosmosDBOperationsList[get] # -------------------------------------------------------------------------- result = self.mgmt_client.operations.list() # -------------------------------------------------------------------------- # /MongoDBResources/post/CosmosDBMongoDBCollectionMigrateToManualThroughput[post] # -------------------------------------------------------------------------- result = self.mgmt_client.mongo_dbresources.begin_migrate_mongo_dbcollection_to_manual_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, collection_name=COLLECTION_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /SqlResources/post/CosmosDBSqlContainerMigrateToManualThroughput[post] # -------------------------------------------------------------------------- result = self.mgmt_client.sql_resources.begin_migrate_sql_container_to_manual_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, container_name=CONTAINER_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /MongoDBResources/post/CosmosDBMongoDBCollectionMigrateToAutoscale[post] # -------------------------------------------------------------------------- result = self.mgmt_client.mongo_dbresources.begin_migrate_mongo_dbcollection_to_autoscale( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, collection_name=COLLECTION_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /CassandraResources/post/CosmosDBCassandraTableMigrateToManualThroughput[post] # -------------------------------------------------------------------------- result = self.mgmt_client.cassandra_resources.begin_migrate_cassandra_table_to_manual_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, keyspace_name=KEYSPACE_NAME, table_name=TABLE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /GremlinResources/post/CosmosDBGremlinGraphMigrateToManualThroughput[post] # -------------------------------------------------------------------------- result = self.mgmt_client.gremlin_resources.begin_migrate_gremlin_graph_to_manual_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, graph_name=GRAPH_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /SqlResources/post/CosmosDBSqlContainerMigrateToAutoscale[post] # -------------------------------------------------------------------------- result = self.mgmt_client.sql_resources.begin_migrate_sql_container_to_autoscale( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, container_name=CONTAINER_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /CassandraResources/post/CosmosDBCassandraTableMigrateToAutoscale[post] # -------------------------------------------------------------------------- result = self.mgmt_client.cassandra_resources.begin_migrate_cassandra_table_to_autoscale( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, keyspace_name=KEYSPACE_NAME, table_name=TABLE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /GremlinResources/post/CosmosDBGremlinGraphMigrateToAutoscale[post] # -------------------------------------------------------------------------- result = self.mgmt_client.gremlin_resources.begin_migrate_gremlin_graph_to_autoscale( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, graph_name=GRAPH_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /CassandraResources/post/CosmosDBCassandraKeyspaceMigrateToManualThroughput[post] # -------------------------------------------------------------------------- result = self.mgmt_client.cassandra_resources.begin_migrate_cassandra_keyspace_to_manual_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, keyspace_name=KEYSPACE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /MongoDBResources/post/CosmosDBMongoDBDatabaseMigrateToManualThroughput[post] # -------------------------------------------------------------------------- result = self.mgmt_client.mongo_dbresources.begin_migrate_mongo_dbdatabase_to_manual_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /GremlinResources/post/CosmosDBGremlinDatabaseMigrateToManualThroughput[post] # -------------------------------------------------------------------------- result = self.mgmt_client.gremlin_resources.begin_migrate_gremlin_database_to_manual_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /SqlResources/post/CosmosDBSqlDatabaseMigrateToManualThroughput[post] # -------------------------------------------------------------------------- result = self.mgmt_client.sql_resources.begin_migrate_sql_database_to_manual_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /CassandraResources/post/CosmosDBCassandraKeyspaceMigrateToAutoscale[post] # -------------------------------------------------------------------------- result = self.mgmt_client.cassandra_resources.begin_migrate_cassandra_keyspace_to_autoscale( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, keyspace_name=KEYSPACE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /MongoDBResources/post/CosmosDBMongoDBDatabaseMigrateToAutoscale[post] # -------------------------------------------------------------------------- result = self.mgmt_client.mongo_dbresources.begin_migrate_mongo_dbdatabase_to_autoscale( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /GremlinResources/post/CosmosDBGremlinDatabaseMigrateToAutoscale[post] # -------------------------------------------------------------------------- result = self.mgmt_client.gremlin_resources.begin_migrate_gremlin_database_to_autoscale( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /SqlResources/post/CosmosDBSqlDatabaseMigrateToAutoscale[post] # -------------------------------------------------------------------------- result = self.mgmt_client.sql_resources.begin_migrate_sql_database_to_autoscale( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /TableResources/post/CosmosDBTableMigrateToManualThroughput[post] # -------------------------------------------------------------------------- result = self.mgmt_client.table_resources.begin_migrate_table_to_manual_throughput( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, table_name=TABLE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /TableResources/post/CosmosDBTableMigrateToAutoscale[post] # -------------------------------------------------------------------------- result = self.mgmt_client.table_resources.begin_migrate_table_to_autoscale( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, table_name=TABLE_NAME, throughput_setting_name=THROUGHPUT_SETTING_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /NotebookWorkspaces/post/CosmosDBNotebookWorkspaceRegenerateAuthToken[post] # -------------------------------------------------------------------------- result = self.mgmt_client.notebook_workspaces.begin_regenerate_auth_token( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, notebook_workspace_name=NOTEBOOK_WORKSPACE_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /NotebookWorkspaces/post/CosmosDBNotebookWorkspaceListConnectionInfo[post] # -------------------------------------------------------------------------- result = self.mgmt_client.notebook_workspaces.list_connection_info( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, notebook_workspace_name=NOTEBOOK_WORKSPACE_NAME, ) # -------------------------------------------------------------------------- # /NotebookWorkspaces/post/CosmosDBNotebookWorkspaceStart[post] # -------------------------------------------------------------------------- result = self.mgmt_client.notebook_workspaces.begin_start( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, notebook_workspace_name=NOTEBOOK_WORKSPACE_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /DatabaseAccounts/post/CosmosDBDatabaseAccountFailoverPriorityChange[post] # -------------------------------------------------------------------------- BODY = { "failover_policies": [ {"location_name": "eastus", "failover_priority": "0"}, {"location_name": "westus", "failover_priority": "1"}, ] } result = self.mgmt_client.database_accounts.begin_failover_priority_change( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, failover_parameters=BODY ) result = result.result() # -------------------------------------------------------------------------- # /DatabaseAccounts/post/CosmosDBDatabaseAccountListConnectionStrings[post] # -------------------------------------------------------------------------- result = self.mgmt_client.database_accounts.list_connection_strings( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME ) # -------------------------------------------------------------------------- # /DatabaseAccounts/post/CosmosDBDatabaseAccountListConnectionStringsMongo[post] # -------------------------------------------------------------------------- result = self.mgmt_client.database_accounts.list_connection_strings( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME ) # -------------------------------------------------------------------------- # /DatabaseAccounts/post/CosmosDBDatabaseAccountOfflineRegion[post] # -------------------------------------------------------------------------- [{"region": "North Europe"}] result = self.mgmt_client.database_accounts.begin_offline_region( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, region_parameter_for_offline=BODY ) result = result.result() # -------------------------------------------------------------------------- # /DatabaseAccounts/post/CosmosDBDatabaseAccountRegenerateKey[post] # -------------------------------------------------------------------------- BODY = {"key_kind": "primary"} result = self.mgmt_client.database_accounts.begin_regenerate_key( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, key_to_regenerate=BODY ) result = result.result() # -------------------------------------------------------------------------- # /DatabaseAccounts/post/CosmosDBDatabaseAccountListReadOnlyKeys[post] # -------------------------------------------------------------------------- result = self.mgmt_client.database_accounts.list_read_only_keys( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME ) # -------------------------------------------------------------------------- # /DatabaseAccounts/post/CosmosDBDatabaseAccountOnlineRegion[post] # -------------------------------------------------------------------------- [{"region": "North Europe"}] result = self.mgmt_client.database_accounts.begin_online_region( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, region_parameter_for_online=BODY ) result = result.result() # -------------------------------------------------------------------------- # /DatabaseAccounts/post/CosmosDBDatabaseAccountListKeys[post] # -------------------------------------------------------------------------- result = self.mgmt_client.database_accounts.list_keys( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME ) # -------------------------------------------------------------------------- # /DatabaseAccounts/patch/CosmosDBDatabaseAccountPatch[patch] # -------------------------------------------------------------------------- BODY = { "location": AZURE_LOCATION, "tags": {"dept": "finance"}, "ip_rules": [{"ip_address_or_range": "23.43.230.120"}, {"ip_address_or_range": "110.12.240.0/12"}], "is_virtual_network_filter_enabled": True, "virtual_network_rules": [ { "id": "/subscriptions/" + SUBSCRIPTION_ID + "/resourceGroups/" + RESOURCE_GROUP + "/providers/Microsoft.Network/virtualNetworks/" + VIRTUAL_NETWORK_NAME + "/subnets/" + SUBNET_NAME, "ignore_missing_vnet_service_endpoint": False, } ], "consistency_policy": { "default_consistency_level": "BoundedStaleness", "max_interval_in_seconds": "10", "max_staleness_prefix": "200", }, "enable_free_tier": False, "enable_analytical_storage": True, } result = self.mgmt_client.database_accounts.begin_update( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, update_parameters=BODY ) result = result.result() # -------------------------------------------------------------------------- # /DatabaseAccounts/head/CosmosDBDatabaseAccountCheckNameExists[head] # -------------------------------------------------------------------------- result = self.mgmt_client.database_accounts.check_name_exists(account_name=ACCOUNT_NAME) # -------------------------------------------------------------------------- # /SqlResources/delete/CosmosDBSqlUserDefinedFunctionDelete[delete] # -------------------------------------------------------------------------- result = self.mgmt_client.sql_resources.begin_delete_sql_user_defined_function( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, container_name=CONTAINER_NAME, user_defined_function_name=USER_DEFINED_FUNCTION_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /SqlResources/delete/CosmosDBSqlStoredProcedureDelete[delete] # -------------------------------------------------------------------------- result = self.mgmt_client.sql_resources.begin_delete_sql_stored_procedure( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, container_name=CONTAINER_NAME, stored_procedure_name=STORED_PROCEDURE_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /SqlResources/delete/CosmosDBSqlTriggerDelete[delete] # -------------------------------------------------------------------------- result = self.mgmt_client.sql_resources.begin_delete_sql_trigger( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, container_name=CONTAINER_NAME, trigger_name=TRIGGER_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /MongoDBResources/delete/CosmosDBMongoDBCollectionDelete[delete] # -------------------------------------------------------------------------- result = self.mgmt_client.mongo_dbresources.begin_delete_mongo_dbcollection( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, collection_name=COLLECTION_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /PrivateEndpointConnections/delete/Deletes a private endpoint connection with a given name.[delete] # -------------------------------------------------------------------------- result = self.mgmt_client.private_endpoint_connections.begin_delete( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, private_endpoint_connection_name=PRIVATE_ENDPOINT_CONNECTION_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /SqlResources/delete/CosmosDBSqlContainerDelete[delete] # -------------------------------------------------------------------------- result = self.mgmt_client.sql_resources.begin_delete_sql_container( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, container_name=CONTAINER_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /CassandraResources/delete/CosmosDBCassandraTableDelete[delete] # -------------------------------------------------------------------------- result = self.mgmt_client.cassandra_resources.begin_delete_cassandra_table( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, keyspace_name=KEYSPACE_NAME, table_name=TABLE_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /GremlinResources/delete/CosmosDBGremlinGraphDelete[delete] # -------------------------------------------------------------------------- result = self.mgmt_client.gremlin_resources.begin_delete_gremlin_graph( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME, graph_name=GRAPH_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /NotebookWorkspaces/delete/CosmosDBNotebookWorkspaceDelete[delete] # -------------------------------------------------------------------------- result = self.mgmt_client.notebook_workspaces.begin_delete( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, notebook_workspace_name=NOTEBOOK_WORKSPACE_NAME, ) result = result.result() # -------------------------------------------------------------------------- # /CassandraResources/delete/CosmosDBCassandraKeyspaceDelete[delete] # -------------------------------------------------------------------------- result = self.mgmt_client.cassandra_resources.begin_delete_cassandra_keyspace( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, keyspace_name=KEYSPACE_NAME ) result = result.result() # -------------------------------------------------------------------------- # /MongoDBResources/delete/CosmosDBMongoDBDatabaseDelete[delete] # -------------------------------------------------------------------------- result = self.mgmt_client.mongo_dbresources.begin_delete_mongo_dbdatabase( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME ) result = result.result() # -------------------------------------------------------------------------- # /GremlinResources/delete/CosmosDBGremlinDatabaseDelete[delete] # -------------------------------------------------------------------------- result = self.mgmt_client.gremlin_resources.begin_delete_gremlin_database( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME ) result = result.result() # -------------------------------------------------------------------------- # /SqlResources/delete/CosmosDBSqlDatabaseDelete[delete] # -------------------------------------------------------------------------- result = self.mgmt_client.sql_resources.begin_delete_sql_database( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, database_name=DATABASE_NAME ) result = result.result() # -------------------------------------------------------------------------- # /TableResources/delete/CosmosDBTableDelete[delete] # -------------------------------------------------------------------------- result = self.mgmt_client.table_resources.begin_delete_table( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME, table_name=TABLE_NAME ) result = result.result() # -------------------------------------------------------------------------- # /DatabaseAccounts/delete/CosmosDBDatabaseAccountDelete[delete] # -------------------------------------------------------------------------- result = self.mgmt_client.database_accounts.begin_delete( resource_group_name=RESOURCE_GROUP, account_name=ACCOUNT_NAME ) result = result.result() # ------------------------------------------------------------------------------ if __name__ == "__main__": unittest.main()