* Added code for BAEL-1972 (#4632)

2018-07-08 05:40:02 -03:00
parent e78ca7c830
commit 3786d4e1b1
10 changed files with 411 additions and 0 deletions
@@ -0,0 +1,35 @@
+package com.baeldung.reactive.responseheaders.controllers;
+
+import org.springframework.http.ResponseEntity;
+import org.springframework.http.server.reactive.ServerHttpResponse;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+import reactor.core.publisher.Mono;
+
+@RestController
+@RequestMapping("/response-header")
+public class ResponseHeaderController {
+
+    @GetMapping("/response-entity")
+    public Mono<ResponseEntity<String>> usingResponseEntityBuilder() {
+        String responseHeaderKey = "Baeldung-Example-Header";
+        String responseHeaderValue = "Value-ResponseEntityBuilder";
+        String responseBody = "Response with header using ResponseEntity (builder)";
+
+        return Mono.just(ResponseEntity.ok()
+            .header(responseHeaderKey, responseHeaderValue)
+            .body(responseBody));
+    }
+    
+    @GetMapping("/server-http-response")
+    public Mono<String> usingServerHttpResponse(ServerHttpResponse response) {
+        String responseHeaderKey = "Baeldung-Example-Header";
+        String responseHeaderValue = "Value-ServerHttpResponse";
+        String responseBody = "Response with header using ServerHttpResponse";
+        
+        response.getHeaders().add(responseHeaderKey, responseHeaderValue);
+        return Mono.just(responseBody);
+    }
+}
@@ -0,0 +1,20 @@
+package com.baeldung.reactive.responseheaders.filter;
+
+import org.springframework.stereotype.Component;
+import org.springframework.web.server.ServerWebExchange;
+import org.springframework.web.server.WebFilter;
+import org.springframework.web.server.WebFilterChain;
+
+import reactor.core.publisher.Mono;
+
+@Component
+public class AddResponseHeaderWebFilter implements WebFilter {
+
+    @Override
+    public Mono<Void> filter(ServerWebExchange exchange, WebFilterChain chain) {
+        exchange.getResponse()
+            .getHeaders()
+            .add("Baeldung-Example-Filter-Header", "Value-Filter");
+        return chain.filter(exchange);
+    }
+}
@@ -0,0 +1,21 @@
+package com.baeldung.reactive.responseheaders.functional.handlers;
+
+import org.springframework.stereotype.Component;
+import org.springframework.web.reactive.function.server.ServerRequest;
+import org.springframework.web.reactive.function.server.ServerResponse;
+
+import reactor.core.publisher.Mono;
+
+@Component
+public class ResponseHeaderHandler {
+    
+    public Mono<ServerResponse> useHandler(final ServerRequest request) {
+        String responseHeaderKey = "Baeldung-Example-Header";
+        String responseHeaderValue = "Value-Handler";
+        String responseBody = "Response with header using Handler";
+
+        return ServerResponse.ok()
+            .header(responseHeaderKey, responseHeaderValue)
+            .body(Mono.just(responseBody),String.class);
+    }
+}
@@ -0,0 +1,20 @@
+package com.baeldung.reactive.responseheaders.functional.routers;
+
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.web.reactive.function.server.RequestPredicates;
+import org.springframework.web.reactive.function.server.RouterFunction;
+import org.springframework.web.reactive.function.server.RouterFunctions;
+import org.springframework.web.reactive.function.server.ServerResponse;
+
+import com.baeldung.reactive.responseheaders.functional.handlers.ResponseHeaderHandler;
+
+@Configuration
+public class ResponseHeadersRouterFunctions {
+
+    @Bean
+    public RouterFunction<ServerResponse> responseHeaderRoute(@Autowired ResponseHeaderHandler handler) {
+        return RouterFunctions.route(RequestPredicates.GET("/functional-response-header/single-handler"), handler::useHandler);
+    }
+}
@@ -0,0 +1,63 @@
+package com.baeldung.reactive.responseheaders;
+
+import org.junit.jupiter.api.BeforeAll;
+import org.junit.jupiter.api.Test;
+import org.junit.runner.RunWith;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.context.junit4.SpringRunner;
+import org.springframework.test.web.reactive.server.WebTestClient;
+import org.springframework.test.web.reactive.server.WebTestClient.ResponseSpec;
+
+@RunWith(SpringRunner.class)
+@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
+public class ResponseHeaderLiveTest {
+
+    private static final String BASE_URL = "http://localhost:8080";
+    private static final String ANNOTATION_BASE_URL = BASE_URL + "/response-header";
+    private static final String FUNCTIONAL_BASE_URL = BASE_URL + "/functional-response-header";
+    private static final String SERVICE_SINGLE_RESPONSE_HEADER = "Baeldung-Example-Header";
+    private static final String SERVICE_FILTER_RESPONSE_HEADER = "Baeldung-Example-Filter-Header";
+    private static final String SERVICE_FILTER_RESPONSE_HEADER_VALUE = "Value-Filter";
+
+    private static WebTestClient client;
+
+    @BeforeAll
+    public static void setup() {
+        client = WebTestClient.bindToServer()
+            .baseUrl(BASE_URL)
+            .build();
+    }
+
+    @Test
+    public void whenUsingResponseEntityBuilderRequest_thenObtainResponseWithCorrectHeaders() {
+        client = WebTestClient.bindToServer()
+            .baseUrl(BASE_URL)
+            .build();
+        ResponseSpec response = client.get()
+            .uri(ANNOTATION_BASE_URL + "/response-entity")
+            .exchange();
+
+        response.expectHeader().valueEquals(SERVICE_SINGLE_RESPONSE_HEADER, "Value-ResponseEntityBuilder")
+            .expectHeader().valueEquals(SERVICE_FILTER_RESPONSE_HEADER, SERVICE_FILTER_RESPONSE_HEADER_VALUE);
+    }
+
+    @Test
+    public void whenUsingServerHttpResponseRequest_thenObtainResponseWithCorrectHeaders() {
+        ResponseSpec response = client.get()
+            .uri(ANNOTATION_BASE_URL + "/server-http-response")
+            .exchange();
+
+        response.expectHeader().valueEquals(SERVICE_SINGLE_RESPONSE_HEADER, "Value-ServerHttpResponse")
+            .expectHeader().valueEquals(SERVICE_FILTER_RESPONSE_HEADER, SERVICE_FILTER_RESPONSE_HEADER_VALUE);
+    }
+
+    @Test
+    public void whenUsingFunctionalHandlerRequest_thenObtainResponseWithCorrectHeaders() {
+        ResponseSpec response = client.get()
+            .uri(FUNCTIONAL_BASE_URL + "/single-handler")
+            .exchange();
+
+        response.expectHeader().valueEquals(SERVICE_SINGLE_RESPONSE_HEADER, "Value-Handler")
+            .expectHeader().valueEquals(SERVICE_FILTER_RESPONSE_HEADER, SERVICE_FILTER_RESPONSE_HEADER_VALUE);
+    }
+}