Probando onErrorResume() Spring Webflux

Answer

1. 

   davy.ai

   June 1, 2023 at 9:11 am

   Para probar el escenario onErrorResume() usando StepVerifier, se puede generar la condición de error simulando que la llamada al webClient produzca una excepción. Aquí se muestra un ejemplo de caso de prueba:

       @Test
       public void testGetInfoError() {
           when(webClient.get()).thenReturn(requestHeadersUriSpec);
   
           // Throw a custom exception to test error scenario
           RuntimeException customError = new RuntimeException("API error");
           when(requestHeadersUriSpec.retrieve()).thenThrow(customError);
   
           service.getInfo(1)
               .as(StepVerifier::create)
               .expectError(WebFaultException.class)
               .verify();
       }
   

   Este caso de prueba primero simula la llamada get() al webClient para que devuelva un requestHeadersUriSpec. Después se configura la llamada retrieve() para simular un error lanzando una RuntimeException. Finalmente, se llama al método getInfo() y se utiliza un StepVerifier para afirmar que el método onErrorResume() captura el error correctamente y devuelve una WebFaultException.

   En cuanto al manejo de excepciones en el controlador, se puede usar switchIfEmpty() como un enfoque alternativo. switchIfEmpty() es útil cuando el Mono que se está utilizando puede devolver un resultado vacío y se quiere manejar ese escenario de manera diferente. Por ejemplo, se puede querer devolver un valor predeterminado o lanzar una excepción personalizada. Aquí se muestra un ejemplo de uso de switchIfEmpty():

       public Mono<sometype> getInfo(Integer id) {
           return webClient
                   .get()
                   .uri(uriBuilder -> uriBuilder.path())
                   .header("", "")
                   .header("", "")
                   .header("", "")
                   .header(HttpHeaders.ACCEPT, MediaType.APPLICATION_JSON_VALUE)
                   .retrieve()
                   .bodyToMono(POJO.class)
                   .onErrorResume(ex -> {
                       if (ex instanceof WebFaultException)
                           return Mono.error(ex);
                       return Mono.error(new WebFaultException(ex.getMessage(), "Error on API Call", HttpStatus.INTERNAL_SERVER_ERROR));
                   })
                   .switchIfEmpty(Mono.error(new WebFaultException("Empty result from API", "Empty Result", HttpStatus.NOT_FOUND)));
       }
   

   En este ejemplo, switchIfEmpty() se usa para manejar el caso en que la llamada bodyToMono() devuelve un Mono vacío. Si esto sucede, se devuelve una WebFaultException personalizada con un estado NOT_FOUND.

Comments are closed.