1 files changed, 11 insertions, 0 deletions

diff --git a/src/test/WixToolsetTest.ManagedHost/TestEngine.cs b/src/test/WixToolsetTest.ManagedHost/TestEngine.cs
index 751ed59c..cda32895 100644
--- a/src/test/WixToolsetTest.ManagedHost/TestEngine.cs
+++ b/src/test/WixToolsetTest.ManagedHost/TestEngine.cs
@@ -13,14 +13,25 @@ namespace WixToolsetTest.ManagedHost
         private static readonly string TestEngineFile = TestData.Get(@"..\Win32\examples\Example.TestEngine\Example.TestEngine.exe");
         public static readonly string BurnStubFile = TestData.Get(@"runtimes\win-x86\native\burn.x86.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) + '"',
             };