added more common exchangeToMono example for article

zdougherty-softwarfare · Jan 26, 2021 · 2073a63 · 2073a63
1 parent 23f8b0a
commit 2073a63
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/spring-5-reactive/src/test/java/com/baeldung/web/client/WebClientIntegrationTest.java b/spring-5-reactive/src/test/java/com/baeldung/web/client/WebClientIntegrationTest.java
@@ -115,8 +115,18 @@ public void givenDifferentScenarios_whenRequestsSent_thenObtainExpectedResponses
             .bodyToMono(String.class);
         Mono<String> responsePostWithBody2 = headerSpecPost2.retrieve()
             .bodyToMono(String.class);
-        Mono<String> responsePostWithBody3 = headerSpecPost3.retrieve()
-            .bodyToMono(String.class);
+        Mono<String> responsePostWithBody3 = headerSpecPost3.exchangeToMono(response -> {
+            if (response.statusCode()
+                .equals(HttpStatus.OK)) {
+                return response.bodyToMono(String.class);
+            } else if (response.statusCode()
+                .is4xxClientError()) {
+                return Mono.just("Error response");
+            } else {
+                return response.createException()
+                    .flatMap(Mono::error);
+            }
+        });
         Mono<String> responsePostFoo = headerSpecFooPost.retrieve()
             .bodyToMono(String.class);
         ParameterizedTypeReference<Map<String, String>> ref = new ParameterizedTypeReference<Map<String, String>>() {