From 4c61d8abfe1b75a5d90eac72051aa96dc7e68364 Mon Sep 17 00:00:00 2001
From: igor-suhorukov <igor.suhorukov@gmail.com>
Date: Wed, 14 Mar 2018 09:46:30 +0300
Subject: [PATCH] avoid check class by string name

---
 .../boot/test/web/client/TestRestTemplate.java                | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git 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
index cf8368bf4f..4c9e845d2e 100644
--- 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));
 		}