SpringCloud/spring-cloud-gateway
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Formatting

Browse Source
This commit is contained in:
spencergibb
2021-03-12 15:37:06 -05:00
parent 779187e6ae
commit 70c50744fb
2 changed files with 25 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
spring-cloud-gateway-server/src/main/java/org/springframework/cloud/gateway/filter/factory/AddRequestHeaderGatewayFilterFactory.java
View File

@@ -40,7 +40,8 @@ public class AddRequestHeaderGatewayFilterFactory
GatewayFilterChain chain) {
String value = ServerWebExchangeUtils.expand(exchange, config.getValue());
ServerHttpRequest request = exchange.getRequest().mutate()
.headers(httpHeaders -> httpHeaders.add(config.getName(), value)).build();
.headers(httpHeaders -> httpHeaders.add(config.getName(), value))
.build();
return chain.filter(exchange.mutate().request(request).build());
}

40
spring-cloud-gateway-server/src/test/java/org/springframework/cloud/gateway/filter/factory/AddRequestHeaderGatewayFilterFactoryTests.java
View File

@@ -63,11 +63,13 @@ public class AddRequestHeaderGatewayFilterFactoryTests extends BaseWebClientTest
@Test
public void addRequestHeaderFilterWorksMultipleValues() {
testClient.get().uri("/multivalueheaders").header("Host", "www.addrequestheader.org").exchange()
.expectBody(Map.class)
.consumeWith(result -> {
Map<String, Object> headers = getMap(result.getResponseBody(), "headers");
assertThat(headers).containsEntry("X-Request-Example", Arrays.asList("ValueA", "ValueB"));
testClient.get().uri("/multivalueheaders")
.header("Host", "www.addrequestheader.org").exchange()
.expectBody(Map.class).consumeWith(result -> {
Map<String, Object> headers = getMap(result.getResponseBody(),
"headers");
assertThat(headers).containsEntry("X-Request-Example",
Arrays.asList("ValueA", "ValueB"));
});
}
@@ -83,11 +85,13 @@ public class AddRequestHeaderGatewayFilterFactoryTests extends BaseWebClientTest
@Test
public void addRequestHeaderFilterMultipleValuesWorksJavaDsl() {
testClient.get().uri("/multivalueheaders").header("Host", "www.addrequestheaderjava.org").exchange()
.expectBody(Map.class)
.consumeWith(result -> {
Map<String, Object> headers = getMap(result.getResponseBody(), "headers");
assertThat(headers).containsEntry("X-Request-Acme", Arrays.asList("ValueB-www", "ValueC-www"));
testClient.get().uri("/multivalueheaders")
.header("Host", "www.addrequestheaderjava.org").exchange()
.expectBody(Map.class).consumeWith(result -> {
Map<String, Object> headers = getMap(result.getResponseBody(),
"headers");
assertThat(headers).containsEntry("X-Request-Acme",
Arrays.asList("ValueB-www", "ValueC-www"));
});
}
@@ -111,14 +115,16 @@ public class AddRequestHeaderGatewayFilterFactoryTests extends BaseWebClientTest
public RouteLocator testRouteLocator(RouteLocatorBuilder builder) {
return builder.routes().route("add_request_header_java_test",
r -> r.path("/headers").and().host("{sub}.addrequestheaderjava.org")
.filters(f -> f.prefixPath("/httpbin").addRequestHeader("X-Request-Acme", "ValueB-{sub}"))
.filters(f -> f.prefixPath("/httpbin")
.addRequestHeader("X-Request-Acme", "ValueB-{sub}"))
.uri(uri))
.route("add_multiple_request_header_java_test", r -> r
.path("/multivalueheaders").and()
.host("{sub}.addrequestheaderjava.org")
.filters(f -> f.prefixPath("/httpbin")
.addRequestHeader("X-Request-Acme", "ValueB-{sub}")
.addRequestHeader("X-Request-Acme", "ValueC-{sub}"))
.uri(uri))
.route("add_multiple_request_header_java_test",
r -> r.path("/multivalueheaders").and().host("{sub}.addrequestheaderjava.org")
.filters(f -> f.prefixPath("/httpbin")
.addRequestHeader("X-Request-Acme", "ValueB-{sub}")
.addRequestHeader("X-Request-Acme", "ValueC-{sub}"))
.uri(uri))
.build();
}