关于测试:API-测试利器-WireMock

之前写了一篇介绍 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();}

进行了如下配置

1. 异样重试默认开启，重试 3 次。
2. 应用了默认的服务不可用重试策略，会重试 1 次，距离 1 秒。
3. 设置 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 的性能，更多的性能能够去官网看文档，十分好懂。