Move Bal.wixext into ext

1 files changed, 74 insertions, 0 deletions

diff --git a/src/ext/Bal/test/WixToolsetTest.ManagedHost/TestEngine.cs b/src/ext/Bal/test/WixToolsetTest.ManagedHost/TestEngine.cs
new file mode 100644
index 00000000..44538227
--- /dev/null
+++ b/src/ext/Bal/test/WixToolsetTest.ManagedHost/TestEngine.cs
@@ -0,0 +1,74 @@
+// 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.ManagedHost
+{
+    using System.Collections.Generic;
+    using System.Diagnostics;
+    using System.IO;
+    using WixBuildTools.TestSupport;
+    using WixToolset.Core.TestPackage;
+
+    public class TestEngine
+    {
+        private static readonly string TestEngineFile = TestData.Get(@"..\Win32\examples\Example.TestEngine\Example.TestEngine.exe");
+
+        public TestEngineResult RunReloadEngine(string bundleFilePath, string tempFolderPath)
+        {
+            return this.RunTestEngine("reload", bundleFilePath, tempFolderPath);
+        }
+
+        public TestEngineResult RunShutdownEngine(string bundleFilePath, string tempFolderPath)
+        {
+            return this.RunTestEngine("shutdown", bundleFilePath, tempFolderPath);
+        }
+
+        private TestEngineResult RunTestEngine(string engineMode, string bundleFilePath, string tempFolderPath)
+        {
+            var baFolderPath = Path.Combine(tempFolderPath, "ba");
+            var extractFolderPath = Path.Combine(tempFolderPath, "extract");
+            var extractResult = BundleExtractor.ExtractBAContainer(null, bundleFilePath, baFolderPath, extractFolderPath);
+            extractResult.AssertSuccess();
+
+            var args = new string[] {
+                engineMode,
+                '"' + bundleFilePath + '"',
+                '"' + extractResult.GetBAFilePath(baFolderPath) + '"',
+            };
+            return RunProcessCaptureOutput(TestEngineFile, args);
+        }
+
+        private static TestEngineResult RunProcessCaptureOutput(string executablePath, string[] arguments = null, string workingFolder = null)
+        {
+            var startInfo = new ProcessStartInfo(executablePath)
+            {
+                Arguments = string.Join(' ', arguments),
+                CreateNoWindow = true,
+                RedirectStandardError = true,
+                RedirectStandardOutput = true,
+                UseShellExecute = false,
+                WorkingDirectory = workingFolder,
+            };
+
+            var exitCode = 0;
+            var output = new List<string>();
+
+            using (var process = Process.Start(startInfo))
+            {
+                process.OutputDataReceived += (s, e) => { if (e.Data != null) { output.Add(e.Data); } };
+                process.ErrorDataReceived += (s, e) => { if (e.Data != null) { output.Add(e.Data); } };
+
+                process.BeginErrorReadLine();
+                process.BeginOutputReadLine();
+
+                process.WaitForExit();
+                exitCode = process.ExitCode;
+            }
+
+            return new TestEngineResult
+            {
+                ExitCode = exitCode,
+                Output = output,
+            };
+        }
+    }
+}