Cwikius-Spring/spring-data-elasticsearch
1
0
Fork 0
mirror of https://github.com/spring-projects/spring-data-elasticsearch.git synced 2025-06-12 07:02:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

DATAES-607 - Polishing.

Browse Source 
Ensure copying HttpHeaders before using these as default to prevent modifications of the original object. Adapt tests. Reuse header constants.

Original pull request: #293.
This commit is contained in:
Mark Paluch 2019-07-16 09:49:01 +02:00
parent c8c3cb7909
commit cf7fc61f7b
2 changed files with 18 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/main/java/org/springframework/data/elasticsearch/client/ClientConfigurationBuilder.java
View File

@ -112,7 +112,9 @@ class ClientConfigurationBuilder
Assert.notNull(defaultHeaders, "Default HTTP headers must not be null");
this.headers = defaultHeaders;
this.headers = new HttpHeaders();
this.headers.addAll(defaultHeaders);
return this;
}
@ -167,6 +169,7 @@ class ClientConfigurationBuilder
}
headers.setBasicAuth(username, password);
}
return new DefaultClientConfiguration(this.hosts, this.headers, this.useSsl, this.sslContext, this.soTimeout,
this.connectTimeout);
}

27
src/test/java/org/springframework/data/elasticsearch/client/ClientConfigurationUnitTests.java
View File

@ -34,8 +34,6 @@ import org.springframework.http.HttpHeaders;
*/
public class ClientConfigurationUnitTests {
private static final String AUTHORIZATION_HEADER = "Authorization";
@Test // DATAES-488
public void shouldCreateSimpleConfiguration() {
@ -85,23 +83,23 @@ public class ClientConfigurationUnitTests {
@Test // DATAES-607
public void shouldAddBasicAuthenticationHeaderWhenNoHeadersAreSet() {
final String username = "secretUser";
final String password = "secretPassword";
String username = "secretUser";
String password = "secretPassword";
ClientConfiguration clientConfiguration = ClientConfiguration.builder() //
.connectedTo("foo", "bar") //
.withBasicAuth(username, password) //
.build();
assertThat(clientConfiguration.getDefaultHeaders().get(AUTHORIZATION_HEADER))
assertThat(clientConfiguration.getDefaultHeaders().get(HttpHeaders.AUTHORIZATION))
.containsOnly(buildBasicAuth(username, password));
}
@Test // DATAES-607
public void shouldAddBasicAuthenticationHeaderAndKeepHeaders() {
final String username = "secretUser";
final String password = "secretPassword";
String username = "secretUser";
String password = "secretPassword";
HttpHeaders defaultHeaders = new HttpHeaders();
defaultHeaders.set("foo", "bar");
@ -109,17 +107,20 @@ public class ClientConfigurationUnitTests {
ClientConfiguration clientConfiguration = ClientConfiguration.builder() //
.connectedTo("foo", "bar") //
.withBasicAuth(username, password) //
.withDefaultHeaders(defaultHeaders).build();
final HttpHeaders httpHeaders = clientConfiguration.getDefaultHeaders();
.withDefaultHeaders(defaultHeaders) //
.build();
assertThat(httpHeaders.get(AUTHORIZATION_HEADER)).containsOnly(buildBasicAuth(username, password));
assertThat(httpHeaders.get("foo")).containsOnly("bar");
HttpHeaders httpHeaders = clientConfiguration.getDefaultHeaders();
assertThat(httpHeaders.get(HttpHeaders.AUTHORIZATION)).containsOnly(buildBasicAuth(username, password));
assertThat(httpHeaders.getFirst("foo")).isEqualTo("bar");
assertThat(defaultHeaders.get(HttpHeaders.AUTHORIZATION)).isNull();
}
private String buildBasicAuth(String username, String password) {
private static String buildBasicAuth(String username, String password) {
HttpHeaders headers = new HttpHeaders();
headers.setBasicAuth(username, password);
return headers.get(AUTHORIZATION_HEADER).get(0);
return headers.getFirst(HttpHeaders.AUTHORIZATION);
}
}