From 2073a63c85bbf9ed6bbe282d67f0db292aa5eb21 Mon Sep 17 00:00:00 2001
From: Gerardo Roza <rozagerardo@gmail.com>
Date: Tue, 26 Jan 2021 17:41:44 -0300
Subject: [PATCH] added more common exchangeToMono example for article

---
 .../web/client/WebClientIntegrationTest.java       | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 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
index c8a306c4bc..9116ff3e63 100644
--- 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 class WebClientIntegrationTest {
             .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>>() {