c# - 如何对使用 Response.OnStarting 的 .NET 中间件进行单元测试

Question

因此，我创建了一个 .net 核心中间件，它为响应添加了 header 。
我没有找到对这个中间件进行单元测试的方法，因为它使用 OnStarting回调，我无法想象如何模拟或强制执行它。

这是中间件的示例:

public async Task InvokeAsync(HttpContext context, RequestDelegate next)
{
    context.Response.OnStarting(() =>
    {
        context.Response.Headers.Add("Name", "Value");

        return Task.CompletedTask;
    });

    await next(context);
}

由于中间件接口(interface)需要 HttpContext 的具体实例，我无法想象使用 FakeItEasy 模拟它的方法， 例如。

Answer 1

您可以使用 Mock HttpContext for unit testing a .NET core MVC controller? 中的示例作为指南:

using Shouldly;
using Xunit;

[Fact]
public async Task Middleware_should_add_header()
{
    // Arrange:

    HttpContext ctx = new DefaultHttpContext();

    RequestDelegate next = ( HttpContext hc ) => Task.CompletedTask;

    MyMiddleware mw = new MyMiddleware();

    // Act - Part 1: InvokeAsync set-up:

    await hw.InvokeAsync( ctx, next );

    // Assert - Part 1

    ctx.Response.Headers.TryGetValue( "Name", out String value0 ).ShouldBeFalse();
    value0.ShouldBeNull();

    // Act - Part 2

    await ctx.Response.StartAsync( default );

    // Asset - Part 2

    ctx.Response.Headers.TryGetValue( "Name", out String value1 ).ShouldBeTrue();
    value1.ShouldBe( "Value" );

}