如何阻止WebApplicationFactory向Program.Main传递参数

问题场景 #

我有一个ASP.NET应用，Program.Main方法通过CommandLine.Parser解析命令行参数，默认会对未知参数报错。但使用Microsoft.AspNetCore.Mvc.Testing.WebApplicationFactory做集成测试时，Main方法收到了额外参数（如--environment=Development、--contentRoot=/path/to/my/app等），导致参数解析失败。尝试过清除配置源、使用空配置等操作均无效，目前只能临时关闭未知参数报错，希望找到彻底阻止额外参数传递的方案。

应用的Program.Main实现：

public class Program
{
    public static async Task Main(string[] args)
    {
        var parserResult = CommandLine.Parser.Default.ParseArguments<CommandLineOptions>(args);
        if (parserResult.Errors.Any())
            return;

        // 执行其余初始化操作
    }
}

测试代码片段：

var client = _factory
    .WithWebHostBuilder(
        builder =>
        {
            builder.ConfigureTestServices(
                services =>
                {
                    services.AddScoped(sp => mockRepository.Object);
                }
            );
        }
    )
    .CreateClient();
const string url = $"/some/path/to/test";

var response = await client.GetAsync(url);

response.EnsureSuccessStatusCode(); // 状态码 200-299

可行解决方案 #

方案1：自定义WebApplicationFactory，控制传入Main的参数 #

WebApplicationFactory默认会传递主机相关参数，我们可以继承它并手动控制传入Program.Main的参数数组：

public class CustomWebApplicationFactory<TProgram> : WebApplicationFactory<TProgram> 
    where TProgram : class
{
    protected override IHost CreateHost(IHostBuilder builder)
    {
        // 构建主机前，手动调用Program.Main并传入空参数数组
        var mainMethod = typeof(TProgram).GetMethod(
            "Main", 
            BindingFlags.Public | BindingFlags.Static
        );
        var task = (Task)mainMethod.Invoke(null, new object[] { Array.Empty<string>() });
        task.Wait();

        return builder.Build();
    }
}

测试时直接使用CustomWebApplicationFactory替代默认的WebApplicationFactory即可。

方案2：在Main方法中区分环境过滤参数 #

在测试项目中设置ASPNETCORE_ENVIRONMENT=Testing环境变量，然后在Program.Main中判断环境，过滤掉非自定义参数：

public static async Task Main(string[] args)
{
    // 测试环境下仅保留自定义参数
    if (Environment.GetEnvironmentVariable("ASPNETCORE_ENVIRONMENT") == "Testing")
    {
        // 这里替换成你的自定义参数前缀
        args = args.Where(arg => arg.StartsWith("--your-custom-option")).ToArray();
    }
    
    var parserResult = CommandLine.Parser.Default.ParseArguments<CommandLineOptions>(args);
    if (parserResult.Errors.Any())
        return;

    // 执行其余初始化操作
}

方案3：配置CommandLineParser忽略特定参数 #

修改参数解析逻辑，直接过滤WebApplicationFactory传递的默认参数：

var parser = new CommandLine.Parser(config =>
{
    config.IgnoreUnknownArguments = false;
    config.AdditionalArgumentsHandler = args =>
    {
        // 过滤掉主机相关的默认参数
        return args.Where(arg => 
            !arg.StartsWith("--environment=") && 
            !arg.StartsWith("--contentRoot=") && 
            !arg.StartsWith("--applicationName=") &&
            !arg.StartsWith("--ENV_VAR=")
        ).ToArray();
    };
});

var parserResult = parser.ParseArguments<CommandLineOptions>(args);

为什么之前的尝试无效？ #

之前使用ConfigureAppConfiguration或UseConfiguration修改的是应用的配置源，但WebApplicationFactory是直接向Program.Main方法传递命令行参数数组，这些参数在主机构建前就已传入，因此修改配置源无法影响到Main收到的args。

内容的提问来源于stack exchange，提问作者OBones