Implement ArpEntry flavored ExePackage.

6772

1 files changed, 24 insertions, 0 deletions

diff --git a/src/test/burn/WixTestTools/ArpEntryVerifier.cs b/src/test/burn/WixTestTools/ArpEntryVerifier.cs
new file mode 100644
index 00000000..b3c70337
--- /dev/null
+++ b/src/test/burn/WixTestTools/ArpEntryVerifier.cs
@@ -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.
+
+namespace WixTestTools
+{
+    using Xunit;
+
+    public partial class ArpEntryInstaller
+    {
+        public bool TryGetRegistration(out GenericArpRegistration registration)
+        {
+            bool success = !this.PerMachine ? GenericArpRegistration.TryGetPerUserRegistrationById(this.ArpId, out registration)
+                                            : GenericArpRegistration.TryGetPerMachineRegistrationById(this.ArpId, this.X64, out registration);
+
+            return success;
+        }
+
+        public void VerifyRegistered(bool registered)
+        {
+            bool success = this.TryGetRegistration(out _);
+
+            Assert.Equal(registered, success);
+        }
+    }
+}