31

我设计了一个 Web 服务，如果请求参数正常，则执行任务，如果请求参数错误或为空，则返回 401 Unauthorized HTTP 状态代码。

我正在使用RestTemplate执行测试，如果 web 服务回复成功，我可以验证 HTTP 200 OK 状态。但是，我无法测试 HTTP 401 错误，因为RestTemplate它本身会引发异常。

我的测试方法是

@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)

如何测试 Web 服务是否回复 HTTP 状态代码 401？

4

5 回答 5

50

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

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;
    }
}

现在你需要在你的测试中做的是，在 RestTemplate 中设置这个 ResponseErrorHandler 像，

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]"));
}
于 2013-03-28T15:33:58.090 回答
31

作为 nilesh 提出的解决方案的替代方案，您还可以使用 spring 类 DefaultResponseErrorHandler。您还需要覆盖它的 hasError(HttpStatus) 方法，这样它就不会在不成功的结果上抛出异常。

restTemplate.setErrorHandler(new DefaultResponseErrorHandler(){
    protected boolean hasError(HttpStatus statusCode) {
        return false;
    }});
于 2014-11-18T09:23:54.097 回答
7

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

catch(HttpStatusCodeException e) {
    log.debug("Status Code", e.getStatusCode());
}
于 2016-01-22T08:24:20.197 回答
5

您可以使用弹簧测试。这要容易得多：

@WebAppConfiguration
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration("classpath:your-context.xml")
public class BasicControllerTest {

        @Autowired
        protected WebApplicationContext wac;
        protected MockMvc mockMvc;

        @Before
        public void setUp() throws Exception {
        mockMvc = MockMvcBuilders.webAppContextSetup(this.wac).build();
        }

        @Test
        public void testUnauthorized(){

        mockMvc.perform(MockMvcRequestBuilders
                            .post("your_url")
                            .param("name", "values")
        .andDo(MockMvcResultHandlers.print())
        .andExpect(MockMvcResultMatchers.status().isUnauthorized()
        .andExpect(MockMvcResultMatchers.content().string(Matchers.notNullValue()));
        }
}
于 2013-03-14T09:55:15.790 回答
4

从 Spring 4.3 开始，有一个RestClientResponseException包含实际的 HTTP 响应数据，例如状态码、响应正文和标头。你可以抓住它。

RestClientResponseException Java 文档

于 2016-09-05T06:47:06.677 回答