Spring RestTem板调用有错误的webservice并分析状态代码

Question

我设计了一个Web服务，在请求参数正常时执行任务，或者在请求参数错误或为空时返回401个未经授权的HTTP状态码。

我使用RestTem板来执行测试，如果webservice成功回复，我可以验证HTTP 200 OK状态。然而，我无法测试HTTP 401错误，因为RestTem板本身会抛出异常。

我的测试方法是

@Test
public void testUnauthorized()
{
    Map<String, Object> params = new HashMap<String, Object>();
    ResponseEntity response = restTemplate.postForEntity(url, params, Map.class);
    Assert.assertEquals(HttpStatus.UNAUTHORIZED, response.getStatusCode());
    Assert.assertNotNull(response.getBody());
}

异常日志是

org.springframework.web.client.HttpClientErrorException: 401 Unauthorized
at org.springframework.web.client.DefaultResponseErrorHandler.handleError(DefaultResponseErrorHandler.java:88)
at org.springframework.web.client.RestTemplate.handleResponseError(RestTemplate.java:533)
at org.springframework.web.client.RestTemplate.doExecute(RestTemplate.java:489)
at org.springframework.web.client.RestTemplate.execute(RestTemplate.java:447)
at org.springframework.web.client.RestTemplate.postForEntity(RestTemplate.java:318)

如何测试webservice是否使用HTTP状态代码401进行回复？

Answer 1

在rest服务中，我捕获HttpStatusCodeException而不是Exception，因为HttpStatusCodeException有一个获取状态代码的方法

catch(HttpStatusCodeException e) {
    log.debug("Status Code", e.getStatusCode());
}

Answer 2

作为nilesh提供的解决方案的替代方案，您还可以使用spring类DefaultResponseErrorHandler。您还需要验证它的hasError（HttpStatus）方法，这样它就不会对不成功的结果抛出异常。

restTemplate.setErrorHandler(new DefaultResponseErrorHandler(){
    protected boolean hasError(HttpStatus statusCode) {
        return false;
    }});

Answer 3

您需要实现ResponseErrorHandler，以便在使用rest模板从服务获取非2xx响应代码时拦截响应代码、正文和标题。复制所有需要的信息，将其附加到自定义异常并抛出，以便在测试中捕获它。

public class CustomResponseErrorHandler implements ResponseErrorHandler {

    private ResponseErrorHandler errorHandler = new DefaultResponseErrorHandler();

    public boolean hasError(ClientHttpResponse response) throws IOException {
        return errorHandler.hasError(response);
    }

    public void handleError(ClientHttpResponse response) throws IOException {
        String theString = IOUtils.toString(response.getBody());
        CustomException exception = new CustomException();
        Map<String, Object> properties = new HashMap<String, Object>();
        properties.put("code", response.getStatusCode().toString());
        properties.put("body", theString);
        properties.put("header", response.getHeaders());
        exception.setProperties(properties);
        throw exception;
    }
}

现在你需要在测试中做的是，在RestTem板中设置这个响应错误处理程序，

RestTemplate restclient = new RestTemplate();
restclient.setErrorHandler(new CustomResponseErrorHandler());
try {
    POJO pojo = restclient.getForObject(url, POJO.class); 
} catch (CustomException e) {
    Assert.isTrue(e.getProperties().get("body")
                    .equals("bad response"));
    Assert.isTrue(e.getProperties().get("code").equals("400"));
    Assert.isTrue(((HttpHeaders) e.getProperties().get("header"))
                    .get("fancyheader").toString().equals("[nilesh]"));
}