[InAccountRestore] Introducing new PowerShell commands for InAccount Restore of Deleted Database/ Containers feature for Table/Gremlin (#21495)

* InAccount Restore changes for tables and gremlin

* generated files using autorest

* update proj files

* update proj file

* update test proj file

* add UXMetadataIssues exception

* update help files

* Revert "generated files using autorest"

This reverts commit 52fbdeb.

* update proj files

* update generated files

* update help file headers

* update help

* update session records except cassandra and account tests

* update help files

* update session records

* update help files

* update help files

* session records for failing tests with updated account name

* cross region restore change

* udpated recordings

* help file updates

* burst capacity changes

* update test sleep time

---------

Co-authored-by: dsapaliga <[email protected]>
Co-authored-by: Richa Sharma <[email protected]>

Author: 3 people

src/CosmosDB/CosmosDB.Sdk/CosmosDB.Sdk.csproj

@@ -8,7 +8,7 @@
 
   <PropertyGroup>
     <TargetFramework>netstandard2.0</TargetFramework>
-    <AssemblyName>Microsoft.Azure.PowerShell.CosmosDB.Sdk</AssemblyName>
+    <AssemblyName>Microsoft.Azure.PowerShell.CosmosDB.Management.Sdk</AssemblyName>
     <RootNamespace>Microsoft.Azure.Management.CosmosDB</RootNamespace>
     <NoWarn>$(NoWarn);CS0108;CS1573</NoWarn>
   </PropertyGroup>

src/CosmosDB/CosmosDB.Sdk/Generated/CassandraResourcesOperations.cs

@@ -180,9 +180,9 @@ public static CassandraKeyspaceGetResults CreateUpdateCassandraKeyspace(this ICa
             /// <param name='keyspaceName'>
             /// Cosmos DB keyspace name.
             /// </param>
-            public static void DeleteCassandraKeyspace(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName)
+            public static CassandraResourcesDeleteCassandraKeyspaceHeaders DeleteCassandraKeyspace(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName)
             {
-                operations.DeleteCassandraKeyspaceAsync(resourceGroupName, accountName, keyspaceName).GetAwaiter().GetResult();
+                return operations.DeleteCassandraKeyspaceAsync(resourceGroupName, accountName, keyspaceName).GetAwaiter().GetResult();
             }
 
             /// <summary>
@@ -203,9 +203,12 @@ public static void DeleteCassandraKeyspace(this ICassandraResourcesOperations op
             /// <param name='cancellationToken'>
             /// The cancellation token.
             /// </param>
-            public static async Task DeleteCassandraKeyspaceAsync(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName, CancellationToken cancellationToken = default(CancellationToken))
+            public static async Task<CassandraResourcesDeleteCassandraKeyspaceHeaders> DeleteCassandraKeyspaceAsync(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName, CancellationToken cancellationToken = default(CancellationToken))
             {
-                (await operations.DeleteCassandraKeyspaceWithHttpMessagesAsync(resourceGroupName, accountName, keyspaceName, null, cancellationToken).ConfigureAwait(false)).Dispose();
+                using (var _result = await operations.DeleteCassandraKeyspaceWithHttpMessagesAsync(resourceGroupName, accountName, keyspaceName, null, cancellationToken).ConfigureAwait(false))
+                {
+                    return _result.Headers;
+                }
             }
 
             /// <summary>
@@ -584,9 +587,9 @@ public static CassandraTableGetResults CreateUpdateCassandraTable(this ICassandr
             /// <param name='tableName'>
             /// Cosmos DB table name.
             /// </param>
-            public static void DeleteCassandraTable(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName, string tableName)
+            public static CassandraResourcesDeleteCassandraTableHeaders DeleteCassandraTable(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName, string tableName)
             {
-                operations.DeleteCassandraTableAsync(resourceGroupName, accountName, keyspaceName, tableName).GetAwaiter().GetResult();
+                return operations.DeleteCassandraTableAsync(resourceGroupName, accountName, keyspaceName, tableName).GetAwaiter().GetResult();
             }
 
             /// <summary>
@@ -610,9 +613,12 @@ public static void DeleteCassandraTable(this ICassandraResourcesOperations opera
             /// <param name='cancellationToken'>
             /// The cancellation token.
             /// </param>
-            public static async Task DeleteCassandraTableAsync(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName, string tableName, CancellationToken cancellationToken = default(CancellationToken))
+            public static async Task<CassandraResourcesDeleteCassandraTableHeaders> DeleteCassandraTableAsync(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName, string tableName, CancellationToken cancellationToken = default(CancellationToken))
             {
-                (await operations.DeleteCassandraTableWithHttpMessagesAsync(resourceGroupName, accountName, keyspaceName, tableName, null, cancellationToken).ConfigureAwait(false)).Dispose();
+                using (var _result = await operations.DeleteCassandraTableWithHttpMessagesAsync(resourceGroupName, accountName, keyspaceName, tableName, null, cancellationToken).ConfigureAwait(false))
+                {
+                    return _result.Headers;
+                }
             }
 
             /// <summary>
@@ -1013,9 +1019,9 @@ public static CassandraViewGetResults CreateUpdateCassandraView(this ICassandraR
             /// <param name='viewName'>
             /// Cosmos DB view name.
             /// </param>
-            public static void DeleteCassandraView(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName, string viewName)
+            public static CassandraResourcesDeleteCassandraViewHeaders DeleteCassandraView(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName, string viewName)
             {
-                operations.DeleteCassandraViewAsync(resourceGroupName, accountName, keyspaceName, viewName).GetAwaiter().GetResult();
+                return operations.DeleteCassandraViewAsync(resourceGroupName, accountName, keyspaceName, viewName).GetAwaiter().GetResult();
             }
 
             /// <summary>
@@ -1039,9 +1045,12 @@ public static void DeleteCassandraView(this ICassandraResourcesOperations operat
             /// <param name='cancellationToken'>
             /// The cancellation token.
             /// </param>
-            public static async Task DeleteCassandraViewAsync(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName, string viewName, CancellationToken cancellationToken = default(CancellationToken))
+            public static async Task<CassandraResourcesDeleteCassandraViewHeaders> DeleteCassandraViewAsync(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName, string viewName, CancellationToken cancellationToken = default(CancellationToken))
             {
-                (await operations.DeleteCassandraViewWithHttpMessagesAsync(resourceGroupName, accountName, keyspaceName, viewName, null, cancellationToken).ConfigureAwait(false)).Dispose();
+                using (var _result = await operations.DeleteCassandraViewWithHttpMessagesAsync(resourceGroupName, accountName, keyspaceName, viewName, null, cancellationToken).ConfigureAwait(false))
+                {
+                    return _result.Headers;
+                }
             }
 
             /// <summary>
@@ -1333,9 +1342,9 @@ public static CassandraKeyspaceGetResults BeginCreateUpdateCassandraKeyspace(thi
             /// <param name='keyspaceName'>
             /// Cosmos DB keyspace name.
             /// </param>
-            public static void BeginDeleteCassandraKeyspace(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName)
+            public static CassandraResourcesDeleteCassandraKeyspaceHeaders BeginDeleteCassandraKeyspace(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName)
             {
-                operations.BeginDeleteCassandraKeyspaceAsync(resourceGroupName, accountName, keyspaceName).GetAwaiter().GetResult();
+                return operations.BeginDeleteCassandraKeyspaceAsync(resourceGroupName, accountName, keyspaceName).GetAwaiter().GetResult();
             }
 
             /// <summary>
@@ -1356,9 +1365,12 @@ public static void BeginDeleteCassandraKeyspace(this ICassandraResourcesOperatio
             /// <param name='cancellationToken'>
             /// The cancellation token.
             /// </param>
-            public static async Task BeginDeleteCassandraKeyspaceAsync(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName, CancellationToken cancellationToken = default(CancellationToken))
+            public static async Task<CassandraResourcesDeleteCassandraKeyspaceHeaders> BeginDeleteCassandraKeyspaceAsync(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName, CancellationToken cancellationToken = default(CancellationToken))
             {
-                (await operations.BeginDeleteCassandraKeyspaceWithHttpMessagesAsync(resourceGroupName, accountName, keyspaceName, null, cancellationToken).ConfigureAwait(false)).Dispose();
+                using (var _result = await operations.BeginDeleteCassandraKeyspaceWithHttpMessagesAsync(resourceGroupName, accountName, keyspaceName, null, cancellationToken).ConfigureAwait(false))
+                {
+                    return _result.Headers;
+                }
             }
 
             /// <summary>
@@ -1587,9 +1599,9 @@ public static CassandraTableGetResults BeginCreateUpdateCassandraTable(this ICas
             /// <param name='tableName'>
             /// Cosmos DB table name.
             /// </param>
-            public static void BeginDeleteCassandraTable(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName, string tableName)
+            public static CassandraResourcesDeleteCassandraTableHeaders BeginDeleteCassandraTable(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName, string tableName)
             {
-                operations.BeginDeleteCassandraTableAsync(resourceGroupName, accountName, keyspaceName, tableName).GetAwaiter().GetResult();
+                return operations.BeginDeleteCassandraTableAsync(resourceGroupName, accountName, keyspaceName, tableName).GetAwaiter().GetResult();
             }
 
             /// <summary>
@@ -1613,9 +1625,12 @@ public static void BeginDeleteCassandraTable(this ICassandraResourcesOperations
             /// <param name='cancellationToken'>
             /// The cancellation token.
             /// </param>
-            public static async Task BeginDeleteCassandraTableAsync(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName, string tableName, CancellationToken cancellationToken = default(CancellationToken))
+            public static async Task<CassandraResourcesDeleteCassandraTableHeaders> BeginDeleteCassandraTableAsync(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName, string tableName, CancellationToken cancellationToken = default(CancellationToken))
             {
-                (await operations.BeginDeleteCassandraTableWithHttpMessagesAsync(resourceGroupName, accountName, keyspaceName, tableName, null, cancellationToken).ConfigureAwait(false)).Dispose();
+                using (var _result = await operations.BeginDeleteCassandraTableWithHttpMessagesAsync(resourceGroupName, accountName, keyspaceName, tableName, null, cancellationToken).ConfigureAwait(false))
+                {
+                    return _result.Headers;
+                }
             }
 
             /// <summary>
@@ -1862,9 +1877,9 @@ public static CassandraViewGetResults BeginCreateUpdateCassandraView(this ICassa
             /// <param name='viewName'>
             /// Cosmos DB view name.
             /// </param>
-            public static void BeginDeleteCassandraView(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName, string viewName)
+            public static CassandraResourcesDeleteCassandraViewHeaders BeginDeleteCassandraView(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName, string viewName)
             {
-                operations.BeginDeleteCassandraViewAsync(resourceGroupName, accountName, keyspaceName, viewName).GetAwaiter().GetResult();
+                return operations.BeginDeleteCassandraViewAsync(resourceGroupName, accountName, keyspaceName, viewName).GetAwaiter().GetResult();
             }
 
             /// <summary>
@@ -1888,9 +1903,12 @@ public static void BeginDeleteCassandraView(this ICassandraResourcesOperations o
             /// <param name='cancellationToken'>
             /// The cancellation token.
             /// </param>
-            public static async Task BeginDeleteCassandraViewAsync(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName, string viewName, CancellationToken cancellationToken = default(CancellationToken))
+            public static async Task<CassandraResourcesDeleteCassandraViewHeaders> BeginDeleteCassandraViewAsync(this ICassandraResourcesOperations operations, string resourceGroupName, string accountName, string keyspaceName, string viewName, CancellationToken cancellationToken = default(CancellationToken))
             {
-                (await operations.BeginDeleteCassandraViewWithHttpMessagesAsync(resourceGroupName, accountName, keyspaceName, viewName, null, cancellationToken).ConfigureAwait(false)).Dispose();
+                using (var _result = await operations.BeginDeleteCassandraViewWithHttpMessagesAsync(resourceGroupName, accountName, keyspaceName, viewName, null, cancellationToken).ConfigureAwait(false))
+                {
+                    return _result.Headers;
+                }
             }
 
             /// <summary>

src/CosmosDB/CosmosDB.Sdk/Generated/CassandraResourcesOperationsExtensions.cs

@@ -535,7 +535,7 @@ private void Initialize()
             RestorableTableResources = new RestorableTableResourcesOperations(this);
             Service = new ServiceOperations(this);
             BaseUri = new System.Uri("https://management.azure.com");
-            ApiVersion = "2022-08-15-preview";
+            ApiVersion = "2022-11-15-preview";
             AcceptLanguage = "en-US";
             LongRunningOperationRetryTimeout = 30;
             GenerateClientRequestId = true;