Sync IEngine.SetUpdateSource with native side.

Fixes wixtoolset/issues#9023

Author: barnson

src/api/burn/WixToolset.BootstrapperApplicationApi/Engine.cs

@@ -259,9 +259,9 @@ public void SetUpdate(string localSource, string downloadSource, long size, Upda
         }
 
         /// <inheritdoc/>
-        public void SetUpdateSource(string url)
+        public void SetUpdateSource(string url, string authorizationHeader)
         {
-            this.engine.SetUpdateSource(url);
+            this.engine.SetUpdateSource(url, authorizationHeader);
         }
 
         /// <inheritdoc/>

src/api/burn/WixToolset.BootstrapperApplicationApi/IBootstrapperEngine.cs

@@ -220,7 +220,8 @@ void LaunchApprovedExe(
         /// Sets the URL to the update feed.
         /// </summary>
         void SetUpdateSource(
-            [MarshalAs(UnmanagedType.LPWStr)] string url
+            [MarshalAs(UnmanagedType.LPWStr)] string url,
+            [MarshalAs(UnmanagedType.LPWStr)] string wzAuthorizationHeader
             );
 
         /// <summary>

src/api/burn/WixToolset.BootstrapperApplicationApi/IEngine.cs

@@ -156,7 +156,8 @@ public interface IEngine
         /// Sets the URL to the update feed.
         /// </summary>
         /// <param name="url">URL of the update feed.</param>
-        void SetUpdateSource(string url);
+        /// <param name="authorizationHeader">Additional proxy authentication header. Not currently used.</param>
+        void SetUpdateSource(string url, string authorizationHeader);
 
         /// <summary>
         /// Set the local source for a package or container.

src/test/burn/TestBA/TestBA.cs

@@ -42,6 +42,7 @@ public class TestBA : BootstrapperApplication
         private int retryExecuteFilesInUse;
         private bool rollingBack;
         private string forceDownloadSource;
+        private string forceUpdateSource;
 
         private IBootstrapperCommand Command { get; set; }
 
@@ -217,9 +218,28 @@ private void ShutdownUiThread(int? exitCode = null)
             }
         }
 
+        protected override void OnDetectBegin(DetectBeginEventArgs args)
+        {
+            this.Log("OnDetectBegin");
+
+            this.forceUpdateSource = this.ReadPackageAction(null, "ForceUpdateSource");
+            if (!String.IsNullOrEmpty(this.forceUpdateSource))
+            {
+                this.Log("    OnDetectBegin::ForceUpdateSource: {0}", this.forceUpdateSource);
+            }
+
+        }
+
         protected override void OnDetectUpdateBegin(DetectUpdateBeginEventArgs args)
         {
             this.Log("OnDetectUpdateBegin");
+
+            if (!String.IsNullOrEmpty(this.forceUpdateSource))
+            {
+                this.Log("    OnDetectUpdateBegin::ForceUpdateSource: {0}", this.forceUpdateSource);
+                this.Engine.SetUpdateSource(this.forceUpdateSource, String.Empty);
+            }
+
             if (LaunchAction.UpdateReplaceEmbedded == this.action || LaunchAction.UpdateReplace == this.action)
             {
                 args.Skip = false;

src/test/burn/TestData/FailureTests/BundleUpdate/Bundle.wxs

@@ -0,0 +1,31 @@
+<!-- Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information. -->
+
+<?ifndef TestVersion?>
+<?define TestVersion = 1.0.0.0?>
+<?endif?>
+
+<Wix xmlns="http://wixtoolset.org/schemas/v4/wxs" xmlns:bal="http://wixtoolset.org/schemas/v4/wxs/bal">
+  <Bundle 
+    Name="~$(var.TestGroupName) - $(var.BundleName)" 
+    Version="$(var.TestVersion)" 
+    Id="WixToolset.Burn.SetUpdateSourceTest" 
+    Compressed="yes">
+    
+    <Update Location="updateurl" />
+    
+    <Log Prefix="~$(var.TestGroupName)_$(var.BundleName)" />
+
+    <Variable Name="TestGroupName" Value="$(var.TestGroupName)" />
+
+    <Chain>
+      <PackageGroupRef Id="TestBA" />
+      <PackageGroupRef Id="BundlePackages" />
+    </Chain>
+  </Bundle>
+
+  <Fragment>
+    <PackageGroup Id="BundlePackages">
+      <MsiPackage Id="PackageA" SourceFile="$(var.PackageA.TargetPath)" />
+    </PackageGroup>
+  </Fragment>
+</Wix>

src/test/burn/TestData/FailureTests/BundleUpdate/BundleUpdate.wixproj

@@ -0,0 +1,16 @@
+<!-- Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information. -->
+<Project Sdk="WixToolset.Sdk">
+  <PropertyGroup>
+    <OutputType>Bundle</OutputType>
+    <UpgradeCode>{C60B9483-CE87-4FDA-AE5A-B39A52E956E8}</UpgradeCode>
+  </PropertyGroup>
+  <ItemGroup>
+    <ProjectReference Include="..\PackageA\PackageA.wixproj" />
+    <ProjectReference Include="..\PackageB\PackageB.wixproj" />
+    <ProjectReference Include="..\..\TestBA\TestBAWixlib\testbawixlib.wixproj" />
+  </ItemGroup>
+  <ItemGroup>
+    <PackageReference Include="WixToolset.BootstrapperApplications.wixext" />
+    <PackageReference Include="WixToolset.NetFx.wixext" />
+  </ItemGroup>
+</Project>

src/test/burn/WixToolsetTest.BurnE2E/FailureTests.cs

@@ -29,6 +29,24 @@ public void CanSetDownloadSourceAndForceDownload()
             packageA.VerifyInstalled(false);
         }
 
+        [RuntimeFact]
+        public void CanSetUpdateSource()
+        {
+            var packageA = this.CreatePackageInstaller("PackageA");
+            var bundleA = this.CreateBundleInstaller("BundleUpdate");
+            var testBAController = this.CreateTestBAController();
+
+            testBAController.SetForceUpdateSource("https://1e1bf2be1c384fd1a0c4c0500eef971b/update_feed.atom.xml");
+
+            packageA.VerifyInstalled(false);
+
+            bundleA.VerifyUnregisteredAndRemovedFromPackageCache();
+
+            bundleA.Install();
+
+            packageA.VerifyInstalled(true);
+        }
+
         [RuntimeFact]
         public void CanCancelExePackageAndAbandonIt()
         {

src/test/burn/WixToolsetTest.BurnE2E/Utilities/TestBAController.cs

@@ -67,6 +67,11 @@ public void SetQuitAfterDetect(string value = "true")
             this.SetBurnTestValue("QuitAfterDetect", value);
         }
 
+        public void SetForceUpdateSource(string url)
+        {
+            this.SetBurnTestValue("ForceUpdateSource", url);
+        }
+
         /// <summary>
         /// Slows the cache progress of a package.
         /// </summary>