之前写了一篇介绍 HttpClient 的两种重试机制,但是否真的会依照预期进行重试咱们不得而知。
他人提供给咱们的 API 往往都是失常的,很多谬误并不能稳固重现,这也造成了咱们无奈进行全面的测试。正是这种状况下,理解到了 WireMock。
本文不打算做一个入门教程,重点在于如何用 WireMock 解决理论的问题。
初始化调用端
首先初始化一个 HttpClient
public ZwroksApi(){RequestConfig requestConfig = RequestConfig.custom().setSocketTimeout(1000).build();
DefaultServiceUnavailableRetryStrategy serviceUnavailableRetryStrategy = new DefaultServiceUnavailableRetryStrategy();
httpClient = HttpClients.custom()
.setServiceUnavailableRetryStrategy(serviceUnavailableRetryStrategy)
.setDefaultRequestConfig(requestConfig)
.build();}
进行了如下配置
- 异样重试默认开启,重试 3 次。
- 应用了默认的服务不可用重试策略,会重试 1 次,距离 1 秒。
- 设置 SocketTimeout 为 1 秒,用于模仿超时。
异样重试
超时
首先来模仿一下超时,设置了固定的 delay 2 秒,所以肯定会超时,依照预期会重试 3 次,所以一共是申请 4 次。
@Test
public void testSocketTimeOut() {stubFor(get(urlEqualTo("/test"))
.willReturn(aResponse()
.withStatus(HttpStatus.SC_OK)
.withHeader("Content-Type", "text/xml")
.withBody("ok")
.withFixedDelay(2000)));
zwroksApi.test();
verify(4, getRequestedFor(urlEqualTo("/test")));
}
然而后果呢
Expected exactly 4 requests matching the following pattern but received 1
具体的起因在之前那篇文章也曾经提过,SocketTimeoutException
是 InterruptedIOException
的子类,不会重试。
如何超时也进行重试,在上一篇中也有讲,这里就不赘述了。
谬误返回
WireMock 中提供了几种谬误返回,应用也很简略
@Test
public void testBadResponse() {stubFor(get(urlEqualTo("/test"))
.willReturn(aResponse().withFault(Fault.CONNECTION_RESET_BY_PEER)));
zwroksApi.test();
verify(4, getRequestedFor(urlEqualTo("/test")));
}
这里在运行时会抛错 java.net.SocketException: Connection reset
,这种异样是会进行重试的,所以这里测试能够通过。
服务不可用重试
调用的代码如下,我心愿在返回 503 的时候进行重试
@Test
public void testServiceUnavailable() {stubFor(get(urlEqualTo("/test"))
.willReturn(aResponse()
.withStatus(HttpStatus.SC_SERVICE_UNAVAILABLE)
.withHeader("Content-Type", "text/xml")
.withBody("service unavailable")));
zwroksApi.test();
verify(2, getRequestedFor(urlEqualTo("/test")));
}
这里测试也是能够通过的。WireMock 提供了获取申请日志的能力,除了次数,咱们能够看看是否两次申请是距离了一秒。
List<ServeEvent> allServeEvents = getAllServeEvents();
long firstTime = allServeEvents.get(1).getRequest().getLoggedDate().getTime();
long lastTime = allServeEvents.get(0).getRequest().getLoggedDate().getTime();
assert lastTime - firstTime > 1000;
这里只是介绍了一小部分 WireMock 的性能,更多的性能能够去官网看文档,十分好懂。