import de.juplo.facebook.errors.GraphApiException.Type;
import java.time.Duration;
-import java.util.List;
import okhttp3.mockwebserver.MockResponse;
import okhttp3.mockwebserver.MockWebServer;
import org.junit.After;
import org.junit.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
+import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.client.reactive.JettyClientHttpConnector;
import org.springframework.web.reactive.function.client.ExchangeFilterFunction;
.builder()
.clientConnector(new JettyClientHttpConnector())
.baseUrl(server.url("/").toString())
- .filter(GraphApiExchangeFilterFunction.INSTANCE)
+ .filter(errorHandlingFilter())
.build();
}
@Test
public void testTest()
{
- WebClient webClient =
- WebClient
- .builder()
- .clientConnector(new JettyClientHttpConnector())
- .baseUrl(server.url("/").toString())
- .filter(ExchangeFilterFunction.ofResponseProcessor(response ->
- {
- List<String> headerValues = response.headers().header("Foo");
- return headerValues.isEmpty() ? Mono.error(
- new Exception("Response does not contain Foo header")) :
- Mono.just(response);
- }))
- .build();
-
server
.enqueue(new MockResponse()
+ .setResponseCode(400)
.setHeader("Content-Type", "text/plain")
.setBody("Hello Spring!"));
- Mono<String> result = webClient.get()
+ Mono<String> result;
+
+ result = webClient.get()
.uri("/greeting?name=Spring")
.retrieve()
.bodyToMono(String.class);
StepVerifier.create(result)
.expectError(Exception.class)
.verify(Duration.ofSeconds(3));
+
+ server
+ .enqueue(new MockResponse()
+ .setResponseCode(200)
+ .setHeader("Content-Type", "text/plain")
+ .setHeader("Foo", "Bar")
+ .setBody("Hello Spring!"));
+
+ result = webClient.get()
+ .uri("/greeting?name=Spring")
+ .retrieve()
+ .bodyToMono(String.class);
+
+ StepVerifier.create(result)
+ .expectNext("Hello Spring!")
+ .expectComplete()
+ .verify(Duration.ofSeconds(3));
+ }
+
+ public static ExchangeFilterFunction errorHandlingFilter()
+ {
+ return ExchangeFilterFunction.ofResponseProcessor(response ->
+ {
+ if (response.statusCode() != null && (response.statusCode()
+ .is5xxServerError() || response.statusCode().is4xxClientError()))
+ {
+ return response.bodyToMono(String.class)
+ .flatMap(errorBody ->
+ {
+ return Mono.error(GraphApiException.create(response.statusCode(), response.headers().asHttpHeaders(), errorBody.getBytes()));
+ });
+ }
+ else
+ {
+ return Mono.just(response);
+ }
+ });
}
}