Spring RestTemplate wywołując webservice z błędami i analizując kod stanu

Question

Zaprojektowałem webservice do wykonania zadania, jeśli parametry żądania są OK, lub zwracają 401 Nieautoryzowany kod statusu HTTP, jeśli parametry żądania są błędne lub puste.

Używam RestTemplate do wykonania testu i jestem w stanie zweryfikować status HTTP 200 OK, jeśli usługa odpowie poprawnie. Nie mogę jednak przetestować błędu HTTP 401, ponieważ sam RestTemplate zgłasza wyjątek.

Moja metoda badawcza jest

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

Wyjątek dziennika jest

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) 

Jak mogę sprawdzić, czy usługa odpowiada z kodem stanu HTTP 401?

Answer 1

Możesz użyć testu sprężynowego. Jest to o wiele łatwiejsze:

@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

Trzeba wdrożyć ResponseErrorHandler w celu przechwycenia kodu odpowiedzi, ciało i nagłówek kiedy masz non-2xx kody odpowiedzi z usługi za pomocą szablonu odpoczynek. Skopiuj wszystkie potrzebne informacje, dołącz je do niestandardowego wyjątku i wyrzuć, aby można było go złapać w teście.

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

Teraz co trzeba zrobić w swoim badaniu jest, ustaw tę ResponseErrorHandler w RestTemplate jak,

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

Jako alternatywę dla rozwiązania przedstawionego przez Nilesh, można również użyć sprężyny klasy DefaultResponseErrorHandler. Trzeba również ovveride jego metodę hasError (HttpStatus), aby nie wyrzucała wyjątku w przypadku nieudanych wyników.

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

Answer 4

W mojej służbie spoczynku złapię HttpStatusCodeException zamiast Exception od HttpStatusCodeException ma metodę uzyskania kodu stanu

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

Answer 5

Od wiosny 4.3 Jest RestClientResponseException który zawiera aktualne dane odpowiedzi HTTP, takie jak kod statusu, treść odpowiedzi i nagłówki. I możesz to złapać.

RestClientResponseException Java Doc