WIP: WebClient - DIE LÖSUNG

[facebook-errors] / src / test / java / de / juplo / facebook / errors / GraphApiExchangeFilterFunctionIntegrationTest.java

diff --git a/src/test/java/de/juplo/facebook/errors/GraphApiExchangeFilterFunctionIntegrationTest.java b/src/test/java/de/juplo/facebook/errors/GraphApiExchangeFilterFunctionIntegrationTest.java
index 743f6be..8d21cc5 100644 (file)
--- a/src/test/java/de/juplo/facebook/errors/GraphApiExchangeFilterFunctionIntegrationTest.java
+++ b/src/test/java/de/juplo/facebook/errors/GraphApiExchangeFilterFunctionIntegrationTest.java
@@ -3,7 +3,6 @@ package de.juplo.facebook.errors;
 
 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;
@@ -12,6 +11,7 @@ import org.junit.Before;
 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;
@@ -42,7 +42,7 @@ public class GraphApiExchangeFilterFunctionIntegrationTest
             .builder()
             .clientConnector(new JettyClientHttpConnector())
             .baseUrl(server.url("/").toString())
-            .filter(GraphApiExchangeFilterFunction.INSTANCE)
+            .filter(errorHandlingFilter())
             .build();
        }
 
@@ -95,26 +95,15 @@ public class GraphApiExchangeFilterFunctionIntegrationTest
   @Test
   public void testTest()
   {
-    WebClient webClient =
-        WebClient
-            .builder()
-            .clientConnector(new JettyClientHttpConnector())
-            .baseUrl(server.url("/").toString())
-            .filter((request, next) -> next.exchange(request).flatMap(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);
@@ -122,5 +111,42 @@ public class GraphApiExchangeFilterFunctionIntegrationTest
                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);
+      }
+    });
   }
 }