{cosmosdb-preview} Update extension for managed identity (#5368)

* initial change * fix the sdk * update recordings * update tests
Azure · Sep 21, 2022 · eedf3f0 · eedf3f0
1 parent 981cc99
commit eedf3f0
Show file tree

Hide file tree

Showing 24 changed files with 5,692 additions and 8,284 deletions.
diff --git a/src/cosmosdb-preview/azext_cosmosdb_preview/custom.py b/src/cosmosdb-preview/azext_cosmosdb_preview/custom.py
@@ -43,7 +43,7 @@
 from azure.cli.core.azclierror import InvalidArgumentValueError
 from azure.core.exceptions import ResourceNotFoundError
 
-from azure.mgmt.cosmosdb.models import (
+from azext_cosmosdb_preview.vendored_sdks.azure_mgmt_cosmosdb.models import (
     Location,
     CreateMode,
     ConsistencyPolicy,

diff --git a/src/cosmosdb-preview/azext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_dts.yaml b/src/cosmosdb-preview/azext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_dts.yaml
diff --git a/...smosdb_preview/tests/latest/recordings/test_cosmosdb_gremlin_account_restore_command.yaml b/...smosdb_preview/tests/latest/recordings/test_cosmosdb_gremlin_account_restore_command.yaml
diff --git a/...b_preview/tests/latest/recordings/test_cosmosdb_gremlin_account_restore_using_create.yaml b/...b_preview/tests/latest/recordings/test_cosmosdb_gremlin_account_restore_using_create.yaml
diff --git a/...zext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_gremlin_graph_backupinfo.yaml b/...zext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_gremlin_graph_backupinfo.yaml
diff --git a/...t_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_gremlin_restorable_commands.yaml b/...t_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_gremlin_restorable_commands.yaml
diff --git a/...osdb-preview/azext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_mongo_role.yaml b/...osdb-preview/azext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_mongo_role.yaml
diff --git a/...cosmosdb_preview/tests/latest/recordings/test_cosmosdb_mongodb_collection_adaptiveru.yaml b/...cosmosdb_preview/tests/latest/recordings/test_cosmosdb_mongodb_collection_adaptiveru.yaml
diff --git a/...cosmosdb_preview/tests/latest/recordings/test_cosmosdb_mongodb_collection_backupinfo.yaml b/...cosmosdb_preview/tests/latest/recordings/test_cosmosdb_mongodb_collection_backupinfo.yaml
diff --git a/...zext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_mongodb_collection_merge.yaml b/...zext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_mongodb_collection_merge.yaml
diff --git a/...zext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_sql_container_adaptiveru.yaml b/...zext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_sql_container_adaptiveru.yaml
diff --git a/...zext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_sql_container_backupinfo.yaml b/...zext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_sql_container_backupinfo.yaml
diff --git a/...iew/azext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_sql_container_merge.yaml b/...iew/azext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_sql_container_merge.yaml
diff --git a/...ew/azext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_sql_continuous30days.yaml b/...ew/azext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_sql_continuous30days.yaml
diff --git a/...iew/azext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_sql_continuous7days.yaml b/...iew/azext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_sql_continuous7days.yaml
diff --git a/...review/tests/latest/recordings/test_cosmosdb_sql_migrate_periodic_to_continuous7days.yaml b/...review/tests/latest/recordings/test_cosmosdb_sql_migrate_periodic_to_continuous7days.yaml
diff --git a/...zext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_sql_oldestRestorableTime.yaml b/...zext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_sql_oldestRestorableTime.yaml
diff --git a/...cosmosdb_preview/tests/latest/recordings/test_cosmosdb_table_account_restore_command.yaml b/...cosmosdb_preview/tests/latest/recordings/test_cosmosdb_table_account_restore_command.yaml
diff --git a/...sdb_preview/tests/latest/recordings/test_cosmosdb_table_account_restore_using_create.yaml b/...sdb_preview/tests/latest/recordings/test_cosmosdb_table_account_restore_using_create.yaml
diff --git a/...review/azext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_table_backupinfo.yaml b/...review/azext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_table_backupinfo.yaml
diff --git a/...ext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_table_restorable_commands.yaml b/...ext_cosmosdb_preview/tests/latest/recordings/test_cosmosdb_table_restorable_commands.yaml
diff --git a/...b_preview/tests/latest/recordings/test_managed_cassandra_cluster_without_datacenters.yaml b/...b_preview/tests/latest/recordings/test_managed_cassandra_cluster_without_datacenters.yaml
diff --git a/...w/azext_cosmosdb_preview/tests/latest/recordings/test_managed_cassandra_verify_lists.yaml b/...w/azext_cosmosdb_preview/tests/latest/recordings/test_managed_cassandra_verify_lists.yaml
diff --git a/...osdb-preview/azext_cosmosdb_preview/tests/latest/test_cosmosdb_sql_adaptiveru_scenario.py b/...osdb-preview/azext_cosmosdb_preview/tests/latest/test_cosmosdb_sql_adaptiveru_scenario.py
@@ -23,8 +23,8 @@ def test_cosmosdb_sql_container_adaptiveru(self, resource_group):
         db_name = self.create_random_name(prefix='cli', length=15)
         # Assumption: There exists a cosmosTest rg with the account adrutest2. This test only creates the database and collection
         self.kwargs.update({
-            'rg' : 'cosmosTest',
-            'acc': 'adrutest2',            
+            'rg' : 'canary-sdk-test',
+            'acc': 'sqladru2',            
             'db_name': db_name,
             'col': col,
             'loc': 'australiaeast',
@@ -64,8 +64,8 @@ def test_cosmosdb_mongodb_collection_adaptiveru(self, resource_group):
         db_name = self.create_random_name(prefix='cli', length=15)
 
         self.kwargs.update({
-            'rg':'cosmosTest',
-            'acc': 'adrutest3',            
+            'rg':'canary-sdk-test',
+            'acc': 'sqladru3',            
             'db_name': db_name,
             'col': col,
             'loc': 'australiaeast',