Hello GitHub

internal/Calculator.UIAutomationLibrary/Utilities/EtwHelper.cs

@@ -0,0 +1,101 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+using System;
+using System.Collections.Generic;
+using System.Diagnostics;
+using System.IO;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using Microsoft.OneCoreUap.Test.AppModel;
+using WEX.Logging.Interop;
+
+namespace Calculator.UIAutomationLibrary
+{
+    public class EtwHelper
+    {
+        private static bool etwServiceWasInstalled = false;
+
+        /// <summary>
+        /// Installs and starts the Etw.Service so that Tests may utilize Etw Waiters.
+        /// Wex.Services.exe is part of TAEF.
+        /// </summary>
+        public static void InstallAndStartETWService()
+        {
+            etwServiceWasInstalled = ServiceHelper.IsInstalled("Etw.Service");
+            if (!etwServiceWasInstalled)
+            {
+                string wexServices = Path.Combine(TAEFHelper.GetTestDeploymentDirectory(), "Wex.Services.exe");
+                if (!File.Exists(wexServices))
+                {
+                    throw new FileNotFoundException(wexServices);
+                }
+
+                Log.Comment("Attempting to install Etw.Service...");
+                var startInfo = new ProcessStartInfo();
+                startInfo.FileName = wexServices;
+                startInfo.Arguments = "/install:Etw.Service";
+
+                var process = new Process();
+                process.StartInfo = startInfo;
+                if (process.Start())
+                {
+                    process.WaitForExit();
+                    Log.Comment("Completed installation of Etw.Service");
+                }
+                else
+                {
+                    throw new Exception("ETW service was not able to be installed. Process didn't start.");
+                }
+            }
+
+            Log.Comment("Attempting to start Etw.Service...");
+            ServiceHelper.Start("Etw.Service");
+            Log.Comment("Etw.Service started");
+        }
+
+        /// <summary>
+        /// Stops the Etw.Service.
+        /// </summary>
+        public static void StopAndRemoveETWService()
+        {
+            if (ServiceHelper.IsInstalled("Etw.Service"))
+            {
+                Log.Comment("Attempting to stop Etw.Service...");
+                ServiceHelper.Stop("Etw.Service");
+                Log.Comment("Etw.Service stopped");
+
+                // if we installed the Etw.Service as part of this test we should also remove it.
+                // This prevents cases where TDP is upgraded but the service tregistration is referencing the old
+                // location that no longer exists.
+                if (!etwServiceWasInstalled)
+                {
+
+                    string wexServices = Path.Combine(TAEFHelper.GetTestDeploymentDirectory(), "Wex.Services.exe");
+                    if (!File.Exists(wexServices))
+                    {
+                        throw new FileNotFoundException(wexServices);
+                    }
+
+                    Log.Comment("Attempting to remove Etw.Service...");
+                    var startInfo = new ProcessStartInfo();
+                    startInfo.FileName = wexServices;
+                    startInfo.Arguments = "/remove:Etw.Service";
+
+                    var process = new Process();
+                    process.StartInfo = startInfo;
+                    if (process.Start())
+                    {
+                        process.WaitForExit();
+                        Log.Comment("Completed removal of Etw.Service");
+                    }
+                    else
+                    {
+                        throw new Exception("ETW service could not be removed. Process didn't start.");
+                    }
+                }
+            }
+        }
+    }
+}