WIP: WebClient -- Syntax angeglichen

[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 2c94a49..27795f9 100644 (file)
--- a/src/test/java/de/juplo/facebook/errors/GraphApiExchangeFilterFunctionIntegrationTest.java
+++ b/src/test/java/de/juplo/facebook/errors/GraphApiExchangeFilterFunctionIntegrationTest.java
@@ -13,6 +13,9 @@ import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.client.reactive.JettyClientHttpConnector;
+import org.springframework.web.reactive.function.client.ClientRequest;
+import org.springframework.web.reactive.function.client.ClientResponse;
+import org.springframework.web.reactive.function.client.ExchangeFunction;
 import org.springframework.web.reactive.function.client.WebClient;
 import reactor.core.publisher.Mono;
 import reactor.test.StepVerifier;
@@ -40,7 +43,7 @@ public class GraphApiExchangeFilterFunctionIntegrationTest
             .builder()
             .clientConnector(new JettyClientHttpConnector())
             .baseUrl(server.url("/").toString())
-            .filter(GraphApiExchangeFilterFunction.INSTANCE)
+            .filter(this::errorHandlingFilter)
             .build();
        }
 
@@ -89,4 +92,59 @@ public class GraphApiExchangeFilterFunctionIntegrationTest
         })
         .verify(Duration.ofSeconds(3));
   }
+
+  @Test
+  public void testTest()
+  {
+    server
+        .enqueue(new MockResponse()
+            .setResponseCode(400)
+            .setHeader("Content-Type", "text/plain")
+            .setBody("Hello Spring!"));
+
+               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 Mono<ClientResponse> errorHandlingFilter(ClientRequest request, ExchangeFunction next)
+  {
+    return
+        next
+            .exchange(request)
+            .flatMap(response ->
+            {
+              return
+                  HttpStatus.Series.CLIENT_ERROR.equals(response.statusCode().series())
+                      ? response.bodyToMono(String.class).flatMap(errorBody ->
+                        {
+                          return Mono.error(GraphApiException.create(response.statusCode(), response.headers().asHttpHeaders(), errorBody.getBytes()));
+                        })
+                      : Mono.just(response);
+            });
+  }
 }