102 lines
3.7 KiB
C#
102 lines
3.7 KiB
C#
// 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.");
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|