java - 如何模拟 REST 请求

Question

我在 JUnit 中使用 Mockito，并且有一个方法使用 RestTemplate 向微服务发出请求。

private static final String REQUESTOR_API_HOST = "http://localhost:8090/requestor/v1/requestors/";

public TokenRequestorPayload getTokenRequestor(Long id) {
    restClient = new RestTemplate();
    return restClient.getForObject(REQUESTOR_API_HOST + id, TokenRequestorPayload.class);
}

此方法返回一个 JSON 对象，该对象将在 TokenRequestorPayload 类中反序列化。

当我执行单元测试时，它们失败了，因为模拟不起作用，并且我得到了org.springframework.web.client.ResourceAccessException。我如何模拟我的 RestTemplate？

测试

RestTemplate restTemplate = Mockito.spy(RestTemplate.class);
Mockito.doReturn(this.tokenRequestorMockJson()).when(restTemplate).getForObject(Mockito.anyString(), Mockito.eq(TokenRequestorPayload.class));

错误

org.springframework.web.client.ResourceAccessException: I/O error on GET request for "http://localhost:8090/requestor/v1/requestors/1": Connection refused (Connection refused); nested exception is java.net.ConnectException: Connection refused (Connection refused)

Answer 1

在您的测试中，您正在为 RestTemplate 的模拟实例定义行为，这很好。

RestTemplate restTemplate = Mockito.spy(RestTemplate.class);

但是，被测试的类不会使用同一个实例，它每次都会创建一个新的 RestTemplate 实例并使用该实例。

public TokenRequestorPayload getTokenRequestor(Long id, RestTemplate restClient) {
    restClient = new RestTemplate(); // it uses this instance, not the mock
    return restClient.getForObject(REQUESTOR_API_HOST + id, TokenRequestorPayload.class);
}

因此您需要找到一种方法将模拟实例引入 getTokenRequestor() 方法。

例如这可以通过将 restClient 转换为方法参数来完成:

public TokenRequestorPayload getTokenRequestor(Long id, RestTemplate restClient) {
    return restClient.getForObject(REQUESTOR_API_HOST + id, TokenRequestorPayload.class);
}

测试可能看起来像这样:

@Test
public void test() {
    RestTemplate restTemplateMock = Mockito.spy(RestTemplate.class);
    Mockito.doReturn(null).when(restTemplateMock).getForObject(Mockito.anyString(), Mockito.eq(TokenRequestorPayload.class));

    // more code

    instance.getTokenRequestor(id, restTemplateMock); // passing in the mock
}