Spring RestTemplate調用帶有錯誤的webservice並分析狀態代碼

Question

我設計了一個web服務來在請求參數正常的情況下執行任務，或者返回401未授權的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回覆我可以測試？

Answer 1

您可以使用彈簧測試。這是很容易：

@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())); 
     } 
} 

Answer 2

你需要爲了攔截響應碼，身體和頭部，當你使用模板休息從服務中獲得非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]")); 
} 

Answer 3

作爲替代由Nilesh製作提出的解決方案，你也可以使用Spring類DefaultResponseErrorHandler。您還需要對其hasError（HttpStatus）方法進行拓展，以便它不會在不成功的結果上拋出異常。

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

Answer 4

在我休息的服務，我趕上HttpStatusCodeException，而不是Exception 因爲HttpStatusCodeException有越來越狀態碼

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

Answer 5

從Spring 4.3的方法，有包含實際的HTTP響應數據RestClientResponseException，如狀態碼，響應正文和標題。你可以抓住它。

RestClientResponseException Java Doc