1 files changed, 68 insertions, 0 deletions

diff --git a/src/test/burn/WixToolsetTest.BurnE2E/LayoutTests.cs b/src/test/burn/WixToolsetTest.BurnE2E/LayoutTests.cs
new file mode 100644
index 00000000..1e36e2a5
--- /dev/null
+++ b/src/test/burn/WixToolsetTest.BurnE2E/LayoutTests.cs
@@ -0,0 +1,68 @@
+// 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.Collections.Generic;
+    using System.IO;
+    using WixBuildTools.TestSupport;
+    using Xunit;
+    using Xunit.Abstractions;
+
+    public class LayoutTests : BurnE2ETests
+    {
+        public LayoutTests(ITestOutputHelper testOutputHelper) : base(testOutputHelper) { }
+
+        [Fact]
+        public void CanLayoutBundleInPlaceWithMissingPayloads()
+        {
+            var bundleA = this.CreateBundleInstaller("BundleA");
+            var webServer = this.CreateWebServer();
+
+            webServer.AddFiles(new Dictionary<string, string>
+            {
+                { "/BundleA/LayoutOnlyPayload", Path.Combine(this.TestContext.TestDataFolder, "BundleA.wxs") },
+                { "/BundleA/packages.cab", Path.Combine(this.TestContext.TestDataFolder, "packages.cab") },
+            });
+            webServer.Start();
+
+            using var dfs = new DisposableFileSystem();
+            var layoutDirectory = dfs.GetFolder(true);
+
+            // Manually copy bundle to layout directory and then run from there so the non-compressed payloads have to be resolved.
+            var bundleAFileInfo = new FileInfo(bundleA.Bundle);
+            var bundleACopiedPath = Path.Combine(layoutDirectory, bundleAFileInfo.Name);
+            bundleAFileInfo.CopyTo(bundleACopiedPath);
+
+            bundleA.Layout(bundleACopiedPath, layoutDirectory);
+            bundleA.VerifyUnregisteredAndRemovedFromPackageCache();
+
+            Assert.True(File.Exists(bundleACopiedPath));
+            Assert.True(File.Exists(Path.Combine(layoutDirectory, "packages.cab")));
+            Assert.True(File.Exists(Path.Combine(layoutDirectory, "BundleA.wxs")));
+        }
+
+        [Fact]
+        public void CanLayoutBundleToNewDirectory()
+        {
+            var bundleA = this.CreateBundleInstaller("BundleA");
+            var webServer = this.CreateWebServer();
+
+            webServer.AddFiles(new Dictionary<string, string>
+            {
+                { "/BundleA/LayoutOnlyPayload", Path.Combine(this.TestContext.TestDataFolder, "BundleA.wxs") },
+                { "/BundleA/packages.cab", Path.Combine(this.TestContext.TestDataFolder, "packages.cab") },
+            });
+            webServer.Start();
+
+            using var dfs = new DisposableFileSystem();
+            var layoutDirectory = dfs.GetFolder();
+
+            bundleA.Layout(layoutDirectory);
+            bundleA.VerifyUnregisteredAndRemovedFromPackageCache();
+
+            Assert.True(File.Exists(Path.Combine(layoutDirectory, "BundleA.exe")));
+            Assert.True(File.Exists(Path.Combine(layoutDirectory, "packages.cab")));
+            Assert.True(File.Exists(Path.Combine(layoutDirectory, "BundleA.wxs")));
+        }
+    }
+}