Question

我正在尝试使用DefaultHttpContext对象对我的异常处理中间件进行单元测试。

我的测试方法如下：

[Fact]
public async Task Invoke_ProductionNonSuredException_ReturnsProductionRequestError()
{
    var logger = new Mock<ILogger<ExceptionHandlerMiddleware>>();
    var middleWare = new ExceptionHandlerMiddleware(next: async (innerHttpContext) =>
    {
        await Task.Run(() =>
        {
            throw new Exception();
        });
    }, logger: logger.Object);

    var mockEnv = new Mock<IHostingEnvironment>();
    mockEnv.Setup(u => u.EnvironmentName).Returns("Production");

    var context = new DefaultHttpContext();

    await middleWare.Invoke(context, mockEnv.Object);

    var reader = new StreamReader(context.Response.Body);
    var streamText = reader.ReadToEnd();

    //TODO: write assert that checks streamtext is the expected production return type and not the verbose development environment version.
}

在我的中间件中，我正在写这样的上下文：

public static Task WriteResponse(HttpContext context, HttpStatusCode statusCode, object responseData, Formatting jsonFormatting)
{
    context.Response.ContentType = "application/json";
    context.Response.StatusCode = (int)statusCode;
    return context.Response.WriteAsync(JsonConvert.SerializeObject(responseData, jsonFormatting));
}

为了让您更深入地了解我采用的中间件方法，我采用this answer here中的方法。

当应用运行正常管道时，工作正常。但是，在测试中使用DefaultHttpContext方法时，响应主体总是返回空，ContentLength为空。因此，测试中的streamText变量是一个空字符串。

在这种情况下，是否可以检查中间件写入上下文的内容？这是适当的方式，还是有更好的方法。

Answer 1

考虑自己设置身体，以便控制流

由@ AndrewStanton-Nurse提供的评论

Response.Body中的DefaultHttpContext流是Stream.Null，这是一个忽略所有读/写的流。您需要在调用方法之前自己设置Stream。

此外 - 这将允许设置正文，但为了正确读取它，我们必须在使用StreamReader之前将指针设置为this answer的开头。

//...code removed for brevity

var context = new DefaultHttpContext();
context.Response.Body = new MemoryStream();

await middleWare.Invoke(context, mockEnv.Object);


context.Response.Body.Seek(0, SeekOrigin.Begin);
var reader = new StreamReader(context.Response.Body);
var streamText = reader.ReadToEnd();

//...code removed for brevity

在单元测试情况下检查DefaultHttpContext主体

1 个答案: