Moving to correct branch: Powershell cmdlet for SyncOwnerInformation

Author: NehaRaje

src/ServiceManagement/RecoveryServices/Commands.RecoveryServices.Test/ScenarioTests/RecoveryServicesTests.ps1

@@ -91,6 +91,7 @@ function Test-RecoveryServicesProtectionTests
 				if ($protectionEntity.Protected)
 				{
 					Set-AzureSiteRecoveryProtectionEntity -ProtectionEntity $protectionEntity -Protection "Enable" -Force
+					Update-AzureSiteRecoveryProtectionEntity -ProtectionEntity $protectionEntity
 				}
 				else
 				{

src/ServiceManagement/RecoveryServices/Commands.RecoveryServices/Commands.RecoveryServices.csproj

@@ -139,6 +139,7 @@
     <Compile Include="Service\GetAzureSiteRecoveryVaultSettings.cs" />
     <Compile Include="Service\GetAzureSiteRecoveryVM.cs" />
     <Compile Include="Service\ImportAzureSiteRecoveryVaultSettingsFile.cs" />
+    <Compile Include="Service\UpdateSyncOwnerInformation.cs" />
   </ItemGroup>
   <ItemGroup>
     <ProjectReference Include="..\..\..\Common\Commands.Common\Commands.Common.csproj">

src/ServiceManagement/RecoveryServices/Commands.RecoveryServices/PSRecoveryServicesClient/PSRecoveryServicesPEClient.cs

@@ -192,5 +192,21 @@ public JobResponse StartAzureSiteRecoveryReprotection(
                 request,
                 this.GetRequestHeaders());
         }
+
+        /// <summary>
+        /// Syncs owner role information on Protection entity.
+        /// </summary>
+        /// <param name="protectionContainerId">Protection Container ID</param>
+        /// <param name="virtualMachineId">Virtual Machine ID</param>
+        /// <returns>Job response</returns>
+        public JobResponse UpdateSyncOwnerInformationOnProtectionEntity(
+            string protectionContainerId,
+            string protectionEntityId)
+        {
+            return this.GetSiteRecoveryClient().ProtectionEntity.SyncOwnerInformation(
+                protectionContainerId,
+                protectionEntityId,
+                this.GetRequestHeaders());
+        }
     }
 }

src/ServiceManagement/RecoveryServices/Commands.RecoveryServices/Service/UpdateSyncOwnerInformation.cs

@@ -0,0 +1,137 @@
+// ----------------------------------------------------------------------------------
+//
+// Copyright Microsoft Corporation
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+// http://www.apache.org/licenses/LICENSE-2.0
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+// ----------------------------------------------------------------------------------
+
+namespace Microsoft.Azure.Commands.RecoveryServices
+{
+    #region Using directives
+    using System;
+    using System.Diagnostics;
+    using System.Management.Automation;
+    using System.Threading;
+    using Microsoft.Azure.Commands.RecoveryServices.SiteRecovery;
+    using Microsoft.WindowsAzure;
+    using Microsoft.WindowsAzure.Management.SiteRecovery.Models;
+    #endregion
+
+    /// <summary>
+    /// Used to initiate a commit operation.
+    /// </summary>
+    [Cmdlet(VerbsData.Update, "AzureSiteRecoveryProtectionEntity")]
+    [OutputType(typeof(ASRJob))]
+    public class UpdateSyncOwnerInformation : RecoveryServicesCmdletBase
+    {
+        #region Parameters
+
+        /// <summary>
+        /// ID of the PE object to Update user information on.
+        /// </summary>
+        private string protectionEntityId;
+
+        /// <summary>
+        /// Protection container ID of the object to Update user information on.
+        /// </summary>
+        private string protectionContainerId;
+
+        /// <summary>
+        /// Recovery Plan object.
+        /// </summary>
+        private ASRProtectionEntity protectionEntity;
+
+        /// <summary>
+        /// Wait / hold prompt till the Job completes.
+        /// </summary>
+        private bool waitForCompletion;
+
+        /// <summary>
+        /// Job response.
+        /// </summary>
+        private JobResponse jobResponse = null;
+
+        /// <summary>
+        /// Gets or sets Protection Entity object.
+        /// </summary>
+        [Parameter(Mandatory = true, ValueFromPipeline = true)]
+        [ValidateNotNullOrEmpty]
+        public ASRProtectionEntity ProtectionEntity
+        {
+            get { return this.protectionEntity; }
+            set { this.protectionEntity = value; }
+        }
+
+        /// <summary>
+        /// Gets or sets switch parameter. This is required to wait for job completion.
+        /// </summary>
+        [Parameter]
+        public SwitchParameter WaitForCompletion
+        {
+            get { return this.waitForCompletion; }
+            set { this.waitForCompletion = value; }
+        }
+
+        #endregion Parameters
+
+        /// <summary>
+        /// ProcessRecord of the command.
+        /// </summary>
+        public override void ExecuteCmdlet()
+        {
+            try
+            {
+                this.protectionContainerId = this.ProtectionEntity.ProtectionContainerId;
+                this.protectionEntityId = this.ProtectionEntity.ID;
+                this.SyncOwnerInformationOnPE();
+            }
+            catch (Exception exception)
+            {
+                this.HandleException(exception);
+            }
+        }
+
+        /// <summary>
+        /// Handles interrupts.
+        /// </summary>
+        protected override void StopProcessing()
+        {
+            // Ctrl + C and etc
+            base.StopProcessing();
+            this.StopProcessingFlag = true;
+        }
+
+        /// <summary>
+        /// Syncs the owner information.
+        /// </summary>
+        private void SyncOwnerInformationOnPE()
+        {
+            this.jobResponse = RecoveryServicesClient.UpdateSyncOwnerInformationOnProtectionEntity(
+                this.protectionContainerId,
+                this.protectionEntityId);
+
+            this.WriteJob(this.jobResponse.Job);
+
+            if (this.waitForCompletion)
+            {
+                this.WaitForJobCompletion(this.jobResponse.Job.ID);
+            }
+        }
+
+        /// <summary>
+        /// Writes Job
+        /// </summary>
+        /// <param name="job">Job object</param>
+        private void WriteJob(Microsoft.WindowsAzure.Management.SiteRecovery.Models.Job job)
+        {
+            this.WriteObject(new ASRJob(job));
+        }
+    }
+}