import org.slf4j.LoggerFactory;
import org.springframework.http.HttpStatus;
import org.springframework.http.client.reactive.JettyClientHttpConnector;
+import org.springframework.web.reactive.function.client.ExchangeFilterFunction;
import org.springframework.web.reactive.function.client.WebClient;
import reactor.core.publisher.Mono;
import reactor.test.StepVerifier;
.builder()
.clientConnector(new JettyClientHttpConnector())
.baseUrl(server.url("/").toString())
- .filter(GraphApiExchangeFilterFunction.INSTANCE)
.build();
}
.get()
.uri("/egal")
.retrieve()
+ .onStatus(status -> status.is4xxClientError(), response -> GraphApiException.create(response))
.bodyToMono(String.class);
StepVerifier
assertEquals("(#613) Calls to stream have exceeded the rate of 600 calls per 600 seconds.", e.getMessage());
assertEquals(Type.OAuthException, e.getType());
})
- .verifyLater();
+ .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()
+ .onStatus(status -> status.is4xxClientError(), response -> GraphApiException.create(response))
+ .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()
+ .onStatus(status -> status.is4xxClientError(), response -> GraphApiException.create(response))
+ .bodyToMono(String.class);
+
+ StepVerifier.create(result)
+ .expectNext("Hello Spring!")
+ .expectComplete()
+ .verify(Duration.ofSeconds(3));
+ }
+
+ public static ExchangeFilterFunction errorHandlingFilter()
+ {
+ return ExchangeFilterFunction.ofResponseProcessor(clientResponse ->
+ {
+ if (clientResponse.statusCode() != null && (clientResponse.statusCode()
+ .is5xxServerError() || clientResponse.statusCode().is4xxClientError()))
+ {
+ return clientResponse.bodyToMono(String.class)
+ .flatMap(errorBody ->
+ {
+ return Mono.error(new Exception(errorBody));
+ });
+ }
+ else
+ {
+ return Mono.just(clientResponse);
+ }
+ });
}
}