39 lines
1.2 KiB
C#
39 lines
1.2 KiB
C#
|
using System.CommandLine;
|
|||
|
|
|
|||
|
|
namespace Gpulse.WCT.DataAnalyzer.Tests;
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 命令执行辅助：运行命令并捕获标准输出/标准错误。
|
|||
|
|
/// 命令 handler 直接使用 Console.WriteLine，因此重定向 System.Console 即可捕获；
|
|||
|
|
/// System.CommandLine 自身的帮助/校验输出也会经默认控制台写入同一流。
|
|||
|
|
/// </summary>
|
|||
|
|
public static class CliTestExtensions
|
|||
|
|
{
|
|||
|
|
/// <summary>
|
|||
|
|
/// 执行命令并返回 (退出码, 标准输出, 标准错误)。
|
|||
|
|
/// </summary>
|
|||
|
|
public static async Task<(int ExitCode, string StdOut, string StdErr)> RunCaptureAsync(
|
|||
|
|
this Command command,
|
|||
|
|
params string[] args)
|
|||
|
|
{
|
|||
|
|
var originalOut = Console.Out;
|
|||
|
|
var originalError = Console.Error;
|
|||
|
|
|
|||
|
|
using var outWriter = new StringWriter();
|
|||
|
|
using var errWriter = new StringWriter();
|
|||
|
|
|
|||
|
|
Console.SetOut(outWriter);
|
|||
|
|
Console.SetError(errWriter);
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
var exitCode = await command.InvokeAsync(args);
|
|||
|
|
return (exitCode, outWriter.ToString(), errWriter.ToString());
|
|||
|
|
}
|
|||
|
|
finally
|
|||
|
|
{
|
|||
|
|
Console.SetOut(originalOut);
|
|||
|
|
Console.SetError(originalError);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|