avoid check class by string name

4c61d8ab · igor-suhorukov · Brian Clozel · 09e0f1e8 · 4c61d8ab
Commit 4c61d8ab authored Mar 14, 2018 by igor-suhorukov Committed by Brian Clozel Mar 14, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

TestRestTemplate.java ...pringframework/boot/test/web/client/TestRestTemplate.java +2 -2

No files found.
--- a/spring-boot-project/spring-boot-test/src/main/java/org/springframework/boot/test/web/client/TestRestTemplate.java
+++ b/spring-boot-project/spring-boot-test/src/main/java/org/springframework/boot/test/web/client/TestRestTemplate.java
@@ -138,8 +138,8 @@ public class TestRestTemplate {
 			HttpClientOption... httpClientOptions) {
 		Assert.notNull(restTemplate, "RestTemplate must not be null");
 		this.httpClientOptions = httpClientOptions;
-		if (getRequestFactoryClass(restTemplate).getName().equals(
-				"org.springframework.http.client.HttpComponentsClientHttpRequestFactory")) {
+		if (getRequestFactoryClass(restTemplate).isAssignableFrom(
+				HttpComponentsClientHttpRequestFactory.class)) {
 			restTemplate.setRequestFactory(
 					new CustomHttpComponentsClientHttpRequestFactory(httpClientOptions));
 		}