17

While testing a RestClient-Implementation I want to simulate a RestClientException that may be thrown by some RestTemplate-methods in that implementation f.e. the delete-method:

@Override
public ResponseEntity<MyResponseModel> documentDelete(String id) {
    template.setErrorHandler(new MyResponseErrorHandler());
    ResponseEntity<MyResponseModel> response = null;
    try {
        String url = baseUrl + "/document/id/{id}";
        response = template.exchange(url, DELETE, null, MyResponseModel.class, id);
    } catch (RestClientException ex) {
        return handleException(ex);
    }
    return response;
}

How can I achieve this?

I define the mock-server in this way:

@Before
public void setUp() {
    mockServer = MockRestServiceServer.createServer(template);
    client = new MyRestClient(template, serverUrl + ":" + serverPort);
}
JMarky
  • 289
  • 2
  • 3
  • 8

4 Answers4

17

You can test throwing runtime exceptions from the MockRestServiceServer, although this class, as of Spring 5.0.0.RC4, is not designed for it (which means it may not work for more complex use cases):

RestTemplate yourApi;
MockRestServiceServer server = MockRestServiceServer.createServer(yourApi);

server.expect(requestTo("http://..."))
    .andRespond((response) -> { throw new ResourceAccessException(
        new ConnectException("Connection reset")); });

It seems to work in tests:

  • where there's only one RestTemplate call,
  • where the exception is thrown as a result of the last expectation.

I wasn't able to expect two consecutive exceptions; the MockRestSeriviceServer (more concrete, the SimpleRequestExpectationManager) throws IllegalStateException on replaying the second expectation.

MaDa
  • 10,511
  • 9
  • 46
  • 84
12

You can take advantage of the MockRestResponseCreators for mocking 4xx or 5xx responses from the mockRestServiceServer.

For example for testing a 5xx - Internal server error:

mockServer.expect(requestTo("your.url"))
                .andExpect(method(HttpMethod.GET/POST....))
                .andRespond(withServerError()...);

In your case the RestClientException is thrown for client-side HTTP errors, so the example above can be fine tuned for a 4xx exception by using: ...andRespond(withBadRequest()); or ...andRespond(withStatus(HttpStatus.NOT_FOUND));

For a more simpler usage of these methods you use static imports for org.springframework.test.web.client.MockRestServiceServer,org.springframework.test.web.client.response.MockRestResponseCreators

Alex Ciocan
  • 2,272
  • 14
  • 20
  • hm ... ok I could use the builder-methods to create finally a ClientHttpResponse but how can I _fine-tune_ such a response via the builder-methods so that an exception is thrown/simulated? – JMarky Mar 03 '17 at 12:23
  • 1
    Adding andRespond(withStatus(HttpStatus.NOT_FOUND)); ahiuld help you – Alex Ciocan Mar 03 '17 at 12:25
  • 1
    Adding _andRespond(withStatus(HttpStatus.NOT_FOUND))_ delivers a ClientHTTPResponse with the defined http-status-code. that means, a call like _response = template.exchange(url, DELETE, null, MyResponseModel.class, id);_ brings back a response and not a _RestClientException_ – JMarky Mar 03 '17 at 12:35
  • The restTemplate is bound to your mockServer. By looking at the source of the restTemplate you can see that when an 4xx code is encountered in the response a specific RestClientException is thrown. So mocking an 4xx response from the server should provide get you in the catch block. – Alex Ciocan Mar 03 '17 at 14:27
  • Hi Alex, I think you're right. I forgot, that I set a custom errorHandler and it catched me this exception away, when the status-code is 4xx or 5xx. So a RestClientException at this place can't be occur. :-) Thanks for your advices! – JMarky Mar 03 '17 at 16:59
7

Answer by Alex Ciocan works for different http status responses, so if you want those, go with that as that's the cleanest way to go. I had a problem that I needed to be able to test also for connection reset and other network-level problems, which are trickier to simulate.

Answer by MaDa works for some use cases, but it didn't work for me when using AsyncRestTemplate, as it throws too early. However it did lead me to right direction. This one seems to work with async calls as well:

import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

// ...

ClientHttpResponse exceptionThrowingResponse = mock(ClientHttpResponse.class);

when(exceptionThrowingResponse.getStatusCode()) // getRawStatusCode() in latest spring
    .thenThrow(new IOException("connection reset"));

mockServer.expect(requestTo("http://localhost:123/callme"))
    .andRespond((response) -> exceptionThrowingResponse);

This seems to also work for consecutive exceptions, as well as different http statuses.

cassiomolin
  • 124,154
  • 35
  • 280
  • 359
eis
  • 51,991
  • 13
  • 150
  • 199
0

How about this :

@Spy
@InjectMocks
ClasstoMock objToMock;

@Test
public void testRestClientException() throws  Exception {       
        try {
        Mockito.when(this.objToMock.perform()).thenThrow(new RestClientException("Rest Client Exception"));
        this.objToMock.perform();
        }
        catch(Exception e){
            Assert.assertEquals(RestClientException.class, e.getClass());
        }  
Alferd Nobel
  • 3,185
  • 2
  • 30
  • 35