Polly 验证 httpclient 在单元测试中超时/重试的次数

Question

这不是一个完全有效的解决方案，只是使问题更清晰、更具可读性的部分代码。

我已经在这里阅读了 Polly 的文档，但是我真的需要测试在 httpClient 超时后，是否击中了以下 TimeoutPolicy 的onTimeoutAsync委托(此外，如果将 TimeoutPolicy 与 RetryPolicy 一起包装 - 它被击中了多少次)：

public static TimeoutPolicy<HttpResponseMessage> TimeoutPolicy
{
get
{
return Policy.TimeoutAsync<HttpResponseMessage>(1, onTimeoutAsync: (context, timeSpan, task) =>
{
Console.WriteLine("Timeout delegate fired after " + timeSpan.TotalMilliseconds);
return Task.CompletedTask;
});
}
}

超时策略设置为在 1 秒未从 HTTP 客户端接收任何内容后超时(HttpClient 的模拟延迟 4 秒，如下所示)

var httpMessageHandler = new Mock<HttpMessageHandler>();
httpMessageHandler.Protected()
.Setup<Task<HttpResponseMessage>>("SendAsync", ItExpr.IsAny<HttpRequestMessage>(), ItExpr.IsAny<CancellationToken>())
.Callback(() => Thread.Sleep(4000))//Delayed here
.Returns(() => Task.FromResult(new HttpResponseMessage
{
StatusCode = HttpStatusCode.InternalServerError,
Content = new StringContent("Some response body", Encoding.UTF8, "text/xml")
}));
var httpClient = new HttpClient(httpMessageHandler.Object);

策略被嘲笑为能够调用 。验证() 用于断言：

var mockedPolicies = new Mock<Policies>();

我使用模拟的HTTP客户端和模拟的超时策略执行调用：

await mockedPolicies.Object.TimeoutPolicy.ExecuteAsync(ct => _httpClient.PostAsync("", new StringContent("Some request body"), ct), CancellationToken.None);

断言：

mockedPolicies.Verify(p => p.TimeoutDelegate(It.IsAny<Context>(), It.IsAny<TimeSpan>(), It.IsAny<Task>()), Times.Exactly(1));

但是，测试结果显示它已被调用 0 次而不是 1。 谢谢你的任何回答。

Answer 1

问题是发布的测试代码没有返回可取消的任务，其值为SendAsync返回的模拟值。在这方面，它没有反映HttpClient.SendAsync(...)的行为方式。

具有默认TimeoutStrategy.Optimistic的 Polly 超时策略通过超时CancellationToken运行，因此您执行的委托必须响应合作取消。

---

细节：

所有async调用都同步运行，直到第一个await语句。 在发布的测试代码中，两者：

// [1]
.Callback(() => Thread.Sleep(4000))//Delayed here

和：

// [2]
.Returns(() => Task.FromResult(new HttpResponseMessage
{
StatusCode = HttpStatusCode.InternalServerError,
Content = new StringContent("Some response body", Encoding.UTF8, " text/xml")
})

不包含await语句。因此，[1] 和 [2] 将在调用线程上完全同步运行(整整四秒)，直到它们返回Task.FromResult返回的已完成Task。 它们在任何时候都不会返回 TimeoutPolicy 可以管理的"热"(正在运行)Task- 在同步四秒执行完成之前，调用策略不会重新获得控制权。并且 [1] 和 [2] 不响应任何CancellationToken.因此，调用超时策略永远不会超时。

Afaik 目前在 Moq 中没有 CallbackAsync(...)，因此您需要将Thread.Sleep移动到Returns(() => )lambda 表达式中，作为可取消的await Task.Delay(...)，以模拟HttpClient.SendAsync(...)的行为方式。 像这样：

httpMessageHandler.Protected()
.Setup<Task<HttpResponseMessage>>("SendAsync", 
ItExpr.IsAny<HttpRequestMessage>(), ItExpr.IsAny<CancellationToken>())
.Returns<object, CancellationToken>(async (request, cancellationToken) => {
await Task.Delay(4000, cancellationToken); // [3]
return new HttpResponseMessage
{
StatusCode = HttpStatusCode.InternalServerError,
Content = new StringContent("Some response body", Encoding.UTF8, "text/xml")
};
});

在此代码中，TimeoutPolicy 在 [3] 处的await处重新获得控制权，并且cancellationToken可以控制取消正在运行的Task。