Implement ArpEntry flavored ExePackage.

6772

1 files changed, 17 insertions, 0 deletions

diff --git a/src/test/burn/WixToolsetTest.BurnE2E/BurnE2ETests.cs b/src/test/burn/WixToolsetTest.BurnE2E/BurnE2ETests.cs
index 6664b849..5caedacb 100644
--- a/src/test/burn/WixToolsetTest.BurnE2E/BurnE2ETests.cs
+++ b/src/test/burn/WixToolsetTest.BurnE2E/BurnE2ETests.cs
@@ -31,6 +31,23 @@ namespace WixToolsetTest.BurnE2E
             this.Installers.Push(installer);
         }
 
+        protected ArpEntryInstaller CreateArpEntryInstaller(string id, bool perMachine = true, bool x64 = false)
+        {
+            var installer = new ArpEntryInstaller(this.TestContext, id, perMachine, x64);
+            this.Installers.Push(installer);
+            return installer;
+        }
+
+        protected ArpEntryInstaller CreateArpEntryInstaller(BundleInstaller bundleInstaller, string packageId)
+        {
+            if (!bundleInstaller.TryGetArpEntryExePackageConfiguration(packageId, out var arpId, out _, out var arpWin64, out var perMachine))
+            {
+                return null;
+            }
+
+            return this.CreateArpEntryInstaller(arpId, perMachine, arpWin64);
+        }
+
         protected BundleInstaller CreateBundleInstaller(string name)
         {
             var installer = new BundleInstaller(this.TestContext, name);