From 36eb3fabd3082912d79dbeafc8c406ac6755e1b9 Mon Sep 17 00:00:00 2001
From: Bob Arnson <bob@firegiant.com>
Date: Tue, 1 Apr 2025 16:48:38 -0400
Subject: Sync IEngine.SetDownloadSource with native side.

Fixes https://github.com/wixtoolset/issues/issues/9018
---
 .../burn/WixToolsetTest.BurnE2E/Utilities/TestBAController.cs  | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'src/test/burn/WixToolsetTest.BurnE2E/Utilities/TestBAController.cs')

diff --git a/src/test/burn/WixToolsetTest.BurnE2E/Utilities/TestBAController.cs b/src/test/burn/WixToolsetTest.BurnE2E/Utilities/TestBAController.cs
index a378545e..ca395f56 100644
--- a/src/test/burn/WixToolsetTest.BurnE2E/Utilities/TestBAController.cs
+++ b/src/test/burn/WixToolsetTest.BurnE2E/Utilities/TestBAController.cs
@@ -107,6 +107,16 @@ namespace WixToolsetTest.BurnE2E
             this.SetPackageState(packageId, "CancelExecuteAtProgress", cancelPoint.HasValue ? cancelPoint.ToString() : null);
         }
 
+        /// <summary>
+        /// Forces a download action and sets the download source.
+        /// </summary>
+        /// <param name="packageId">Package identity.</param>
+        /// <param name="actionName">The URL format string.</param>
+        public void SetPackageForceDownloadSource(string packageId, string url)
+        {
+            this.SetPackageState(packageId, "ForceDownloadSource", url);
+        }
+
         /// <summary>
         /// Cancels the execute of a package at the next progess after the specified MSI action start.
         /// </summary>
-- 
cgit v1.2.3-55-g6feb