[CosmosDB] Refactoring Code (#11803)

* Refactoring code

* renamed PSDatabaseAccount to PSDatabaseAccountGetResults

* Updated session records

* Renaming PSDatabaseAccount

* Renaming references

* Updating helps

* Updating changelog

* nits

* relevant changes in DatabaseAccount models

Co-authored-by: Meha Kaushik <mekaushi@microsoft.com>

src/CosmosDB/CosmosDB/Cassandra/GetAzCosmosDBCassandraKeyspace.cs

@@ -41,7 +41,7 @@ public class GetAzCosmosDBCassandraKeyspace : AzureCosmosDBCmdletBase
 
         [Parameter(Mandatory = true, ValueFromPipeline = true, ParameterSetName = ParentObjectParameterSet, HelpMessage = Constants.AccountObjectHelpMessage)]
         [ValidateNotNull]
-        public PSDatabaseAccount ParentObject { get; set; }
+        public PSDatabaseAccountGetResults ParentObject { get; set; }
 
         public override void ExecuteCmdlet()
         {

src/CosmosDB/CosmosDB/Cassandra/GetAzCosmosDBCassandraKeyspaceThroughput.cs

@@ -40,7 +40,7 @@ public class GetAzCosmosDBCassandraKeyspaceThroughput : AzureCosmosDBCmdletBase
 
         [Parameter(Mandatory = true, ValueFromPipeline = true, ParameterSetName = ParentObjectParameterSet, HelpMessage = Constants.AccountObjectHelpMessage)]
         [ValidateNotNull]
-        public PSDatabaseAccount InputObject { get; set; }
+        public PSDatabaseAccountGetResults InputObject { get; set; }
 
         public override void ExecuteCmdlet()
         {

src/CosmosDB/CosmosDB/Cassandra/NewAzCosmosDBCassandraKeyspace.cs

@@ -48,7 +48,7 @@ public class NewAzCosmosDBCassandraKeyspace : AzureCosmosDBCmdletBase
 
         [Parameter(Mandatory = true, ValueFromPipeline = true, ParameterSetName = ParentObjectParameterSet, HelpMessage = Constants.AccountObjectHelpMessage)]
         [ValidateNotNull]
-        public PSDatabaseAccount ParentObject { get; set; }
+        public PSDatabaseAccountGetResults ParentObject { get; set; }
 
         public override void ExecuteCmdlet()
         {

src/CosmosDB/CosmosDB/Cassandra/NewAzCosmosDBCassandraTable.cs

@@ -100,7 +100,8 @@ public override void ExecuteCmdlet()
                 Id = Name,
                 DefaultTtl = TtlInSeconds
             };
-            cassandraTableResource.Schema = PSCassandraSchema.ConvertPSCassandraSchemaToCassandraSchema(Schema);
+
+            cassandraTableResource.Schema = PSCassandraSchema.ToSDKModel(Schema);
 
             CassandraTableCreateUpdateParameters cassandraTableCreateUpdateParameters = new CassandraTableCreateUpdateParameters
             {

src/CosmosDB/CosmosDB/Cassandra/UpdateAzCosmosDBCassandraKeyspace.cs

@@ -48,7 +48,7 @@ public class UpdateAzCosmosDBCassandraKeyspace : AzureCosmosDBCmdletBase
 
         [Parameter(Mandatory = true, ValueFromPipeline = true, ParameterSetName = ParentObjectParameterSet, HelpMessage = Constants.AccountObjectHelpMessage)]
         [ValidateNotNull]
-        public PSDatabaseAccount ParentObject { get; set; }
+        public PSDatabaseAccountGetResults ParentObject { get; set; }
 
         [Parameter(Mandatory = true, ValueFromPipeline = true, ParameterSetName = ObjectParameterSet, HelpMessage = Constants.CassandraKeyspaceObjectHelpMessage)]
         [ValidateNotNull]

src/CosmosDB/CosmosDB/Cassandra/UpdateAzCosmosDBCassandraKeyspaceThroughput.cs

@@ -45,7 +45,7 @@ public class UpdateAzCosmosDBCassandraKeyspaceThroughput : AzureCosmosDBCmdletBa
 
         [Parameter(Mandatory = true, ValueFromPipeline = true, ParameterSetName = ParentObjectParameterSet, HelpMessage = Constants.AccountObjectHelpMessage)]
         [ValidateNotNull]
-        public PSDatabaseAccount ParentObject { get; set; }
+        public PSDatabaseAccountGetResults ParentObject { get; set; }
 
         [Parameter(Mandatory = true, ValueFromPipeline = true, ParameterSetName = ObjectParameterSet, HelpMessage = Constants.CassandraKeyspaceObjectHelpMessage)]
         [ValidateNotNull]

src/CosmosDB/CosmosDB/Cassandra/UpdateAzCosmosDBCassandraTable.cs

@@ -106,7 +106,7 @@ public override void ExecuteCmdlet()
 
             if (Schema != null)
             {
-                cassandraTableResource.Schema = PSCassandraSchema.ConvertPSCassandraSchemaToCassandraSchema(Schema);
+                cassandraTableResource.Schema = PSCassandraSchema.ToSDKModel(Schema);
             }          
 
             CreateUpdateOptions options = new CreateUpdateOptions();

src/CosmosDB/CosmosDB/ChangeLog.md

@@ -19,14 +19,15 @@
 -->
 
 ## Upcoming Release
+* PSDatabaseAccount is renamed to PSDatabaseAccountGetResults
 * Detailed parameter is deprecated in the following cmdlets:
     Get-AzCosmosDBSqlContainer,
     Get-AzCosmosDBSqlDatabase, Get-AzCosmosDBGremlinDatabase,
     Get-AzCosmosDBGremlinGraph, Get-AzCosmosDBTable,
     Get-AzCosmosDBCassandraKeyspace, Get-AzCosmosDBCassandraTable,
     Get-AzCosmosDBMongoDBCollection, Get-AzCosmosDBMongoDBDatabase
 * Introduces KeyVaultKeyUri as a paramter in PSCosmosDBAccount, enabling BYOK feature
-* Updated the Azure.Management.CosmosDB Sdk Version to 1.1.1 
+* Updates the Azure.Management.CosmosDB Sdk Version to 1.1.1 
 * Replaces Set-AzCosmosDB* cmdlets with New-AzCosmosDB* and Update-AzComsosDB* cmdlets.
 The following cmdlets are added: 
     New-AzCosmosDBSqlStoredProcedure, New-AzCosmosDBSqlTrigger, 

src/CosmosDB/CosmosDB/CosmosDBAccount/GetAzCosmosDBAccount.cs

@@ -23,7 +23,7 @@
 
 namespace Microsoft.Azure.Commands.CosmosDB
 {
-    [Cmdlet(VerbsCommon.Get, ResourceManager.Common.AzureRMConstants.AzureRMPrefix + "CosmosDBAccount", DefaultParameterSetName = NameParameterSet), OutputType(typeof(PSDatabaseAccount))]
+    [Cmdlet(VerbsCommon.Get, ResourceManager.Common.AzureRMConstants.AzureRMPrefix + "CosmosDBAccount", DefaultParameterSetName = NameParameterSet), OutputType(typeof(PSDatabaseAccountGetResults))]
     public class GetAzCosmosDBAccount : AzureCosmosDBCmdletBase
     {
         [Parameter(Mandatory = true, ParameterSetName = NameParameterSet, HelpMessage = Constants.ResourceGroupNameHelpMessage)]
@@ -48,7 +48,7 @@ public override void ExecuteCmdlet()
                     IEnumerable<DatabaseAccountGetResults>  DatabaseAccounts = CosmosDBManagementClient.DatabaseAccounts.ListByResourceGroupWithHttpMessagesAsync(ResourceGroupName).GetAwaiter().GetResult().Body;
                     foreach (DatabaseAccountGetResults DatabaseAccount in DatabaseAccounts)
                     {
-                        WriteObject(new PSDatabaseAccount(DatabaseAccount));
+                        WriteObject(new PSDatabaseAccountGetResults(DatabaseAccount));
                     }
 
                     return;
@@ -62,7 +62,7 @@ public override void ExecuteCmdlet()
             }
 
             DatabaseAccountGetResults databaseAccount = CosmosDBManagementClient.DatabaseAccounts.GetWithHttpMessagesAsync(ResourceGroupName, Name).GetAwaiter().GetResult().Body;
-            WriteObject(new PSDatabaseAccount(databaseAccount));
+            WriteObject(new PSDatabaseAccountGetResults(databaseAccount));
 
             return;
         }

src/CosmosDB/CosmosDB/CosmosDBAccount/GetAzCosmosDBAccountKey.cs

@@ -44,7 +44,7 @@ public class GetAzCosmosDBAccountKey : AzureCosmosDBCmdletBase
 
         [Parameter(Mandatory = true, ValueFromPipeline = true, ParameterSetName = ObjectParameterSet, HelpMessage = Constants.AccountObjectHelpMessage)]
         [ValidateNotNull]
-        public PSDatabaseAccount InputObject { get; set; }
+        public PSDatabaseAccountGetResults InputObject { get; set; }
 
         public override void ExecuteCmdlet()
         {

src/CosmosDB/CosmosDB/CosmosDBAccount/NewAzCosmosDBAccount.cs

@@ -26,7 +26,7 @@
 
 namespace Microsoft.Azure.Commands.CosmosDB
 {
-    [Cmdlet(VerbsCommon.New, ResourceManager.Common.AzureRMConstants.AzureRMPrefix + "CosmosDBAccount", DefaultParameterSetName = NameParameterSet, SupportsShouldProcess = true), OutputType(typeof(PSDatabaseAccount))]
+    [Cmdlet(VerbsCommon.New, ResourceManager.Common.AzureRMConstants.AzureRMPrefix + "CosmosDBAccount", DefaultParameterSetName = NameParameterSet, SupportsShouldProcess = true), OutputType(typeof(PSDatabaseAccountGetResults))]
     public class NewAzCosmosDBAccount : AzureCosmosDBCmdletBase
     {
         [Parameter(Mandatory = true, ParameterSetName = NameParameterSet, HelpMessage = Constants.ResourceGroupNameHelpMessage)]
@@ -160,7 +160,7 @@ public override void ExecuteCmdlet()
 
                 foreach (PSLocation psLocation in LocationObject)
                 {
-                    LocationCollection.Add(PSLocation.ConvertPSLocationToLocation(psLocation));
+                    LocationCollection.Add(PSLocation.ToSDKModel(psLocation));
                     if (psLocation.FailoverPriority == 0)
                     {
                         writeLocation = psLocation.LocationName;
@@ -195,7 +195,7 @@ public override void ExecuteCmdlet()
             { 
                 foreach (PSVirtualNetworkRule psVirtualNetworkRule in VirtualNetworkRuleObject)
                 {
-                    virtualNetworkRule.Add(PSVirtualNetworkRule.ConvertPSVirtualNetworkRuleToVirtualNetworkRule(psVirtualNetworkRule));
+                    virtualNetworkRule.Add(PSVirtualNetworkRule.ToSDKModel(psVirtualNetworkRule));
                 }
             }
 
@@ -250,7 +250,7 @@ public override void ExecuteCmdlet()
             if (ShouldProcess(Name, "Creating Database Account"))
             {
                 DatabaseAccountGetResults cosmosDBAccount = CosmosDBManagementClient.DatabaseAccounts.CreateOrUpdateWithHttpMessagesAsync(ResourceGroupName, Name, databaseAccountCreateUpdateParameters).GetAwaiter().GetResult().Body;
-                WriteObject(new PSDatabaseAccount(cosmosDBAccount));
+                WriteObject(new PSDatabaseAccountGetResults(cosmosDBAccount));
             }
 
             return;

src/CosmosDB/CosmosDB/CosmosDBAccount/NewAzCosmosDBAccountKey.cs

@@ -45,7 +45,7 @@ public class NewAzCosmosDBAccountKey : AzureCosmosDBCmdletBase
 
         [Parameter(Mandatory = true, ValueFromPipeline = true, ParameterSetName = ObjectParameterSet, HelpMessage = Constants.AccountObjectHelpMessage)]
         [ValidateNotNullOrEmpty]
-        public PSDatabaseAccount InputObject { get; set; }
+        public PSDatabaseAccountGetResults InputObject { get; set; }
 
         [Parameter(Mandatory = false, HelpMessage = Constants.AsJobHelpMessage)]
         public SwitchParameter AsJob { get; set; }

src/CosmosDB/CosmosDB/CosmosDBAccount/RemoveAzCosmosDBAccount.cs

@@ -39,7 +39,7 @@ public class RemoveAzCosmosDBAccount : AzureCosmosDBCmdletBase
 
         [Parameter(Mandatory = true, ValueFromPipeline = true, ParameterSetName = ObjectParameterSet, HelpMessage = Constants.AccountObjectHelpMessage)]
         [ValidateNotNullOrEmpty]
-        public PSDatabaseAccount InputObject { get; set; }
+        public PSDatabaseAccountGetResults InputObject { get; set; }
 
         [Parameter(Mandatory = false, HelpMessage = Constants.AsJobHelpMessage)]
         public SwitchParameter AsJob { get; set; }

src/CosmosDB/CosmosDB/CosmosDBAccount/UpdateAzCosmosDBAccount.cs

@@ -26,7 +26,7 @@
 
 namespace Microsoft.Azure.Commands.CosmosDB
 {
-    [Cmdlet("Update", ResourceManager.Common.AzureRMConstants.AzureRMPrefix + "CosmosDBAccount", DefaultParameterSetName = NameParameterSet, SupportsShouldProcess = true), OutputType(typeof(PSDatabaseAccount))]
+    [Cmdlet("Update", ResourceManager.Common.AzureRMConstants.AzureRMPrefix + "CosmosDBAccount", DefaultParameterSetName = NameParameterSet, SupportsShouldProcess = true), OutputType(typeof(PSDatabaseAccountGetResults))]
     public class UpdateAzCosmosDBAccount : AzureCosmosDBCmdletBase
     {
         [Parameter(Mandatory = true, ParameterSetName = NameParameterSet, HelpMessage = Constants.ResourceGroupNameHelpMessage)]
@@ -44,7 +44,7 @@ public class UpdateAzCosmosDBAccount : AzureCosmosDBCmdletBase
 
         [Parameter(Mandatory = true, ValueFromPipeline = true, ParameterSetName = ObjectParameterSet, HelpMessage = Constants.AccountObjectHelpMessage)]
         [ValidateNotNull]
-        public PSDatabaseAccount InputObject { get; set; }
+        public PSDatabaseAccountGetResults InputObject { get; set; }
 
         [Parameter(Mandatory = false, HelpMessage = Constants.DefaultConsistencyLevelHelpMessage)]
         [PSArgumentCompleter("BoundedStaleness", "ConsistentPrefix", "Eventual", "Session", "Strong")]
@@ -197,7 +197,7 @@ public override void ExecuteCmdlet()
                 {
                     foreach (PSVirtualNetworkRule psVirtualNetworkRule in VirtualNetworkRuleObject)
                     {
-                        virtualNetworkRule.Add(PSVirtualNetworkRule.ConvertPSVirtualNetworkRuleToVirtualNetworkRule(psVirtualNetworkRule));
+                        virtualNetworkRule.Add(PSVirtualNetworkRule.ToSDKModel(psVirtualNetworkRule));
                     }
                 }
                 databaseAccountUpdateParameters.VirtualNetworkRules = virtualNetworkRule;
@@ -212,7 +212,7 @@ public override void ExecuteCmdlet()
             if (ShouldProcess(Name, "Updating Database Account"))
             {
                 DatabaseAccountGetResults cosmosDBAccount = CosmosDBManagementClient.DatabaseAccounts.UpdateWithHttpMessagesAsync(ResourceGroupName, Name, databaseAccountUpdateParameters).GetAwaiter().GetResult().Body;
-                WriteObject(new PSDatabaseAccount(cosmosDBAccount));
+                WriteObject(new PSDatabaseAccountGetResults(cosmosDBAccount));
             }
 
             return;

src/CosmosDB/CosmosDB/CosmosDBAccount/UpdateAzCosmosDBAccountFailoverPriority.cs

@@ -24,7 +24,7 @@
 
 namespace Microsoft.Azure.Commands.CosmosDB
 {
-    [Cmdlet("Update", ResourceManager.Common.AzureRMConstants.AzureRMPrefix + "CosmosDBAccountFailoverPriority", DefaultParameterSetName = NameParameterSet, SupportsShouldProcess = true), OutputType(typeof(PSDatabaseAccount))]
+    [Cmdlet("Update", ResourceManager.Common.AzureRMConstants.AzureRMPrefix + "CosmosDBAccountFailoverPriority", DefaultParameterSetName = NameParameterSet, SupportsShouldProcess = true), OutputType(typeof(PSDatabaseAccountGetResults))]
     public class UpdateAzCosmosDBAccountFailoverPriority : AzureCosmosDBCmdletBase
     {
         [Parameter(Mandatory = true, ParameterSetName = NameParameterSet, HelpMessage = Constants.ResourceGroupNameHelpMessage)]
@@ -46,7 +46,7 @@ public class UpdateAzCosmosDBAccountFailoverPriority : AzureCosmosDBCmdletBase
 
         [Parameter(Mandatory = true, ValueFromPipeline = true, ParameterSetName = ObjectParameterSet, HelpMessage = Constants.AccountObjectHelpMessage)]
         [ValidateNotNull] 
-        public PSDatabaseAccount InputObject { get; set; }
+        public PSDatabaseAccountGetResults InputObject { get; set; }
 
         [Parameter(Mandatory = false, HelpMessage = Constants.AsJobHelpMessage)]
         public SwitchParameter AsJob { get; set; }
@@ -81,7 +81,7 @@ public override void ExecuteCmdlet()
                 CosmosDBManagementClient.DatabaseAccounts.FailoverPriorityChangeAsync(ResourceGroupName, Name, new FailoverPolicies(failoverPolicies)).GetAwaiter().GetResult();
 
                 DatabaseAccountGetResults databaseAccount = CosmosDBManagementClient.DatabaseAccounts.GetWithHttpMessagesAsync(ResourceGroupName, Name).GetAwaiter().GetResult().Body;
-                WriteObject(new PSDatabaseAccount(databaseAccount));
+                WriteObject(new PSDatabaseAccountGetResults(databaseAccount));
             }
 
             return;

src/CosmosDB/CosmosDB/CosmosDBAccount/UpdateAzCosmosDBAccountRegion.cs

@@ -24,7 +24,7 @@
 
 namespace Microsoft.Azure.Commands.CosmosDB
 {
-    [Cmdlet("Update", ResourceManager.Common.AzureRMConstants.AzureRMPrefix + "CosmosDBAccountRegion", DefaultParameterSetName = NameParameterSet, SupportsShouldProcess = true), OutputType(typeof(PSDatabaseAccount))]
+    [Cmdlet("Update", ResourceManager.Common.AzureRMConstants.AzureRMPrefix + "CosmosDBAccountRegion", DefaultParameterSetName = NameParameterSet, SupportsShouldProcess = true), OutputType(typeof(PSDatabaseAccountGetResults))]
     public class UpdateAzCosmosDBAccountRegion : AzureCosmosDBCmdletBase
     { 
         [Parameter(Mandatory = true, ParameterSetName = NameParameterSet, HelpMessage = Constants.ResourceGroupNameHelpMessage)]
@@ -48,7 +48,7 @@ public class UpdateAzCosmosDBAccountRegion : AzureCosmosDBCmdletBase
 
         [Parameter(Mandatory = true, ValueFromPipeline = true, ParameterSetName = ObjectParameterSet, HelpMessage = Constants.AccountObjectHelpMessage)]
         [ValidateNotNull]
-        public PSDatabaseAccount InputObject { get; set; }
+        public PSDatabaseAccountGetResults InputObject { get; set; }
 
         [Parameter(Mandatory = false, HelpMessage = Constants.AsJobHelpMessage)]
         public SwitchParameter AsJob { get; set; }
@@ -87,7 +87,7 @@ public override void ExecuteCmdlet()
                 {
                     foreach (PSLocation psLocation in LocationObject)
                     {
-                        locations.Add(PSLocation.ConvertPSLocationToLocation(psLocation));
+                        locations.Add(PSLocation.ToSDKModel(psLocation));
                     }
                 }
             }
@@ -104,7 +104,7 @@ public override void ExecuteCmdlet()
                 CosmosDBManagementClient.DatabaseAccounts.UpdateWithHttpMessagesAsync(ResourceGroupName, Name, createUpdateParameters).GetAwaiter().GetResult();
 
                 DatabaseAccountGetResults databaseAccount = CosmosDBManagementClient.DatabaseAccounts.GetWithHttpMessagesAsync(ResourceGroupName, Name).GetAwaiter().GetResult().Body;
-                WriteObject(new PSDatabaseAccount(databaseAccount));
+                WriteObject(new PSDatabaseAccountGetResults(databaseAccount));
             }
 
             return;

src/CosmosDB/CosmosDB/Gremlin/GetAzCosmosDBGremlinDatabase.cs

@@ -41,7 +41,7 @@ public class GetAzCosmosDBGremlinDatabase : AzureCosmosDBCmdletBase
 
         [Parameter(Mandatory = true, ValueFromPipeline = true, ParameterSetName = ParentObjectParameterSet, HelpMessage = Constants.AccountObjectHelpMessage)]
         [ValidateNotNull]
-        public PSDatabaseAccount ParentObject { get; set; }
+        public PSDatabaseAccountGetResults ParentObject { get; set; }
 
         public override void ExecuteCmdlet()
         {

src/CosmosDB/CosmosDB/Gremlin/GetAzCosmosDBGremlinDatabaseThroughput.cs

@@ -40,7 +40,7 @@ public class GetAzCosmosDBGremlinDatabaseThroughput : AzureCosmosDBCmdletBase
 
         [Parameter(Mandatory = true, ParameterSetName = ParentObjectParameterSet, HelpMessage = Constants.AccountObjectHelpMessage)]
         [ValidateNotNull]
-        public PSDatabaseAccount InputObject { get; set; }
+        public PSDatabaseAccountGetResults InputObject { get; set; }
 
         public override void ExecuteCmdlet()
         {

src/CosmosDB/CosmosDB/Gremlin/NewAzCosmosDBGremlinDatabase.cs

@@ -48,7 +48,7 @@ public class NewAzCosmosDBGremlinDatabase : AzureCosmosDBCmdletBase
 
         [Parameter(Mandatory = true, ValueFromPipeline = true, ParameterSetName = ParentObjectParameterSet, HelpMessage = Constants.AccountObjectHelpMessage)]
         [ValidateNotNull]
-        public PSDatabaseAccount ParentObject { get; set; }
+        public PSDatabaseAccountGetResults ParentObject { get; set; }
 
         public override void ExecuteCmdlet()
         {

src/CosmosDB/CosmosDB/Gremlin/NewAzCosmosDBGremlinGraph.cs

@@ -135,7 +135,7 @@ public override void ExecuteCmdlet()
 
             if (UniqueKeyPolicy != null)
             {
-                gremlinGraphResource.UniqueKeyPolicy = PSUniqueKeyPolicy.ConvertPSUniqueKeyPolicyToUniqueKeyPolicy(UniqueKeyPolicy);
+                gremlinGraphResource.UniqueKeyPolicy = PSUniqueKeyPolicy.ToSDKModel(UniqueKeyPolicy);
             }
 
             if (TtlInSeconds != null)
@@ -145,7 +145,7 @@ public override void ExecuteCmdlet()
 
             if (ConflictResolutionPolicy != null)
             {
-                gremlinGraphResource.ConflictResolutionPolicy = PSConflictResolutionPolicy.ConvertPSConflictResolutionPolicyToConflictResolutionPolicy(ConflictResolutionPolicy);
+                gremlinGraphResource.ConflictResolutionPolicy = PSConflictResolutionPolicy.ToSDKModel(ConflictResolutionPolicy);
             }
             else if (ConflictResolutionPolicyMode != null)
             {
@@ -168,7 +168,7 @@ public override void ExecuteCmdlet()
 
             if (IndexingPolicy != null)
             {
-                gremlinGraphResource.IndexingPolicy = PSIndexingPolicy.ConvertPSIndexingToIndexingPolicy(IndexingPolicy);
+                gremlinGraphResource.IndexingPolicy = PSIndexingPolicy.ToSDKModel(IndexingPolicy);
             }
 
             CreateUpdateOptions options = new CreateUpdateOptions();

src/CosmosDB/CosmosDB/Gremlin/NewAzCosmosDBGremlinIndexingPolicy.cs

@@ -76,7 +76,9 @@ public override void ExecuteCmdlet()
             indexingPolicy.Automatic = Automatic;
 
             if (IndexingMode != null)
+            {
                 indexingPolicy.IndexingMode = IndexingMode;
+            }
 
             WriteObject(indexingPolicy);
             return;

src/CosmosDB/CosmosDB/Gremlin/NewAzCosmosDBGremlinUniqueKey.cs

@@ -19,20 +19,7 @@
 namespace Microsoft.Azure.Commands.CosmosDB
 {
     [Cmdlet(VerbsCommon.New, ResourceManager.Common.AzureRMConstants.AzureRMPrefix + "CosmosDBGremlinUniqueKey"), OutputType(typeof(PSUniqueKey))]
-    public class NewAzCosmosDBGremlinUniqueKey : AzureCosmosDBCmdletBase
-    {
-        [Parameter(Mandatory = true, HelpMessage = Constants.UniqueKeyPathHelpMessage)]
-        public string[] Path { get; set; }
-
-        public override void ExecuteCmdlet()
-        {
-            PSUniqueKey uniqueKey = new PSUniqueKey
-            {
-                Paths = Path
-            };
-
-            WriteObject(uniqueKey);
-            return;
-        }
+    public class NewAzCosmosDBGremlinUniqueKey : NewAzCosmosDBSqlUniqueKey 
+    { 
     }
 }