1 files changed, 15 insertions, 0 deletions

diff --git a/src/test/burn/WixToolsetTest.BurnE2E/Utilities/TestBAController.cs b/src/test/burn/WixToolsetTest.BurnE2E/Utilities/TestBAController.cs
index 8e6611a2..3f9d76b8 100644
--- a/src/test/burn/WixToolsetTest.BurnE2E/Utilities/TestBAController.cs
+++ b/src/test/burn/WixToolsetTest.BurnE2E/Utilities/TestBAController.cs
@@ -42,6 +42,11 @@ namespace WixToolsetTest.BurnE2E
             }
         }
 
+        public void SetAllowAcquireAfterValidationFailure(string value = "true")
+        {
+            this.SetBurnTestValue("AllowAcquireAfterValidationFailure", value);
+        }
+
         public void SetExplicitlyElevateAndPlanFromOnElevateBegin(string value = "true")
         {
             this.SetBurnTestValue("ExplicitlyElevateAndPlanFromOnElevateBegin", value);
@@ -133,6 +138,16 @@ namespace WixToolsetTest.BurnE2E
         }
 
         /// <summary>
+        /// Sets the requested cache type for a package that the TestBA will return to the engine during plan.
+        /// </summary>
+        /// <param name="packageId">Package identity.</param>
+        /// <param name="type">Cache type to request.</param>
+        public void SetPackageRequestedCacheType(string packageId, BOOTSTRAPPER_CACHE_TYPE type)
+        {
+            this.SetPackageState(packageId, "CacheRequested", type.ToString());
+        }
+
+        /// <summary>
         /// Sets the requested state for a package that the TestBA will return to the engine during plan.
         /// </summary>
         /// <param name="packageId">Package identity.</param>