UFO/TFO

Author: sanjkuma

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

@@ -13,11 +13,9 @@
 // ----------------------------------------------------------------------------------
 
 using System;
-using System.Diagnostics;
 using System.Management.Automation;
-using System.Threading;
 using Microsoft.Azure.Commands.RecoveryServices.SiteRecovery;
-using Microsoft.WindowsAzure;
+using Microsoft.Azure.Portal.RecoveryServices.Models.Common;
 using Microsoft.WindowsAzure.Management.SiteRecovery.Models;
 
 namespace Microsoft.Azure.Commands.RecoveryServices
@@ -52,6 +50,11 @@ public class StartAzureSiteRecoveryTestFailoverJob : RecoveryServicesCmdletBase
         [ValidateNotNullOrEmpty]
         public string RpId { get; set; }
 
+        /// <summary>
+        /// Gets or sets ID of the Recovery Plan.
+        /// </summary>
+        public ASRNetwork Network { get; set; }
+
         /// <summary>
         /// Gets or sets Recovery Plan object.
         /// </summary>
@@ -175,11 +178,55 @@ public override void ExecuteCmdlet()
         /// </summary>
         private void StartRpTestFailover()
         {
-            RpTestFailoverRequest recoveryPlanTestFailoverRequest = new RpTestFailoverRequest();
-            recoveryPlanTestFailoverRequest.FailoverDirection = this.Direction;
+            RpTestFailoverRequest request = new RpTestFailoverRequest();
+
+            if (this.Network != null)
+            {
+                request.NetworkID = this.Network.ID;
+                request.NetworkType = "UseVMNetworkTypeForTestFailover";
+            }
+            else
+            {
+                request.NetworkID = this.networkId;
+            }
+
+            if (this.RecoveryPlan == null)
+            {
+                var rp = RecoveryServicesClient.GetAzureSiteRecoveryRecoveryPlan(
+                    this.RpId);
+                this.RecoveryPlan = new ASRRecoveryPlan(rp.RecoveryPlan);
+
+                this.ValidateUsageById(this.RecoveryPlan.ReplicationProvider);
+            }
+
+            request.ReplicationProviderSettings = string.Empty;
+
+            if (this.RecoveryPlan.ReplicationProvider == Constants.HyperVReplicaAzure)
+            {
+                request.ReplicationProvider = this.RecoveryPlan.ReplicationProvider;
+                if (this.Direction == Constants.PrimaryToRecovery)
+                {
+                    var blob = new AzureFailoverInput();
+                    blob.VaultLocation = this.GetCurrentValutLocation();
+                    request.ReplicationProviderSettings = DataContractUtils.Serialize<AzureFailoverInput>(blob);
+                }
+            }
+
+            request.FailoverDirection = this.Direction;
+
+            if (this.Network != null)
+            {
+                request.NetworkID = this.Network.ID;
+                this.networkType = "UseVMNetworkTypeForTestFailover";
+            }
+            else
+            {
+                request.NetworkID = this.networkId;
+            }
+
             this.jobResponse = RecoveryServicesClient.StartAzureSiteRecoveryTestFailover(
                 this.RpId, 
-                recoveryPlanTestFailoverRequest);
+                request);
 
             this.WriteJob(this.jobResponse.Job);
 
@@ -194,18 +241,51 @@ private void StartRpTestFailover()
         /// </summary>
         private void StartPETestFailover()
         {
-            var tfoReqeust = new TestFailoverRequest();
-            tfoReqeust.NetworkID = this.networkId;
-            tfoReqeust.FailoverDirection = this.Direction;
-            tfoReqeust.NetworkType = this.networkType;
-            tfoReqeust.ReplicationProvider = string.Empty;
-            tfoReqeust.ReplicationProviderSettings = string.Empty;
+            var request = new TestFailoverRequest();
+
+            if (this.ProtectionEntity == null)
+            {
+                var pe = RecoveryServicesClient.GetAzureSiteRecoveryProtectionEntity(
+                    this.ProtectionContainerId,
+                    this.ProtectionEntityId);
+                this.ProtectionEntity = new ASRProtectionEntity(pe.ProtectionEntity);
+
+                this.ValidateUsageById(this.ProtectionEntity.ReplicationProvider);
+            }
+
+            request.ReplicationProviderSettings = string.Empty;
+
+            if (this.ProtectionEntity.ReplicationProvider == Constants.HyperVReplicaAzure)
+            {
+                request.ReplicationProvider = this.ProtectionEntity.ReplicationProvider;
+                if (this.Direction == Constants.PrimaryToRecovery)
+                {
+                    var blob = new AzureFailoverInput();
+                    blob.VaultLocation = this.GetCurrentValutLocation();
+                    request.ReplicationProviderSettings = DataContractUtils.Serialize<AzureFailoverInput>(blob);
+                }
+            }
+
+            request.FailoverDirection = this.Direction;
+
+            if (this.Network != null)
+            {
+                request.NetworkID = this.Network.ID;
+                this.networkType = "UseVMNetworkTypeForTestFailover";
+            }
+            else
+            {
+                request.NetworkID = this.networkId;
+            }
+
+            request.FailoverDirection = this.Direction;
+            request.NetworkType = this.networkType;
 
             this.jobResponse =
                 RecoveryServicesClient.StartAzureSiteRecoveryTestFailover(
                 this.ProtectionContainerId,
                 this.ProtectionEntityId,
-                tfoReqeust);
+                request);
             this.WriteJob(this.jobResponse.Job);
 
             if (this.WaitForCompletion.IsPresent)

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

@@ -17,6 +17,7 @@
 using System.Management.Automation;
 using System.Threading;
 using Microsoft.Azure.Commands.RecoveryServices.SiteRecovery;
+using Microsoft.Azure.Portal.RecoveryServices.Models.Common;
 using Microsoft.WindowsAzure;
 using Microsoft.WindowsAzure.Management.SiteRecovery.Models;
 
@@ -140,14 +141,35 @@ public override void ExecuteCmdlet()
         /// </summary>
         private void StartPEUnplannedFailover()
         {
-            var ufoReqeust = new UnplannedFailoverRequest();
-            ufoReqeust.FailoverDirection = this.Direction;
-            ufoReqeust.SourceSiteOperations = this.PerformSourceSiteOperations;
+            var request = new UnplannedFailoverRequest();
+            if (this.ProtectionEntity == null)
+            {
+                var pe = RecoveryServicesClient.GetAzureSiteRecoveryProtectionEntity(
+                    this.ProtectionContainerId,
+                    this.ProtectionEntityId);
+                this.ProtectionEntity = new ASRProtectionEntity(pe.ProtectionEntity);
+
+                this.ValidateUsageById(this.ProtectionEntity.ReplicationProvider);
+            }
+
+            if (this.ProtectionEntity.ReplicationProvider == Constants.HyperVReplicaAzure)
+            {
+                request.ReplicationProvider = this.ProtectionEntity.ReplicationProvider;
+                if (this.Direction == Constants.PrimaryToRecovery)
+                {
+                    var blob = new AzureFailoverInput();
+                    blob.VaultLocation = this.GetCurrentValutLocation();
+                    request.ReplicationProviderSettings = DataContractUtils.Serialize<AzureFailoverInput>(blob);
+                }
+            }
+
+            request.FailoverDirection = this.Direction;
+            request.SourceSiteOperations = this.PerformSourceSiteOperations;
             this.jobResponse =
                 RecoveryServicesClient.StartAzureSiteRecoveryUnplannedFailover(
                 this.ProtectionContainerId,
                 this.ProtectionEntityId,
-                ufoReqeust);
+                request);
             this.WriteJob(this.jobResponse.Job);
 
             if (this.WaitForCompletion.IsPresent)
@@ -161,12 +183,34 @@ private void StartPEUnplannedFailover()
         /// </summary>
         private void StartRpUnPlannedFailover()
         {
-            RpUnplannedFailoverRequest recoveryPlanUnPlannedFailoverRequest = new RpUnplannedFailoverRequest();
-            recoveryPlanUnPlannedFailoverRequest.FailoverDirection = this.Direction;
-            recoveryPlanUnPlannedFailoverRequest.PrimaryAction = this.PrimaryAction;
+            RpUnplannedFailoverRequest request = new RpUnplannedFailoverRequest();
+
+            if (this.RecoveryPlan == null)
+            {
+                var rp = RecoveryServicesClient.GetAzureSiteRecoveryRecoveryPlan(
+                    this.RPId);
+                this.RecoveryPlan = new ASRRecoveryPlan(rp.RecoveryPlan);
+
+                this.ValidateUsageById(this.RecoveryPlan.ReplicationProvider);
+            }
+
+            if (this.RecoveryPlan.ReplicationProvider == Constants.HyperVReplicaAzure)
+            {
+                request.ReplicationProvider = this.RecoveryPlan.ReplicationProvider;
+                if (this.Direction == Constants.PrimaryToRecovery)
+                {
+                    var blob = new AzureFailoverInput();
+                    blob.VaultLocation = this.GetCurrentValutLocation();
+                    request.ReplicationProviderSettings = DataContractUtils.Serialize<AzureFailoverInput>(blob);
+                }
+            }
+
+            request.FailoverDirection = this.Direction;
+            request.PrimaryAction = this.PrimaryAction;
+
             this.jobResponse = RecoveryServicesClient.StartAzureSiteRecoveryUnplannedFailover(
                 this.RPId,
-                recoveryPlanUnPlannedFailoverRequest);
+                request);
 
             this.WriteJob(this.jobResponse.Job);