3 files changed, 158 insertions, 0 deletions

diff --git a/src/WixToolset.Core.TestPackage/BundleExtractor.cs b/src/WixToolset.Core.TestPackage/BundleExtractor.cs
new file mode 100644
index 00000000..3d7b2932
--- /dev/null
+++ b/src/WixToolset.Core.TestPackage/BundleExtractor.cs
@@ -0,0 +1,44 @@
+// 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 WixToolset.Core.TestPackage
+{
+    using System.IO;
+    using System.Xml;
+    using WixToolset.Core.Burn.Bundles;
+    using WixToolset.Extensibility.Services;
+
+    public class BundleExtractor
+    {
+        public static ExtractBAContainerResult ExtractBAContainer(IMessaging messaging, string bundleFilePath, string destinationFolderPath, string tempFolderPath)
+        {
+            var result = new ExtractBAContainerResult();
+            Directory.CreateDirectory(tempFolderPath);
+            using (var burnReader = BurnReader.Open(messaging, bundleFilePath))
+            {
+                result.Success = burnReader.ExtractUXContainer(destinationFolderPath, tempFolderPath);
+            }
+
+            if (result.Success)
+            {
+                result.ManifestDocument = LoadBurnManifest(destinationFolderPath);
+                result.ManifestNamespaceManager = GetBurnNamespaceManager(result.ManifestDocument, "burn");
+            }
+
+            return result;
+        }
+
+        public static XmlNamespaceManager GetBurnNamespaceManager(XmlDocument document, string prefix)
+        {
+            var namespaceManager = new XmlNamespaceManager(document.NameTable);
+            namespaceManager.AddNamespace(prefix, BurnCommon.BurnNamespace);
+            return namespaceManager;
+        }
+
+        public static XmlDocument LoadBurnManifest(string baFolderPath)
+        {
+            var document = new XmlDocument();
+            document.Load(Path.Combine(baFolderPath, "manifest.xml"));
+            return document;
+        }
+    }
+}
diff --git a/src/WixToolset.Core.TestPackage/ExtractBAContainerResult.cs b/src/WixToolset.Core.TestPackage/ExtractBAContainerResult.cs
new file mode 100644
index 00000000..6d2ea943
--- /dev/null
+++ b/src/WixToolset.Core.TestPackage/ExtractBAContainerResult.cs
@@ -0,0 +1,39 @@
+// 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 WixToolset.Core.TestPackage
+{
+    using System.IO;
+    using System.Xml;
+    using Xunit;
+
+    public class ExtractBAContainerResult
+    {
+        public XmlDocument ManifestDocument { get; set; }
+        public XmlNamespaceManager ManifestNamespaceManager { get; set; }
+        public bool Success { get; set; }
+
+        public ExtractBAContainerResult AssertSuccess()
+        {
+            Assert.True(this.Success);
+            return this;
+        }
+
+        public string GetBAFilePath(string extractedBAContainerFolderPath)
+        {
+            var uxPayloads = this.SelectManifestNodes("/burn:BurnManifest/burn:UX/burn:Payload");
+            var baPayload = uxPayloads[0];
+            var relativeBAPath = baPayload.Attributes["FilePath"].Value;
+            return Path.Combine(extractedBAContainerFolderPath, relativeBAPath);
+        }
+
+        /// <summary>
+        /// 
+        /// </summary>
+        /// <param name="xpath">elements must have the 'burn' prefix</param>
+        /// <returns></returns>
+        public XmlNodeList SelectManifestNodes(string xpath)
+        {
+            return this.ManifestDocument.SelectNodes(xpath, this.ManifestNamespaceManager);
+        }
+    }
+}
diff --git a/src/WixToolset.Core.TestPackage/XmlNodeExtensions.cs b/src/WixToolset.Core.TestPackage/XmlNodeExtensions.cs
new file mode 100644
index 00000000..a7f04508
--- /dev/null
+++ b/src/WixToolset.Core.TestPackage/XmlNodeExtensions.cs
@@ -0,0 +1,75 @@
+// 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 WixToolset.Core.TestPackage
+{
+    using System.Collections.Generic;
+    using System.IO;
+    using System.Text.RegularExpressions;
+    using System.Xml;
+
+    public static class XmlNodeExtensions
+    {
+        public static string GetTestXml(this XmlNode node, Dictionary<string, List<string>> ignoredAttributesByElementName = null)
+        {
+            return node.OuterXml.GetTestXml(ignoredAttributesByElementName);
+        }
+
+        public static string GetTestXml(this string xml, Dictionary<string, List<string>> ignoredAttributesByElementName = null)
+        {
+            string formattedXml;
+            using (var sw = new StringWriter())
+            using (var writer = new TestXmlWriter(sw))
+            {
+                var doc = new XmlDocument();
+                doc.LoadXml(xml);
+
+                if (ignoredAttributesByElementName != null)
+                {
+                    HandleIgnoredAttributes(doc, ignoredAttributesByElementName);
+                }
+
+                doc.Save(writer);
+                formattedXml = sw.ToString();
+            }
+
+            return Regex.Replace(formattedXml, " xmlns(:[^=]+)?='[^']*'", "");
+        }
+
+        private static void HandleIgnoredAttributes(XmlNode node, Dictionary<string, List<string>> ignoredAttributesByElementName)
+        {
+            if (node.Attributes != null && ignoredAttributesByElementName.TryGetValue(node.LocalName, out var ignoredAttributes))
+            {
+                foreach (var ignoredAttribute in ignoredAttributes)
+                {
+                    var attribute = node.Attributes[ignoredAttribute];
+                    if (attribute != null)
+                    {
+                        attribute.Value = "*";
+                    }
+                }
+            }
+
+            if (node.ChildNodes != null)
+            {
+                foreach (XmlNode childNode in node.ChildNodes)
+                {
+                    HandleIgnoredAttributes(childNode, ignoredAttributesByElementName);
+                }
+            }
+        }
+
+        private class TestXmlWriter : XmlTextWriter
+        {
+            public TestXmlWriter(TextWriter w)
+                : base(w)
+            {
+                this.QuoteChar = '\'';
+            }
+
+            public override void WriteStartDocument()
+            {
+                //OmitXmlDeclaration
+            }
+        }
+    }
+}