Add ability to skip a local path candidate if it failed verification.

Fixes 6818

1 files changed, 52 insertions, 1 deletions

diff --git a/src/test/burn/WixToolsetTest.BurnE2E/Utilities/TestBAController.cs b/src/test/burn/WixToolsetTest.BurnE2E/Utilities/TestBAController.cs
index 3f9d76b8..115f3b54 100644
--- a/src/test/burn/WixToolsetTest.BurnE2E/Utilities/TestBAController.cs
+++ b/src/test/burn/WixToolsetTest.BurnE2E/Utilities/TestBAController.cs
@@ -183,6 +183,26 @@ namespace WixToolsetTest.BurnE2E
         }
 
         /// <summary>
+        /// At startup, set the payload's source path to the given path.
+        /// </summary>
+        /// <param name="payloadId"></param>
+        /// <param name="sourcePath"></param>
+        public void SetPayloadInitialLocalSource(string payloadId, string sourcePath)
+        {
+            this.SetPayloadState(payloadId, "InitialLocalSource", sourcePath);
+        }
+
+        /// <summary>
+        /// At startup, set the container's source path to the given path.
+        /// </summary>
+        /// <param name="containerId"></param>
+        /// <param name="sourcePath"></param>
+        public void SetContainerInitialLocalSource(string containerId, string sourcePath)
+        {
+            this.SetContainerState(containerId, "InitialLocalSource", sourcePath);
+        }
+
+        /// <summary>
         /// Sets the number of times to re-run the Detect phase.
         /// </summary>
         /// <param name="state">Number of times to run Detect (after the first, normal, Detect).</param>
@@ -204,7 +224,6 @@ namespace WixToolsetTest.BurnE2E
         public void SetVerifyArguments(string verifyArguments)
         {
             this.SetBurnTestValue("VerifyArguments", verifyArguments);
-
         }
 
         private void SetPackageState(string packageId, string name, string value)
@@ -223,6 +242,38 @@ namespace WixToolsetTest.BurnE2E
             }
         }
 
+        private void SetContainerState(string containerId, string name, string value)
+        {
+            var key = String.Format(@"{0}\container\{1}", this.TestBaseRegKeyPath, containerId);
+            using (var containerKey = Registry.LocalMachine.CreateSubKey(key))
+            {
+                if (String.IsNullOrEmpty(value))
+                {
+                    containerKey.DeleteValue(name, false);
+                }
+                else
+                {
+                    containerKey.SetValue(name, value);
+                }
+            }
+        }
+
+        private void SetPayloadState(string payloadId, string name, string value)
+        {
+            var key = String.Format(@"{0}\payload\{1}", this.TestBaseRegKeyPath, payloadId ?? String.Empty);
+            using (var payloadKey = Registry.LocalMachine.CreateSubKey(key))
+            {
+                if (String.IsNullOrEmpty(value))
+                {
+                    payloadKey.DeleteValue(name, false);
+                }
+                else
+                {
+                    payloadKey.SetValue(name, value);
+                }
+            }
+        }
+
         public void Dispose()
         {
             Registry.LocalMachine.DeleteSubKeyTree($@"{this.BaseRegKeyPath}\{this.TestGroupName}", false);