Use request factory to support Basic Authentication
Update `RestTemplateBuilder` to use a custom request factory to add authentication headers rather than an interceptor. Prior to this commit, the use of the `BasicAuthenticationInterceptor` interceptor could cause `OutOfMemoryError` whenever a large file is uploaded. See gh-17010
This commit is contained in:
committed by
Phillip Webb
parent
4ac1407af1
commit
18a0a7a2e8
@@ -19,11 +19,8 @@ package org.springframework.boot.test.web.client;
|
||||
import java.io.IOException;
|
||||
import java.lang.reflect.Field;
|
||||
import java.net.URI;
|
||||
import java.util.ArrayList;
|
||||
import java.util.Arrays;
|
||||
import java.util.Collections;
|
||||
import java.util.HashSet;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.Set;
|
||||
import java.util.function.Supplier;
|
||||
@@ -41,6 +38,8 @@ import org.apache.http.ssl.SSLContextBuilder;
|
||||
|
||||
import org.springframework.beans.BeanInstantiationException;
|
||||
import org.springframework.beans.BeanUtils;
|
||||
import org.springframework.boot.web.client.BasicAuthentication;
|
||||
import org.springframework.boot.web.client.BasicAuthenticationClientHttpRequestFactory;
|
||||
import org.springframework.boot.web.client.ClientHttpRequestFactorySupplier;
|
||||
import org.springframework.boot.web.client.RestTemplateBuilder;
|
||||
import org.springframework.boot.web.client.RootUriTemplateHandler;
|
||||
@@ -50,12 +49,11 @@ import org.springframework.http.HttpHeaders;
|
||||
import org.springframework.http.HttpMethod;
|
||||
import org.springframework.http.RequestEntity;
|
||||
import org.springframework.http.ResponseEntity;
|
||||
import org.springframework.http.client.AbstractClientHttpRequestFactoryWrapper;
|
||||
import org.springframework.http.client.ClientHttpRequestFactory;
|
||||
import org.springframework.http.client.ClientHttpRequestInterceptor;
|
||||
import org.springframework.http.client.ClientHttpResponse;
|
||||
import org.springframework.http.client.HttpComponentsClientHttpRequestFactory;
|
||||
import org.springframework.http.client.InterceptingClientHttpRequestFactory;
|
||||
import org.springframework.http.client.support.BasicAuthenticationInterceptor;
|
||||
import org.springframework.util.Assert;
|
||||
import org.springframework.util.ReflectionUtils;
|
||||
import org.springframework.web.client.DefaultResponseErrorHandler;
|
||||
@@ -86,6 +84,7 @@ import org.springframework.web.util.UriTemplateHandler;
|
||||
* @author Phillip Webb
|
||||
* @author Andy Wilkinson
|
||||
* @author Kristine Jetzke
|
||||
* @author Dmytro Nosan
|
||||
* @since 1.4.0
|
||||
*/
|
||||
public class TestRestTemplate {
|
||||
@@ -154,31 +153,35 @@ public class TestRestTemplate {
|
||||
|
||||
private Class<? extends ClientHttpRequestFactory> getRequestFactoryClass(
|
||||
RestTemplate restTemplate) {
|
||||
return getRequestFactory(restTemplate).getClass();
|
||||
}
|
||||
|
||||
private ClientHttpRequestFactory getRequestFactory(RestTemplate restTemplate) {
|
||||
ClientHttpRequestFactory requestFactory = restTemplate.getRequestFactory();
|
||||
if (InterceptingClientHttpRequestFactory.class
|
||||
.isAssignableFrom(requestFactory.getClass())) {
|
||||
Field requestFactoryField = ReflectionUtils.findField(RestTemplate.class,
|
||||
"requestFactory");
|
||||
ReflectionUtils.makeAccessible(requestFactoryField);
|
||||
requestFactory = (ClientHttpRequestFactory) ReflectionUtils
|
||||
.getField(requestFactoryField, restTemplate);
|
||||
while (requestFactory instanceof InterceptingClientHttpRequestFactory
|
||||
|| requestFactory instanceof BasicAuthenticationClientHttpRequestFactory) {
|
||||
requestFactory = unwrapRequestFactory(
|
||||
((AbstractClientHttpRequestFactoryWrapper) requestFactory));
|
||||
}
|
||||
return requestFactory.getClass();
|
||||
return requestFactory;
|
||||
}
|
||||
|
||||
private ClientHttpRequestFactory unwrapRequestFactory(
|
||||
AbstractClientHttpRequestFactoryWrapper requestFactory) {
|
||||
Field field = ReflectionUtils.findField(
|
||||
AbstractClientHttpRequestFactoryWrapper.class, "requestFactory");
|
||||
ReflectionUtils.makeAccessible(field);
|
||||
return (ClientHttpRequestFactory) ReflectionUtils.getField(field, requestFactory);
|
||||
}
|
||||
|
||||
private void addAuthentication(RestTemplate restTemplate, String username,
|
||||
String password) {
|
||||
if (username == null) {
|
||||
if (username == null || password == null) {
|
||||
return;
|
||||
}
|
||||
List<ClientHttpRequestInterceptor> interceptors = restTemplate.getInterceptors();
|
||||
if (interceptors == null) {
|
||||
interceptors = Collections.emptyList();
|
||||
}
|
||||
interceptors = new ArrayList<>(interceptors);
|
||||
interceptors.removeIf(BasicAuthenticationInterceptor.class::isInstance);
|
||||
interceptors.add(new BasicAuthenticationInterceptor(username, password));
|
||||
restTemplate.setInterceptors(interceptors);
|
||||
ClientHttpRequestFactory requestFactory = getRequestFactory(restTemplate);
|
||||
restTemplate.setRequestFactory(new BasicAuthenticationClientHttpRequestFactory(
|
||||
new BasicAuthentication(username, password), requestFactory));
|
||||
}
|
||||
|
||||
/**
|
||||
|
||||
@@ -20,13 +20,13 @@ import java.io.IOException;
|
||||
import java.lang.reflect.Method;
|
||||
import java.lang.reflect.Modifier;
|
||||
import java.net.URI;
|
||||
import java.util.List;
|
||||
|
||||
import org.apache.http.client.config.RequestConfig;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
import org.springframework.boot.test.web.client.TestRestTemplate.CustomHttpComponentsClientHttpRequestFactory;
|
||||
import org.springframework.boot.test.web.client.TestRestTemplate.HttpClientOption;
|
||||
import org.springframework.boot.web.client.BasicAuthenticationClientHttpRequestFactory;
|
||||
import org.springframework.boot.web.client.RestTemplateBuilder;
|
||||
import org.springframework.core.ParameterizedTypeReference;
|
||||
import org.springframework.http.HttpEntity;
|
||||
@@ -35,12 +35,9 @@ import org.springframework.http.HttpStatus;
|
||||
import org.springframework.http.RequestEntity;
|
||||
import org.springframework.http.client.ClientHttpRequest;
|
||||
import org.springframework.http.client.ClientHttpRequestFactory;
|
||||
import org.springframework.http.client.ClientHttpRequestInterceptor;
|
||||
import org.springframework.http.client.HttpComponentsClientHttpRequestFactory;
|
||||
import org.springframework.http.client.InterceptingClientHttpRequestFactory;
|
||||
import org.springframework.http.client.OkHttp3ClientHttpRequestFactory;
|
||||
import org.springframework.http.client.SimpleClientHttpRequestFactory;
|
||||
import org.springframework.http.client.support.BasicAuthenticationInterceptor;
|
||||
import org.springframework.mock.env.MockEnvironment;
|
||||
import org.springframework.mock.http.client.MockClientHttpRequest;
|
||||
import org.springframework.mock.http.client.MockClientHttpResponse;
|
||||
@@ -150,7 +147,7 @@ public class TestRestTemplateTests {
|
||||
public void authenticated() {
|
||||
assertThat(new TestRestTemplate("user", "password").getRestTemplate()
|
||||
.getRequestFactory())
|
||||
.isInstanceOf(InterceptingClientHttpRequestFactory.class);
|
||||
.isInstanceOf(BasicAuthenticationClientHttpRequestFactory.class);
|
||||
}
|
||||
|
||||
@Test
|
||||
@@ -227,7 +224,7 @@ public class TestRestTemplateTests {
|
||||
}
|
||||
|
||||
@Test
|
||||
public void withBasicAuthAddsBasicAuthInterceptorWhenNotAlreadyPresent() {
|
||||
public void withBasicAuthAddsBasicAuthClientFactoryWhenNotAlreadyPresent() {
|
||||
TestRestTemplate originalTemplate = new TestRestTemplate();
|
||||
TestRestTemplate basicAuthTemplate = originalTemplate.withBasicAuth("user",
|
||||
"password");
|
||||
@@ -235,20 +232,19 @@ public class TestRestTemplateTests {
|
||||
.containsExactlyElementsOf(
|
||||
originalTemplate.getRestTemplate().getMessageConverters());
|
||||
assertThat(basicAuthTemplate.getRestTemplate().getRequestFactory())
|
||||
.isInstanceOf(InterceptingClientHttpRequestFactory.class);
|
||||
.isInstanceOf(BasicAuthenticationClientHttpRequestFactory.class);
|
||||
assertThat(ReflectionTestUtils.getField(
|
||||
basicAuthTemplate.getRestTemplate().getRequestFactory(),
|
||||
"requestFactory"))
|
||||
.isInstanceOf(CustomHttpComponentsClientHttpRequestFactory.class);
|
||||
assertThat(basicAuthTemplate.getRestTemplate().getUriTemplateHandler())
|
||||
.isSameAs(originalTemplate.getRestTemplate().getUriTemplateHandler());
|
||||
assertThat(basicAuthTemplate.getRestTemplate().getInterceptors()).hasSize(1);
|
||||
assertBasicAuthorizationInterceptorCredentials(basicAuthTemplate, "user",
|
||||
"password");
|
||||
assertThat(basicAuthTemplate.getRestTemplate().getInterceptors()).isEmpty();
|
||||
assertBasicAuthorizationCredentials(basicAuthTemplate, "user", "password");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void withBasicAuthReplacesBasicAuthInterceptorWhenAlreadyPresent() {
|
||||
public void withBasicAuthReplacesBasicAuthClientFactoryWhenAlreadyPresent() {
|
||||
TestRestTemplate original = new TestRestTemplate("foo", "bar")
|
||||
.withBasicAuth("replace", "replace");
|
||||
TestRestTemplate basicAuth = original.withBasicAuth("user", "password");
|
||||
@@ -256,14 +252,14 @@ public class TestRestTemplateTests {
|
||||
.containsExactlyElementsOf(
|
||||
original.getRestTemplate().getMessageConverters());
|
||||
assertThat(basicAuth.getRestTemplate().getRequestFactory())
|
||||
.isInstanceOf(InterceptingClientHttpRequestFactory.class);
|
||||
.isInstanceOf(BasicAuthenticationClientHttpRequestFactory.class);
|
||||
assertThat(ReflectionTestUtils.getField(
|
||||
basicAuth.getRestTemplate().getRequestFactory(), "requestFactory"))
|
||||
.isInstanceOf(CustomHttpComponentsClientHttpRequestFactory.class);
|
||||
assertThat(basicAuth.getRestTemplate().getUriTemplateHandler())
|
||||
.isSameAs(original.getRestTemplate().getUriTemplateHandler());
|
||||
assertThat(basicAuth.getRestTemplate().getInterceptors()).hasSize(1);
|
||||
assertBasicAuthorizationInterceptorCredentials(basicAuth, "user", "password");
|
||||
assertThat(basicAuth.getRestTemplate().getInterceptors()).isEmpty();
|
||||
assertBasicAuthorizationCredentials(basicAuth, "user", "password");
|
||||
}
|
||||
|
||||
@Test
|
||||
@@ -394,17 +390,14 @@ public class TestRestTemplateTests {
|
||||
verify(requestFactory).createRequest(eq(absoluteUri), any(HttpMethod.class));
|
||||
}
|
||||
|
||||
private void assertBasicAuthorizationInterceptorCredentials(
|
||||
TestRestTemplate testRestTemplate, String username, String password) {
|
||||
@SuppressWarnings("unchecked")
|
||||
List<ClientHttpRequestInterceptor> requestFactoryInterceptors = (List<ClientHttpRequestInterceptor>) ReflectionTestUtils
|
||||
.getField(testRestTemplate.getRestTemplate().getRequestFactory(),
|
||||
"interceptors");
|
||||
assertThat(requestFactoryInterceptors).hasSize(1);
|
||||
ClientHttpRequestInterceptor interceptor = requestFactoryInterceptors.get(0);
|
||||
assertThat(interceptor).isInstanceOf(BasicAuthenticationInterceptor.class);
|
||||
assertThat(interceptor).hasFieldOrPropertyWithValue("username", username);
|
||||
assertThat(interceptor).hasFieldOrPropertyWithValue("password", password);
|
||||
private void assertBasicAuthorizationCredentials(TestRestTemplate testRestTemplate,
|
||||
String username, String password) {
|
||||
ClientHttpRequestFactory requestFactory = testRestTemplate.getRestTemplate()
|
||||
.getRequestFactory();
|
||||
Object authentication = ReflectionTestUtils.getField(requestFactory,
|
||||
"authentication");
|
||||
assertThat(authentication).hasFieldOrPropertyWithValue("username", username);
|
||||
assertThat(authentication).hasFieldOrPropertyWithValue("password", password);
|
||||
|
||||
}
|
||||
|
||||
|
||||
@@ -0,0 +1,86 @@
|
||||
/*
|
||||
* Copyright 2012-2019 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.springframework.boot.web.client;
|
||||
|
||||
import java.nio.charset.Charset;
|
||||
|
||||
import org.springframework.util.Assert;
|
||||
|
||||
/**
|
||||
* Basic authentication properties which are used by
|
||||
* {@link BasicAuthenticationClientHttpRequestFactory}.
|
||||
*
|
||||
* @author Dmytro Nosan
|
||||
* @since 2.2.0
|
||||
* @see BasicAuthenticationClientHttpRequestFactory
|
||||
*/
|
||||
public class BasicAuthentication {
|
||||
|
||||
private final String username;
|
||||
|
||||
private final String password;
|
||||
|
||||
private final Charset charset;
|
||||
|
||||
/**
|
||||
* Create a new {@link BasicAuthentication}.
|
||||
* @param username the username to use
|
||||
* @param password the password to use
|
||||
*/
|
||||
public BasicAuthentication(String username, String password) {
|
||||
this(username, password, null);
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a new {@link BasicAuthentication}.
|
||||
* @param username the username to use
|
||||
* @param password the password to use
|
||||
* @param charset the charset to use
|
||||
*/
|
||||
public BasicAuthentication(String username, String password, Charset charset) {
|
||||
Assert.notNull(username, "Username must not be null");
|
||||
Assert.notNull(password, "Password must not be null");
|
||||
this.username = username;
|
||||
this.password = password;
|
||||
this.charset = charset;
|
||||
}
|
||||
|
||||
/**
|
||||
* The username to use.
|
||||
* @return the username, never {@code null} or {@code empty}.
|
||||
*/
|
||||
public String getUsername() {
|
||||
return this.username;
|
||||
}
|
||||
|
||||
/**
|
||||
* The password to use.
|
||||
* @return the password, never {@code null} or {@code empty}.
|
||||
*/
|
||||
public String getPassword() {
|
||||
return this.password;
|
||||
}
|
||||
|
||||
/**
|
||||
* The charset to use.
|
||||
* @return the charset, or {@code null}.
|
||||
*/
|
||||
public Charset getCharset() {
|
||||
return this.charset;
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,67 @@
|
||||
/*
|
||||
* Copyright 2012-2019 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.springframework.boot.web.client;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.net.URI;
|
||||
|
||||
import org.springframework.http.HttpHeaders;
|
||||
import org.springframework.http.HttpMethod;
|
||||
import org.springframework.http.client.AbstractClientHttpRequestFactoryWrapper;
|
||||
import org.springframework.http.client.ClientHttpRequest;
|
||||
import org.springframework.http.client.ClientHttpRequestFactory;
|
||||
import org.springframework.util.Assert;
|
||||
|
||||
/**
|
||||
* {@link ClientHttpRequestFactory} to apply a given HTTP Basic Authentication
|
||||
* username/password pair, unless a custom Authorization header has been set before.
|
||||
*
|
||||
* @author Dmytro Nosan
|
||||
* @since 2.2.0
|
||||
*/
|
||||
public class BasicAuthenticationClientHttpRequestFactory
|
||||
extends AbstractClientHttpRequestFactoryWrapper {
|
||||
|
||||
private final BasicAuthentication authentication;
|
||||
|
||||
/**
|
||||
* Create a new {@link BasicAuthenticationClientHttpRequestFactory} which adds
|
||||
* {@link HttpHeaders#AUTHORIZATION} header for the given authentication.
|
||||
* @param authentication the authentication to use
|
||||
* @param clientHttpRequestFactory the factory to use
|
||||
*/
|
||||
public BasicAuthenticationClientHttpRequestFactory(BasicAuthentication authentication,
|
||||
ClientHttpRequestFactory clientHttpRequestFactory) {
|
||||
super(clientHttpRequestFactory);
|
||||
Assert.notNull(authentication, "Authentication must not be null");
|
||||
this.authentication = authentication;
|
||||
}
|
||||
|
||||
@Override
|
||||
protected ClientHttpRequest createRequest(URI uri, HttpMethod httpMethod,
|
||||
ClientHttpRequestFactory requestFactory) throws IOException {
|
||||
BasicAuthentication authentication = this.authentication;
|
||||
ClientHttpRequest request = requestFactory.createRequest(uri, httpMethod);
|
||||
HttpHeaders headers = request.getHeaders();
|
||||
if (!headers.containsKey(HttpHeaders.AUTHORIZATION)) {
|
||||
headers.setBasicAuth(authentication.getUsername(),
|
||||
authentication.getPassword(), authentication.getCharset());
|
||||
}
|
||||
return request;
|
||||
}
|
||||
|
||||
}
|
||||
@@ -33,7 +33,7 @@ import org.springframework.beans.BeanUtils;
|
||||
import org.springframework.http.client.AbstractClientHttpRequestFactoryWrapper;
|
||||
import org.springframework.http.client.ClientHttpRequestFactory;
|
||||
import org.springframework.http.client.ClientHttpRequestInterceptor;
|
||||
import org.springframework.http.client.support.BasicAuthenticationInterceptor;
|
||||
import org.springframework.http.client.InterceptingClientHttpRequestFactory;
|
||||
import org.springframework.http.converter.HttpMessageConverter;
|
||||
import org.springframework.util.Assert;
|
||||
import org.springframework.util.CollectionUtils;
|
||||
@@ -58,6 +58,7 @@ import org.springframework.web.util.UriTemplateHandler;
|
||||
* @author Phillip Webb
|
||||
* @author Andy Wilkinson
|
||||
* @author Brian Clozel
|
||||
* @author Dmytro Nosan
|
||||
* @since 1.4.0
|
||||
*/
|
||||
public class RestTemplateBuilder {
|
||||
@@ -74,7 +75,7 @@ public class RestTemplateBuilder {
|
||||
|
||||
private final ResponseErrorHandler errorHandler;
|
||||
|
||||
private final BasicAuthenticationInterceptor basicAuthentication;
|
||||
private final BasicAuthentication basicAuthentication;
|
||||
|
||||
private final Set<RestTemplateCustomizer> restTemplateCustomizers;
|
||||
|
||||
@@ -106,7 +107,7 @@ public class RestTemplateBuilder {
|
||||
Set<HttpMessageConverter<?>> messageConverters,
|
||||
Supplier<ClientHttpRequestFactory> requestFactorySupplier,
|
||||
UriTemplateHandler uriTemplateHandler, ResponseErrorHandler errorHandler,
|
||||
BasicAuthenticationInterceptor basicAuthentication,
|
||||
BasicAuthentication basicAuthentication,
|
||||
Set<RestTemplateCustomizer> restTemplateCustomizers,
|
||||
RequestFactoryCustomizer requestFactoryCustomizer,
|
||||
Set<ClientHttpRequestInterceptor> interceptors) {
|
||||
@@ -372,17 +373,28 @@ public class RestTemplateBuilder {
|
||||
|
||||
/**
|
||||
* Add HTTP basic authentication to requests. See
|
||||
* {@link BasicAuthenticationInterceptor} for details.
|
||||
* {@link BasicAuthenticationClientHttpRequestFactory} for details.
|
||||
* @param username the user name
|
||||
* @param password the password
|
||||
* @return a new builder instance
|
||||
* @since 2.1.0
|
||||
*/
|
||||
public RestTemplateBuilder basicAuthentication(String username, String password) {
|
||||
return basicAuthentication(new BasicAuthentication(username, password));
|
||||
}
|
||||
|
||||
/**
|
||||
* Add HTTP basic authentication to requests. See
|
||||
* {@link BasicAuthenticationClientHttpRequestFactory} for details.
|
||||
* @param basicAuthentication the authentication
|
||||
* @return a new builder instance
|
||||
* @since 2.2.0
|
||||
*/
|
||||
public RestTemplateBuilder basicAuthentication(
|
||||
BasicAuthentication basicAuthentication) {
|
||||
return new RestTemplateBuilder(this.detectRequestFactory, this.rootUri,
|
||||
this.messageConverters, this.requestFactorySupplier,
|
||||
this.uriTemplateHandler, this.errorHandler,
|
||||
new BasicAuthenticationInterceptor(username, password),
|
||||
this.uriTemplateHandler, this.errorHandler, basicAuthentication,
|
||||
this.restTemplateCustomizers, this.requestFactoryCustomizer,
|
||||
this.interceptors);
|
||||
}
|
||||
@@ -534,7 +546,7 @@ public class RestTemplateBuilder {
|
||||
RootUriTemplateHandler.addTo(restTemplate, this.rootUri);
|
||||
}
|
||||
if (this.basicAuthentication != null) {
|
||||
restTemplate.getInterceptors().add(this.basicAuthentication);
|
||||
configureBasicAuthentication(restTemplate);
|
||||
}
|
||||
restTemplate.getInterceptors().addAll(this.interceptors);
|
||||
if (!CollectionUtils.isEmpty(this.restTemplateCustomizers)) {
|
||||
@@ -561,6 +573,25 @@ public class RestTemplateBuilder {
|
||||
}
|
||||
}
|
||||
|
||||
private void configureBasicAuthentication(RestTemplate restTemplate) {
|
||||
ClientHttpRequestFactory requestFactory = restTemplate.getRequestFactory();
|
||||
while (requestFactory instanceof InterceptingClientHttpRequestFactory
|
||||
|| requestFactory instanceof BasicAuthenticationClientHttpRequestFactory) {
|
||||
requestFactory = unwrapRequestFactory(
|
||||
((AbstractClientHttpRequestFactoryWrapper) requestFactory));
|
||||
}
|
||||
restTemplate.setRequestFactory(new BasicAuthenticationClientHttpRequestFactory(
|
||||
this.basicAuthentication, requestFactory));
|
||||
}
|
||||
|
||||
private static ClientHttpRequestFactory unwrapRequestFactory(
|
||||
AbstractClientHttpRequestFactoryWrapper requestFactory) {
|
||||
Field field = ReflectionUtils.findField(
|
||||
AbstractClientHttpRequestFactoryWrapper.class, "requestFactory");
|
||||
ReflectionUtils.makeAccessible(field);
|
||||
return (ClientHttpRequestFactory) ReflectionUtils.getField(field, requestFactory);
|
||||
}
|
||||
|
||||
private <T> Set<T> append(Set<T> set, Collection<? extends T> additions) {
|
||||
Set<T> result = new LinkedHashSet<>((set != null) ? set : Collections.emptySet());
|
||||
result.addAll(additions);
|
||||
@@ -607,18 +638,11 @@ public class RestTemplateBuilder {
|
||||
|
||||
private ClientHttpRequestFactory unwrapRequestFactoryIfNecessary(
|
||||
ClientHttpRequestFactory requestFactory) {
|
||||
if (!(requestFactory instanceof AbstractClientHttpRequestFactoryWrapper)) {
|
||||
return requestFactory;
|
||||
}
|
||||
ClientHttpRequestFactory unwrappedRequestFactory = requestFactory;
|
||||
Field field = ReflectionUtils.findField(
|
||||
AbstractClientHttpRequestFactoryWrapper.class, "requestFactory");
|
||||
ReflectionUtils.makeAccessible(field);
|
||||
do {
|
||||
unwrappedRequestFactory = (ClientHttpRequestFactory) ReflectionUtils
|
||||
.getField(field, unwrappedRequestFactory);
|
||||
while (unwrappedRequestFactory instanceof AbstractClientHttpRequestFactoryWrapper) {
|
||||
unwrappedRequestFactory = unwrapRequestFactory(
|
||||
((AbstractClientHttpRequestFactoryWrapper) unwrappedRequestFactory));
|
||||
}
|
||||
while (unwrappedRequestFactory instanceof AbstractClientHttpRequestFactoryWrapper);
|
||||
return unwrappedRequestFactory;
|
||||
}
|
||||
|
||||
|
||||
@@ -0,0 +1,81 @@
|
||||
/*
|
||||
* Copyright 2012-2019 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.springframework.boot.web.client;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.net.URI;
|
||||
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
|
||||
import org.springframework.http.HttpHeaders;
|
||||
import org.springframework.http.HttpMethod;
|
||||
import org.springframework.http.client.ClientHttpRequest;
|
||||
import org.springframework.http.client.ClientHttpRequestFactory;
|
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat;
|
||||
import static org.mockito.ArgumentMatchers.any;
|
||||
import static org.mockito.BDDMockito.when;
|
||||
import static org.mockito.Mockito.mock;
|
||||
|
||||
/**
|
||||
* Tests for {@link BasicAuthenticationClientHttpRequestFactory}.
|
||||
*
|
||||
* @author Dmytro Nosan
|
||||
*/
|
||||
public class BasicAuthenticationClientHttpRequestFactoryTests {
|
||||
|
||||
private final HttpHeaders headers = new HttpHeaders();
|
||||
|
||||
private final BasicAuthentication authentication = new BasicAuthentication("spring",
|
||||
"boot");
|
||||
|
||||
private ClientHttpRequestFactory requestFactory;
|
||||
|
||||
@Before
|
||||
public void setUp() throws IOException {
|
||||
ClientHttpRequestFactory requestFactory = mock(ClientHttpRequestFactory.class);
|
||||
ClientHttpRequest request = mock(ClientHttpRequest.class);
|
||||
when(requestFactory.createRequest(any(), any())).thenReturn(request);
|
||||
when(request.getHeaders()).thenReturn(this.headers);
|
||||
this.requestFactory = new BasicAuthenticationClientHttpRequestFactory(
|
||||
this.authentication, requestFactory);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void shouldAddAuthorizationHeader() throws IOException {
|
||||
ClientHttpRequest request = createRequest();
|
||||
assertThat(request.getHeaders().get(HttpHeaders.AUTHORIZATION))
|
||||
.containsExactly("Basic c3ByaW5nOmJvb3Q=");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void shouldNotAddAuthorizationHeaderAuthorizationAlreadySet()
|
||||
throws IOException {
|
||||
this.headers.setBasicAuth("boot", "spring");
|
||||
ClientHttpRequest request = createRequest();
|
||||
assertThat(request.getHeaders().get(HttpHeaders.AUTHORIZATION))
|
||||
.doesNotContain("Basic c3ByaW5nOmJvb3Q=");
|
||||
|
||||
}
|
||||
|
||||
private ClientHttpRequest createRequest() throws IOException {
|
||||
return this.requestFactory.createRequest(URI.create("http://localhost:8080"),
|
||||
HttpMethod.POST);
|
||||
}
|
||||
|
||||
}
|
||||
@@ -16,6 +16,7 @@
|
||||
|
||||
package org.springframework.boot.web.client;
|
||||
|
||||
import java.nio.charset.StandardCharsets;
|
||||
import java.time.Duration;
|
||||
import java.util.Collections;
|
||||
import java.util.Set;
|
||||
@@ -35,7 +36,6 @@ import org.springframework.http.client.HttpComponentsClientHttpRequestFactory;
|
||||
import org.springframework.http.client.InterceptingClientHttpRequestFactory;
|
||||
import org.springframework.http.client.OkHttp3ClientHttpRequestFactory;
|
||||
import org.springframework.http.client.SimpleClientHttpRequestFactory;
|
||||
import org.springframework.http.client.support.BasicAuthenticationInterceptor;
|
||||
import org.springframework.http.converter.HttpMessageConverter;
|
||||
import org.springframework.http.converter.ResourceHttpMessageConverter;
|
||||
import org.springframework.http.converter.StringHttpMessageConverter;
|
||||
@@ -324,12 +324,13 @@ public class RestTemplateBuilderTests {
|
||||
|
||||
@Test
|
||||
public void basicAuthenticationShouldApply() {
|
||||
RestTemplate template = this.builder.basicAuthentication("spring", "boot")
|
||||
BasicAuthentication basicAuthentication = new BasicAuthentication("spring",
|
||||
"boot", StandardCharsets.UTF_8);
|
||||
RestTemplate template = this.builder.basicAuthentication(basicAuthentication)
|
||||
.build();
|
||||
ClientHttpRequestInterceptor interceptor = template.getInterceptors().get(0);
|
||||
assertThat(interceptor).isInstanceOf(BasicAuthenticationInterceptor.class);
|
||||
assertThat(interceptor).extracting("username").containsExactly("spring");
|
||||
assertThat(interceptor).extracting("password").containsExactly("boot");
|
||||
ClientHttpRequestFactory requestFactory = template.getRequestFactory();
|
||||
assertThat(requestFactory).hasFieldOrPropertyWithValue("authentication",
|
||||
basicAuthentication);
|
||||
}
|
||||
|
||||
@Test
|
||||
@@ -406,19 +407,19 @@ public class RestTemplateBuilderTests {
|
||||
.messageConverters(this.messageConverter).rootUri("http://localhost:8080")
|
||||
.errorHandler(errorHandler).basicAuthentication("spring", "boot")
|
||||
.requestFactory(() -> requestFactory).customizers((restTemplate) -> {
|
||||
assertThat(restTemplate.getInterceptors()).hasSize(2)
|
||||
.contains(this.interceptor).anyMatch(
|
||||
(ic) -> ic instanceof BasicAuthenticationInterceptor);
|
||||
assertThat(restTemplate.getInterceptors()).hasSize(1);
|
||||
assertThat(restTemplate.getMessageConverters())
|
||||
.contains(this.messageConverter);
|
||||
assertThat(restTemplate.getUriTemplateHandler())
|
||||
.isInstanceOf(RootUriTemplateHandler.class);
|
||||
assertThat(restTemplate.getErrorHandler()).isEqualTo(errorHandler);
|
||||
ClientHttpRequestFactory actualRequestFactory = restTemplate
|
||||
ClientHttpRequestFactory interceptingRequestFactory = restTemplate
|
||||
.getRequestFactory();
|
||||
assertThat(actualRequestFactory)
|
||||
assertThat(interceptingRequestFactory)
|
||||
.isInstanceOf(InterceptingClientHttpRequestFactory.class);
|
||||
assertThat(actualRequestFactory).hasFieldOrPropertyWithValue(
|
||||
Object basicAuthRequestFactory = ReflectionTestUtils
|
||||
.getField(interceptingRequestFactory, "requestFactory");
|
||||
assertThat(basicAuthRequestFactory).hasFieldOrPropertyWithValue(
|
||||
"requestFactory", requestFactory);
|
||||
}).build();
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user