113 lines
3.6 KiB
C#
113 lines
3.6 KiB
C#
// Licensed to the .NET Foundation under one or more agreements.
|
|
// The .NET Foundation licenses this file to you under the MIT license.
|
|
// See the LICENSE file in the project root for more information.
|
|
|
|
using System;
|
|
using System.Diagnostics;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using Xunit.Abstractions;
|
|
|
|
namespace Microsoft.DotNet.SourceBuild.SmokeTests;
|
|
|
|
internal static class ExecuteHelper
|
|
{
|
|
public static (Process Process, string StdOut, string StdErr) ExecuteProcess(
|
|
string fileName,
|
|
string args,
|
|
ITestOutputHelper outputHelper,
|
|
bool logOutput = false,
|
|
Action<Process>? configure = null,
|
|
int millisecondTimeout = -1)
|
|
{
|
|
outputHelper.WriteLine($"Executing: {fileName} {args}");
|
|
|
|
Process process = new()
|
|
{
|
|
EnableRaisingEvents = true,
|
|
StartInfo =
|
|
{
|
|
FileName = fileName,
|
|
Arguments = args,
|
|
RedirectStandardOutput = true,
|
|
RedirectStandardError = true,
|
|
}
|
|
};
|
|
|
|
// The `dotnet test` execution context sets a number of dotnet related ENVs that cause issues when executing
|
|
// dotnet commands. Clear these to avoid side effects.
|
|
foreach (string key in process.StartInfo.Environment.Keys.Where(key => key != "HOME").ToList())
|
|
{
|
|
process.StartInfo.Environment.Remove(key);
|
|
}
|
|
|
|
configure?.Invoke(process);
|
|
|
|
StringBuilder stdOutput = new();
|
|
process.OutputDataReceived += new DataReceivedEventHandler(
|
|
(sender, e) =>
|
|
{
|
|
lock (stdOutput)
|
|
{
|
|
stdOutput.AppendLine(e.Data);
|
|
}
|
|
});
|
|
|
|
StringBuilder stdError = new();
|
|
process.ErrorDataReceived += new DataReceivedEventHandler(
|
|
(sender, e) =>
|
|
{
|
|
lock (stdError)
|
|
{
|
|
stdError.AppendLine(e.Data);
|
|
}
|
|
});
|
|
|
|
process.Start();
|
|
process.BeginOutputReadLine();
|
|
process.BeginErrorReadLine();
|
|
process.WaitForExit(millisecondTimeout);
|
|
|
|
if (!process.HasExited)
|
|
{
|
|
outputHelper.WriteLine($"Killing: {fileName} {args}");
|
|
process.Kill(true);
|
|
process.WaitForExit();
|
|
}
|
|
|
|
string output = stdOutput.ToString().Trim();
|
|
if (logOutput && !string.IsNullOrWhiteSpace(output))
|
|
{
|
|
outputHelper.WriteLine(output);
|
|
}
|
|
|
|
string error = stdError.ToString().Trim();
|
|
if (logOutput && !string.IsNullOrWhiteSpace(error))
|
|
{
|
|
outputHelper.WriteLine(error);
|
|
}
|
|
|
|
return (process, output, error);
|
|
}
|
|
|
|
public static string ExecuteProcessValidateExitCode(string fileName, string args, ITestOutputHelper outputHelper)
|
|
{
|
|
(Process Process, string StdOut, string StdErr) result = ExecuteHelper.ExecuteProcess(fileName, args, outputHelper);
|
|
ValidateExitCode(result);
|
|
|
|
return result.StdOut;
|
|
}
|
|
|
|
public static void ValidateExitCode((Process Process, string StdOut, string StdErr) result, int expectedExitCode = 0)
|
|
{
|
|
if (result.Process.ExitCode != expectedExitCode)
|
|
{
|
|
ProcessStartInfo startInfo = result.Process.StartInfo;
|
|
string msg = $"Failed to execute {startInfo.FileName} {startInfo.Arguments}" +
|
|
$"{Environment.NewLine}Exit code: {result.Process.ExitCode}" +
|
|
$"{Environment.NewLine}{result.StdOut}" +
|
|
$"{Environment.NewLine}{result.StdErr}";
|
|
throw new InvalidOperationException(msg);
|
|
}
|
|
}
|
|
}
|