X-Git-Url: https://juplo.de/gitweb/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fde%2Fjuplo%2Ffacebook%2Ferrors%2FGraphApiExchangeFilterFunctionIntegrationTest.java;h=b9636ce12bb920b7c25352dab0493112f6e4aaa9;hb=f16862b51e98a30710fe20dc867110d48bb03be3;hp=a9ba58c6c1a385dcc3b15c66b4c721eb11ac94a3;hpb=9a19a543217345145934e913b113c8e50f899ffd;p=facebook-errors

diff --git a/src/test/java/de/juplo/facebook/errors/GraphApiExchangeFilterFunctionIntegrationTest.java b/src/test/java/de/juplo/facebook/errors/GraphApiExchangeFilterFunctionIntegrationTest.java
index a9ba58c..b9636ce 100644
--- a/src/test/java/de/juplo/facebook/errors/GraphApiExchangeFilterFunctionIntegrationTest.java
+++ b/src/test/java/de/juplo/facebook/errors/GraphApiExchangeFilterFunctionIntegrationTest.java
@@ -13,9 +13,6 @@ 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;
@@ -43,7 +40,7 @@ public class GraphApiExchangeFilterFunctionIntegrationTest
             .builder()
             .clientConnector(new JettyClientHttpConnector())
             .baseUrl(server.url("/").toString())
-            .filter(this::errorHandlingFilter)
+            .filter(new GraphApiExchangeFilterFunction())
             .build();
 	}
 
@@ -130,24 +127,4 @@ public class GraphApiExchangeFilterFunctionIntegrationTest
         .expectComplete()
 				.verify(Duration.ofSeconds(3));
   }
-
-  public Mono<ClientResponse> errorHandlingFilter(ClientRequest request, ExchangeFunction next)
-  {
-    return next.exchange(request).flatMap(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);
-      }
-    });
-  }
 }