配置AspNetCore TestServer返回500而不是引发异常

时间:2018-11-28 07:40:58

标签: c# asp.net-core integration-testing xunit.net

我正在开发一个Web API,在某些情况下它会响应500(我知道这是丑陋的设计,但对此无能为力)。在测试中,有一个包含AspNetCore.TestHost的ApiFixture:

public class ApiFixture
{
    public TestServer ApiServer { get; }
    public HttpClient HttpClient { get; }

    public ApiFixture()
    {
        var config = new ConfigurationBuilder()
            .AddEnvironmentVariables()
            .Build();

        var path = Assembly.GetAssembly(typeof(ApiFixture)).Location;
        var hostBuilder = new WebHostBuilder()
            .UseContentRoot(Path.GetDirectoryName(path))
            .UseConfiguration(config)
            .UseStartup<Startup>();

        ApiServer = new TestServer(hostBuilder);
        HttpClient = ApiServer.CreateClient();
    }
}

当我从这个灯具中使用HttpClient调用API端点时,它应该以500响应,而我正在测试的控制器中引发异常。我知道在测试中这可能是一个不错的功能,但我不希望这样做-我想测试控制器的实际行为,这将返回内部服务器错误。有没有一种方法可以重新配置TestServer以返回响应?

控制器操作中的代码无关,可以为throw new Exception();

1 个答案:

答案 0 :(得分:4)

您可以创建一个异常处理中间件,并在测试中使用它,或者更好地始终使用它

public class ExceptionMiddleware
{
    private readonly RequestDelegate next;

    public ExceptionMiddleware(RequestDelegate next)
    {
        this.next = next;
    }

    public async Task Invoke(HttpContext httpContext)
    {
        try
        {
            await this.next(httpContext);
        }
        catch (Exception ex)
        {
            httpContext.Response.ContentType = MediaTypeNames.Text.Plain;
            httpContext.Response.StatusCode = (int)HttpStatusCode.InternalServerError;
            await httpContext.Response.WriteAsync("Internal server error!");
        }
    }
}

现在您可以在Startup.cs中注册此中间件:

...
public void Configure(IApplicationBuilder app, IHostingEnvironment env)
{
    app.UseMiddleware<ExceptionMiddleware>();
    ...
    app.UseMvc();
}

如果您不想一直使用它,可以创建TestStartup-Startup的子类,并重写Configure方法来调用UseMiddleware只有那里。然后,您仅需要在测试中使用新的TestStartup类。