Port the patch tests from the old repo.

22 files changed, 387 insertions, 12 deletions

diff --git a/src/TestData/PatchTests/BundleA/BundleA.wixproj b/src/TestData/PatchTests/BundleA/BundleA.wixproj
new file mode 100644
index 00000000..bbdace55
--- /dev/null
+++ b/src/TestData/PatchTests/BundleA/BundleA.wixproj
@@ -0,0 +1,18 @@
+<!-- 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>{486FC795-69A5-4130-8727-4068F645A0A1}</UpgradeCode>
+  </PropertyGroup>
+  <ItemGroup>
+    <Compile Include="..\..\Templates\Bundle.wxs" Link="Bundle.wxs" />
+  </ItemGroup>
+  <ItemGroup>
+    <ProjectReference Include="..\PackageAv1\PackageAv1.wixproj" />
+    <ProjectReference Include="..\..\TestBA\TestBAWixlib\testbawixlib.wixproj" />
+  </ItemGroup>
+  <ItemGroup>
+    <PackageReference Include="WixToolset.Bal.wixext" Version="4.0.81" />
+    <PackageReference Include="WixToolset.NetFx.wixext" Version="4.0.57" />
+  </ItemGroup>
+</Project>
\ No newline at end of file
diff --git a/src/TestData/PatchTests/BundleA/BundleA.wxs b/src/TestData/PatchTests/BundleA/BundleA.wxs
new file mode 100644
index 00000000..d40351a1
--- /dev/null
+++ b/src/TestData/PatchTests/BundleA/BundleA.wxs
@@ -0,0 +1,11 @@
+<!-- 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. -->
+
+<Wix xmlns="http://wixtoolset.org/schemas/v4/wxs" xmlns:swid="http://wixtoolset.org/schemas/v4/wxs/tag">
+  <Fragment>    
+    <PackageGroup Id="BundlePackages">
+      <MsiPackage Id="PackageA" SourceFile="$(var.PackageAv1.TargetPath)" />
+    </PackageGroup>
+
+    <!--<swid:Tag Regid="regid.1995-08.com.example" />-->
+  </Fragment>
+</Wix>
diff --git a/src/TestData/PatchTests/BundlePatchA/BundlePatchA.wixproj b/src/TestData/PatchTests/BundlePatchA/BundlePatchA.wixproj
new file mode 100644
index 00000000..5846fce5
--- /dev/null
+++ b/src/TestData/PatchTests/BundlePatchA/BundlePatchA.wixproj
@@ -0,0 +1,18 @@
+<!-- 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>{AA083618-6280-44B8-9899-57BCC57906A5}</UpgradeCode>
+  </PropertyGroup>
+  <ItemGroup>
+    <Compile Include="..\..\Templates\Bundle.wxs" Link="Bundle.wxs" />
+  </ItemGroup>
+  <ItemGroup>
+    <ProjectReference Include="..\PatchA\PatchA.wixproj" />
+    <ProjectReference Include="..\..\TestBA\TestBAWixlib\testbawixlib.wixproj" />
+  </ItemGroup>
+  <ItemGroup>
+    <PackageReference Include="WixToolset.Bal.wixext" Version="4.0.81" />
+    <PackageReference Include="WixToolset.NetFx.wixext" Version="4.0.57" />
+  </ItemGroup>
+</Project>
\ No newline at end of file
diff --git a/src/TestData/PatchTests/BundlePatchA/BundlePatchA.wxs b/src/TestData/PatchTests/BundlePatchA/BundlePatchA.wxs
new file mode 100644
index 00000000..89f24936
--- /dev/null
+++ b/src/TestData/PatchTests/BundlePatchA/BundlePatchA.wxs
@@ -0,0 +1,11 @@
+<!-- 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. -->
+
+<Wix xmlns="http://wixtoolset.org/schemas/v4/wxs" xmlns:swid="http://wixtoolset.org/schemas/v4/wxs/tag">
+  <Fragment>    
+    <PackageGroup Id="BundlePackages">
+      <MspPackage Id="PatchA" SourceFile="$(var.PatchA.TargetPath)" PerMachine="yes" />
+    </PackageGroup>
+
+    <!--<swid:Tag Regid="regid.1995-08.com.example" Type="patch" />-->
+  </Fragment>
+</Wix>
diff --git a/src/TestData/PatchTests/BundlePatchA2/BundlePatchA2.wixproj b/src/TestData/PatchTests/BundlePatchA2/BundlePatchA2.wixproj
new file mode 100644
index 00000000..dee4a0f3
--- /dev/null
+++ b/src/TestData/PatchTests/BundlePatchA2/BundlePatchA2.wixproj
@@ -0,0 +1,19 @@
+<!-- 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>{1BE09331-2327-4534-9223-59B54EFAE7A5}</UpgradeCode>
+  </PropertyGroup>
+  <ItemGroup>
+    <Compile Include="..\..\Templates\Bundle.wxs" Link="Bundle.wxs" />
+  </ItemGroup>
+  <ItemGroup>
+    <ProjectReference Include="..\PatchA\PatchA.wixproj" />
+    <ProjectReference Include="..\PatchA2\PatchA2.wixproj" />
+    <ProjectReference Include="..\..\TestBA\TestBAWixlib\testbawixlib.wixproj" />
+  </ItemGroup>
+  <ItemGroup>
+    <PackageReference Include="WixToolset.Bal.wixext" Version="4.0.81" />
+    <PackageReference Include="WixToolset.NetFx.wixext" Version="4.0.57" />
+  </ItemGroup>
+</Project>
\ No newline at end of file
diff --git a/src/TestData/PatchTests/BundlePatchA2/BundlePatchA2.wxs b/src/TestData/PatchTests/BundlePatchA2/BundlePatchA2.wxs
new file mode 100644
index 00000000..f2c44ad4
--- /dev/null
+++ b/src/TestData/PatchTests/BundlePatchA2/BundlePatchA2.wxs
@@ -0,0 +1,12 @@
+<!-- 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. -->
+
+<Wix xmlns="http://wixtoolset.org/schemas/v4/wxs" xmlns:swid="http://wixtoolset.org/schemas/v4/wxs/tag">
+  <Fragment>    
+    <PackageGroup Id="BundlePackages">
+      <MspPackage Id="PatchA" SourceFile="$(var.PatchA.TargetPath)" PerMachine="yes" />
+      <MspPackage Id="PatchA2" SourceFile="$(var.PatchA2.TargetPath)" PerMachine="yes" />
+    </PackageGroup>
+
+    <!--<swid:Tag Regid="regid.1995-08.com.example" Type="patch" />-->
+  </Fragment>
+</Wix>
diff --git a/src/TestData/PatchTests/PackageAv1/PackageA.props b/src/TestData/PatchTests/PackageAv1/PackageA.props
new file mode 100644
index 00000000..1209d09e
--- /dev/null
+++ b/src/TestData/PatchTests/PackageAv1/PackageA.props
@@ -0,0 +1,12 @@
+<!-- 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>
+  <PropertyGroup>
+    <PackageName>PackageA</PackageName>
+    <ProductComponentsRef>true</ProductComponentsRef>
+    <ProductCode>{724F9BA5-DD9D-4851-855E-ECC35B27BF11}</ProductCode>
+    <UpgradeCode>{C56DA396-7A9A-4177-8264-638161CE9EB8}</UpgradeCode>
+  </PropertyGroup>
+  <ItemGroup>
+    <Compile Include="..\..\Templates\Package.wxs" Link="Package.wxs" />
+  </ItemGroup>
+</Project>
\ No newline at end of file
diff --git a/src/TestData/PatchTests/PackageAv1/PackageAv1.wixproj b/src/TestData/PatchTests/PackageAv1/PackageAv1.wixproj
new file mode 100644
index 00000000..45d3b2c8
--- /dev/null
+++ b/src/TestData/PatchTests/PackageAv1/PackageAv1.wixproj
@@ -0,0 +1,4 @@
+<!-- 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">
+  <Import Project="PackageA.props" />
+</Project>
\ No newline at end of file
diff --git a/src/TestData/PatchTests/PackageAv1/ProductComponents.wxs b/src/TestData/PatchTests/PackageAv1/ProductComponents.wxs
new file mode 100644
index 00000000..81c6da0c
--- /dev/null
+++ b/src/TestData/PatchTests/PackageAv1/ProductComponents.wxs
@@ -0,0 +1,17 @@
+<!-- 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. -->
+
+<Wix xmlns="http://wixtoolset.org/schemas/v4/wxs" xmlns:swid="http://wixtoolset.org/schemas/v4/wxs/tag">
+  <Fragment>
+    <ComponentGroup Id="ProductComponents">
+      <ComponentRef Id="RegistryComponent2" />
+    </ComponentGroup>
+
+    <!-- <swid:Tag Regid="regid.1995-08.com.example" /> -->
+  </Fragment>
+
+  <Fragment>
+    <Component Id="RegistryComponent2" Directory="INSTALLFOLDER">
+      <RegistryValue Root="HKLM" Key="Software\WiX\Tests\$(var.TestGroupName)" Name="$(var.PackageName)2" Value="!(bind.Property.TestVersion)" Type="string" />
+    </Component>
+  </Fragment>
+</Wix>
diff --git a/src/TestData/PatchTests/PackageAv1_0_1/PackageAv1_0_1.wixproj b/src/TestData/PatchTests/PackageAv1_0_1/PackageAv1_0_1.wixproj
new file mode 100644
index 00000000..9ceda117
--- /dev/null
+++ b/src/TestData/PatchTests/PackageAv1_0_1/PackageAv1_0_1.wixproj
@@ -0,0 +1,10 @@
+<!-- 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">
+  <Import Project="..\PackageAv1\PackageA.props" />
+  <PropertyGroup>
+    <Version>1.0.1.0</Version>
+  </PropertyGroup>
+  <ItemGroup>
+    <Compile Include="..\PackageAv1\ProductComponents.wxs" Link="ProductComponents.wxs" />
+  </ItemGroup>
+</Project>
\ No newline at end of file
diff --git a/src/TestData/PatchTests/PatchA/PatchA.wixproj b/src/TestData/PatchTests/PatchA/PatchA.wixproj
new file mode 100644
index 00000000..da9acb5e
--- /dev/null
+++ b/src/TestData/PatchTests/PatchA/PatchA.wixproj
@@ -0,0 +1,12 @@
+<!-- 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>PatchCreation</OutputType>
+    <TargetExt>.msp</TargetExt>
+    <SuppressSpecificWarnings>1079</SuppressSpecificWarnings>
+  </PropertyGroup>
+  <ItemGroup>
+    <ProjectReference Include="..\PackageAv1\PackageAv1.wixproj" />
+    <ProjectReference Include="..\PackageAv1_0_1\PackageAv1_0_1.wixproj" />
+  </ItemGroup>
+</Project>
\ No newline at end of file
diff --git a/src/TestData/PatchTests/PatchA/PatchA.wxs b/src/TestData/PatchTests/PatchA/PatchA.wxs
new file mode 100644
index 00000000..067f25ca
--- /dev/null
+++ b/src/TestData/PatchTests/PatchA/PatchA.wxs
@@ -0,0 +1,24 @@
+<!-- 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 Version?>
+<?define Version = 1.0.0.0?>
+<?endif?>
+
+<Wix xmlns="http://wixtoolset.org/schemas/v4/wxs" xmlns:swid="http://wixtoolset.org/schemas/v4/wxs/tag">
+    <Patch AllowRemoval="yes" Classification="Update" Description="Patch A in test $(var.TestGroupName)" DisplayName="$(var.TestGroupName) - Patch A" Manufacturer="Example Corporation" MinorUpdateTargetRTM="yes">
+        <Media Id="100" Cabinet="PatchA" EmbedCab="yes">
+            <PatchBaseline
+              Id="PatchA"
+              BaselineFile="$(var.PackageAv1.TargetDir)$(var.PackageAv1.TargetName).wixpdb"
+              UpdateFile="$(var.PackageAv1_0_1.TargetDir)$(var.PackageAv1_0_1.TargetName).wixpdb"
+            />
+        </Media>
+
+        <PatchFamily Id="A" Version="$(var.Version)" Supersede="yes">
+          <!-- <swid:TagRef Regid="regid.1995-08.com.example" /> -->
+
+          <ComponentRef Id="RegistryComponent"/>
+          <PropertyRef Id="TestVersion"/>
+        </PatchFamily>
+    </Patch>
+</Wix>
diff --git a/src/TestData/PatchTests/PatchA2/PatchA2.wixproj b/src/TestData/PatchTests/PatchA2/PatchA2.wixproj
new file mode 100644
index 00000000..da9acb5e
--- /dev/null
+++ b/src/TestData/PatchTests/PatchA2/PatchA2.wixproj
@@ -0,0 +1,12 @@
+<!-- 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>PatchCreation</OutputType>
+    <TargetExt>.msp</TargetExt>
+    <SuppressSpecificWarnings>1079</SuppressSpecificWarnings>
+  </PropertyGroup>
+  <ItemGroup>
+    <ProjectReference Include="..\PackageAv1\PackageAv1.wixproj" />
+    <ProjectReference Include="..\PackageAv1_0_1\PackageAv1_0_1.wixproj" />
+  </ItemGroup>
+</Project>
\ No newline at end of file
diff --git a/src/TestData/PatchTests/PatchA2/PatchA2.wxs b/src/TestData/PatchTests/PatchA2/PatchA2.wxs
new file mode 100644
index 00000000..a06401cc
--- /dev/null
+++ b/src/TestData/PatchTests/PatchA2/PatchA2.wxs
@@ -0,0 +1,23 @@
+<!-- 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 Version?>
+<?define Version = 1.0.0.0?>
+<?endif?>
+
+<Wix xmlns="http://wixtoolset.org/schemas/v4/wxs" xmlns:swid="http://wixtoolset.org/schemas/v4/wxs/tag">
+    <Patch AllowRemoval="yes" Classification="Update" Description="Patch A2 in test $(var.TestGroupName)" DisplayName="$(var.TestGroupName) - Patch A2" Manufacturer="Example Corporation" MinorUpdateTargetRTM="yes">
+        <Media Id="100" Cabinet="PatchA2" EmbedCab="yes">
+            <PatchBaseline
+              Id="PatchA2"
+              BaselineFile="$(var.PackageAv1.TargetDir)$(var.PackageAv1.TargetName).wixpdb"
+              UpdateFile="$(var.PackageAv1_0_1.TargetDir)$(var.PackageAv1_0_1.TargetName).wixpdb"
+            />
+        </Media>
+
+        <PatchFamily Id="A2" Version="$(var.Version)" Supersede="yes">
+          <!-- <swid:TagRef Regid="regid.1995-08.com.example" /> -->
+
+          <ComponentRef Id="RegistryComponent2"/>
+        </PatchFamily>
+    </Patch>
+</Wix>
diff --git a/src/TestData/Templates/Package.wxs b/src/TestData/Templates/Package.wxs
index 5522b9a3..4923e1aa 100644
--- a/src/TestData/Templates/Package.wxs
+++ b/src/TestData/Templates/Package.wxs
@@ -6,8 +6,12 @@
 <?define Version = 1.0.0.0?>
 <?endif?>
 
+<?ifndef ProductCode?>
+<?define ProductCode = *?>
+<?endif?>
+
 <Wix xmlns="http://wixtoolset.org/schemas/v4/wxs">
-  <Package Name="~$(var.TestGroupName) - $(var.PackageName)" Language="1033" Version="$(var.Version)" Manufacturer="Microsoft Corporation" UpgradeCode="$(var.UpgradeCode)" Compressed="yes">
+  <Package Name="~$(var.TestGroupName) - $(var.PackageName)" Language="1033" Version="$(var.Version)" Manufacturer="Microsoft Corporation" UpgradeCode="$(var.UpgradeCode)" Compressed="yes" ProductCode="$(var.ProductCode)">
     <MajorUpgrade DowngradeErrorMessage="A newer version of [ProductName] is already installed." />
     <?ifndef CabPrefix?>
     <MediaTemplate EmbedCab="yes" />
@@ -20,6 +24,9 @@
     <Feature Id="Complete" Level="1">
       <ComponentRef Id="FileComponent" />
       <ComponentRef Id="RegistryComponent" />
+      <?ifdef var.ProductComponents?>
+      <ComponentGroupRef Id="ProductComponents" />
+      <?endif?>
     </Feature>
   </Package>
 
diff --git a/src/Wix.Build.targets b/src/Wix.Build.targets
index e6018b67..3fe78ae3 100644
--- a/src/Wix.Build.targets
+++ b/src/Wix.Build.targets
@@ -7,6 +7,8 @@
     <DefineConstants>TestGroupName=$(TestGroupName);PackageName=$(PackageName);BundleName=$(BundleName);$(DefineConstants)</DefineConstants>
     <DefineConstants Condition=" '$(BA)' != '' ">BA=$(BA);$(DefineConstants)</DefineConstants>
     <DefineConstants Condition=" '$(CabPrefix)' != '' ">CabPrefix=$(CabPrefix);$(DefineConstants)</DefineConstants>
+    <DefineConstants Condition=" '$(ProductCode)' != '' ">ProductCode=$(ProductCode);$(DefineConstants)</DefineConstants>
+    <DefineConstants Condition=" '$(ProductComponentsRef)' == 'true' ">ProductComponents=1;$(DefineConstants)</DefineConstants>
     <DefineConstants Condition=" '$(UpgradeCode)' != '' ">UpgradeCode=$(UpgradeCode);$(DefineConstants)</DefineConstants>
     <DefineConstants Condition=" '$(Version)' != '' ">Version=$(Version);$(DefineConstants)</DefineConstants>
   </PropertyGroup>
diff --git a/src/WixToolsetTest.BurnE2E/BurnE2ETests.cs b/src/WixToolsetTest.BurnE2E/BurnE2ETests.cs
index 8d1cca23..f3b3e583 100644
--- a/src/WixToolsetTest.BurnE2E/BurnE2ETests.cs
+++ b/src/WixToolsetTest.BurnE2E/BurnE2ETests.cs
@@ -21,9 +21,9 @@ namespace WixToolsetTest.BurnE2E
             return installer;
         }
 
-        protected PackageInstaller CreatePackageInstaller(string name)
+        protected PackageInstaller CreatePackageInstaller(string filename)
         {
-            var installer = new PackageInstaller(this.TestContext, name);
+            var installer = new PackageInstaller(this.TestContext, filename);
             this.Installers.Enqueue(installer);
             return installer;
         }
diff --git a/src/WixToolsetTest.BurnE2E/PackageInstaller.cs b/src/WixToolsetTest.BurnE2E/PackageInstaller.cs
index 95d6cd90..bd1e30cc 100644
--- a/src/WixToolsetTest.BurnE2E/PackageInstaller.cs
+++ b/src/WixToolsetTest.BurnE2E/PackageInstaller.cs
@@ -8,11 +8,10 @@ namespace WixToolsetTest.BurnE2E
 
     public partial class PackageInstaller : IDisposable
     {
-        public PackageInstaller(WixTestContext testContext, string name)
+        public PackageInstaller(WixTestContext testContext, string filename)
         {
-            this.Package = Path.Combine(testContext.TestDataFolder, $"{name}.msi");
-            this.PackagePdb = Path.Combine(testContext.TestDataFolder, $"{name}.wixpdb");
-            this.PackageName = name;
+            this.Package = Path.Combine(testContext.TestDataFolder, $"{filename}.msi");
+            this.PackagePdb = Path.Combine(testContext.TestDataFolder, $"{filename}.wixpdb");
             this.TestContext = testContext;
         }
 
diff --git a/src/WixToolsetTest.BurnE2E/PackageVerifier.cs b/src/WixToolsetTest.BurnE2E/PackageVerifier.cs
index 7b4bbfef..7a4fea88 100644
--- a/src/WixToolsetTest.BurnE2E/PackageVerifier.cs
+++ b/src/WixToolsetTest.BurnE2E/PackageVerifier.cs
@@ -12,15 +12,13 @@ namespace WixToolsetTest.BurnE2E
 
     public partial class PackageInstaller
     {
-        private string PackageName { get; }
-
         public string PackagePdb { get; }
 
         private WindowsInstallerData WiData { get; set; }
 
         public string GetInstalledFilePath(string filename)
         {
-            return this.TestContext.GetTestInstallFolder(Path.Combine(this.PackageName, filename));
+            return this.TestContext.GetTestInstallFolder(Path.Combine(this.GetInstallFolderName(), filename));
         }
 
         private WindowsInstallerData GetWindowsInstallerData()
@@ -34,6 +32,19 @@ namespace WixToolsetTest.BurnE2E
             return this.WiData;
         }
 
+        public string GetInstallFolderName()
+        {
+            var wiData = this.GetWindowsInstallerData();
+            var row = wiData.Tables["Directory"].Rows.Single(r => r.FieldAsString(0) == "INSTALLFOLDER");
+            var value = row.FieldAsString(2);
+            var longNameIndex = value.IndexOf('|') + 1;
+            if (longNameIndex > 0)
+            {
+                return value.Substring(longNameIndex);
+            }
+            return value;
+        }
+
         public string GetProperty(string name)
         {
             var wiData = this.GetWindowsInstallerData();
@@ -46,5 +57,21 @@ namespace WixToolsetTest.BurnE2E
             var productCode = this.GetProperty("ProductCode");
             Assert.Equal(installed, MsiUtilities.IsProductInstalled(productCode));
         }
+
+        public void VerifyTestRegistryRootDeleted()
+        {
+            using var testRegistryRoot = this.TestContext.GetTestRegistryRoot();
+            Assert.Null(testRegistryRoot);
+        }
+
+        public void VerifyTestRegistryValue(string name, string expectedValue)
+        {
+            using (var root = this.TestContext.GetTestRegistryRoot())
+            {
+                Assert.NotNull(root);
+                var actualValue = root.GetValue(name) as string;
+                Assert.Equal(expectedValue, actualValue);
+            }
+        }
     }
 }
diff --git a/src/WixToolsetTest.BurnE2E/PatchTests.cs b/src/WixToolsetTest.BurnE2E/PatchTests.cs
new file mode 100644
index 00000000..19adf8db
--- /dev/null
+++ b/src/WixToolsetTest.BurnE2E/PatchTests.cs
@@ -0,0 +1,137 @@
+// 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.
+
+namespace WixToolsetTest.BurnE2E
+{
+    using System;
+    using System.IO;
+    using System.Xml;
+    using Xunit;
+    using Xunit.Abstractions;
+
+    public class PatchTests : BurnE2ETests
+    {
+        public PatchTests(ITestOutputHelper testOutputHelper) : base(testOutputHelper) { }
+
+        [Fact]
+        public void CanRunDetectMultipleTimesWithPatches()
+        {
+            var testBAController = this.CreateTestBAController();
+            testBAController.SetRedetectCount(1);
+
+            this.CanInstallBundleWithPatchThenRemoveIt();
+        }
+
+        [Fact]
+        public void CanInstallBundleWithPatchThenRemoveIt()
+        {
+            var originalVersion = "1.0.0.0";
+            var patchedVersion = "1.0.1.0";
+            var testRegistryValue = "PackageA";
+
+            var packageAv1 = this.CreatePackageInstaller("PackageAv1");
+            var bundleA = this.CreateBundleInstaller("BundleA");
+            var bundlePatchA = this.CreateBundleInstaller("BundlePatchA");
+
+            bundleA.Install();
+            bundleA.VerifyRegisteredAndInPackageCache();
+
+            packageAv1.VerifyInstalled(true);
+            packageAv1.VerifyTestRegistryValue(testRegistryValue, originalVersion);
+
+            bundlePatchA.Install();
+            bundlePatchA.VerifyRegisteredAndInPackageCache();
+
+            packageAv1.VerifyTestRegistryValue(testRegistryValue, patchedVersion);
+
+            bundlePatchA.Uninstall();
+            bundlePatchA.VerifyUnregisteredAndRemovedFromPackageCache();
+
+            packageAv1.VerifyTestRegistryValue(testRegistryValue, originalVersion);
+
+            bundleA.Uninstall();
+            bundleA.VerifyUnregisteredAndRemovedFromPackageCache();
+
+            packageAv1.VerifyInstalled(false);
+            packageAv1.VerifyTestRegistryRootDeleted();
+        }
+
+        [Fact(Skip = "https://github.com/wixtoolset/issues/issues/5949")]
+        public void CanPatchSwidTag()
+        {
+            var originalVersion = "1.0.0.0";
+            var patchedVersion = "1.0.1.0";
+            var packageTagName = "~PatchTests - PackageA";
+            var bundleTagName = "~PatchTests - BundleA";
+            var bundlePatchTagName = "~PatchTests - BundlePatchA";
+
+            this.CreatePackageInstaller("PackageAv1");
+            var bundleA = this.CreateBundleInstaller("BundleA");
+            var bundlePatchA = this.CreateBundleInstaller("BundlePatchA");
+
+            bundleA.Install();
+            VerifySwidTagVersion(bundleTagName, originalVersion);
+            VerifySwidTagVersion(packageTagName, originalVersion);
+
+            bundlePatchA.Install();
+            VerifySwidTagVersion(bundlePatchTagName, patchedVersion);
+            VerifySwidTagVersion(packageTagName, patchedVersion);
+
+            bundlePatchA.Uninstall();
+            VerifySwidTagVersion(packageTagName, originalVersion);
+
+            bundleA.Uninstall();
+            VerifySwidTagVersion(bundleTagName, null);
+            VerifySwidTagVersion(packageTagName, null);
+        }
+
+        [Fact]
+        public void CanInstallBundleWithPatchesTargetingSingleProductThenRemoveIt()
+        {
+            var originalVersion = "1.0.0.0";
+            var patchedVersion = "1.0.1.0";
+            var testRegistryValue = "PackageA";
+            var testRegistryValue2 = "PackageA2";
+
+            var packageAv1 = this.CreatePackageInstaller("PackageAv1");
+            var bundlePatchA2 = this.CreateBundleInstaller("BundlePatchA2");
+
+            packageAv1.InstallProduct();
+            packageAv1.VerifyInstalled(true);
+            packageAv1.VerifyTestRegistryValue(testRegistryValue, originalVersion);
+            packageAv1.VerifyTestRegistryValue(testRegistryValue2, originalVersion);
+
+            bundlePatchA2.Install();
+            bundlePatchA2.VerifyRegisteredAndInPackageCache();
+
+            packageAv1.VerifyTestRegistryValue(testRegistryValue, patchedVersion);
+            packageAv1.VerifyTestRegistryValue(testRegistryValue2, patchedVersion);
+
+            bundlePatchA2.Uninstall();
+            bundlePatchA2.VerifyUnregisteredAndRemovedFromPackageCache();
+
+            packageAv1.VerifyTestRegistryValue(testRegistryValue, originalVersion);
+            packageAv1.VerifyTestRegistryValue(testRegistryValue2, originalVersion);
+        }
+
+        private static void VerifySwidTagVersion(string tagName, string expectedVersion)
+        {
+            var regidFolder = Environment.ExpandEnvironmentVariables(@"%ProgramData%\regid.1995-08.com.example");
+            var tagPath = Path.Combine(regidFolder, "regid.1995-08.com.example " + tagName + ".swidtag");
+            string version = null;
+
+            if (File.Exists(tagPath))
+            {
+                var doc = new XmlDocument();
+                doc.Load(tagPath);
+
+                var ns = new XmlNamespaceManager(doc.NameTable);
+                ns.AddNamespace("s", "http://standards.iso.org/iso/19770/-2/2009/schema.xsd");
+
+                var versionNode = doc.SelectSingleNode("/s:software_identification_tag/s:product_version/s:name", ns);
+                version = versionNode?.InnerText ?? String.Empty;
+            }
+
+            Assert.Equal(expectedVersion, version);
+        }
+    }
+}
diff --git a/src/WixToolsetTest.BurnE2E/WixTestContext.cs b/src/WixToolsetTest.BurnE2E/WixTestContext.cs
index 5620c6a7..0ae99a77 100644
--- a/src/WixToolsetTest.BurnE2E/WixTestContext.cs
+++ b/src/WixToolsetTest.BurnE2E/WixTestContext.cs
@@ -57,7 +57,7 @@ namespace WixToolsetTest.BurnE2E
         /// </remarks>
         public RegistryKey GetTestRegistryRoot(string additionalPath = null)
         {
-            var key = String.Format(@"Software\WiX\Tests\{0}\{1}", this.TestName, additionalPath ?? String.Empty);
+            var key = String.Format(@"Software\WOW6432Node\WiX\Tests\{0}\{1}", this.TestGroupName, additionalPath ?? String.Empty);
             return Registry.LocalMachine.OpenSubKey(key, true);
         }
 
diff --git a/src/WixToolsetTest.BurnE2E/WixToolsetTest.BurnE2E.csproj b/src/WixToolsetTest.BurnE2E/WixToolsetTest.BurnE2E.csproj
index 86ad4410..3ffc892f 100644
--- a/src/WixToolsetTest.BurnE2E/WixToolsetTest.BurnE2E.csproj
+++ b/src/WixToolsetTest.BurnE2E/WixToolsetTest.BurnE2E.csproj
@@ -16,7 +16,7 @@
     <PackageReference Include="Microsoft.Win32.Registry" Version="5.0.0" />
     <PackageReference Include="System.Security.Principal.Windows" Version="5.0.0" />
     <PackageReference Include="WixBuildTools.TestSupport" Version="4.0.48" />
-    <PackageReference Include="WixToolset.Data" Version="4.0.180" />
+    <PackageReference Include="WixToolset.Data" Version="4.0.181" />
     <PackageReference Include="WixToolset.Mba.Core" Version="4.0.47" />
   </ItemGroup>