Spring/spring-boot
Spring/spring-boot
1
0
Fork 0
mirror of https://github.com/spring-projects/spring-boot.git synced 2024-09-03 04:26:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

Provide access to root URI from TestRestTemplate

Browse Source 
Closes gh-10641
This commit is contained in:
tinexw 2017-10-14 13:29:45 +02:00 committed by Andy Wilkinson
parent e92e56dda5
commit c1205c3243
2 changed files with 36 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
spring-boot-project/spring-boot-test/src/main/java/org/springframework/boot/test/web/client/TestRestTemplate.java
View File

@ -78,6 +78,7 @@ import org.springframework.web.util.UriTemplateHandler;
* @author Dave Syer
* @author Phillip Webb
* @author Andy Wilkinson
* @author Kristine Jetzke
* @since 1.4.0
*/
public class TestRestTemplate {
@ -165,6 +166,15 @@ public class TestRestTemplate {
this.restTemplate.setUriTemplateHandler(handler);
}
public String getRootUri() {
UriTemplateHandler uriTemplateHandler = this.restTemplate.getUriTemplateHandler();
if (RootUriTemplateHandler.class
.isAssignableFrom(uriTemplateHandler.getClass())) {
return ((RootUriTemplateHandler) uriTemplateHandler).getRootUri();
}
return "";
}
/**
* Retrieve a representation by doing a GET on the specified URL. The response (if
* any) is converted and returned.

26
spring-boot-project/spring-boot-test/src/test/java/org/springframework/boot/test/web/client/TestRestTemplateTests.java
View File

@ -63,6 +63,7 @@ import static org.mockito.Mockito.verify;
* @author Phillip Webb
* @author Stephane Nicoll
* @author Andy Wilkinson
* @author Kristine Jetzke
*/
public class TestRestTemplateTests {
@ -81,6 +82,31 @@ public class TestRestTemplateTests {
.isInstanceOf(HttpComponentsClientHttpRequestFactory.class);
}
@Test
public void getRootUriRootUriSetViaRestTemplateBuilder() {
String rootUri = "http://example.com";
RestTemplate delegate = new RestTemplateBuilder().rootUri(rootUri).build();
assertThat(new TestRestTemplate(delegate).getRootUri()).isEqualTo(rootUri);
}
@Test
public void getRootUriRootUriSetViaLocalHostUriTemplateHandler() {
String rootUri = "http://example.com";
TestRestTemplate template = new TestRestTemplate();
LocalHostUriTemplateHandler templateHandler = mock(
LocalHostUriTemplateHandler.class);
given(templateHandler.getRootUri()).willReturn(rootUri);
template.setUriTemplateHandler(templateHandler);
assertThat(template.getRootUri()).isEqualTo(rootUri);
}
@Test
public void getRootUriRootUriNotSet() {
assertThat(new TestRestTemplate().getRootUri()).isEqualTo("");
}
@Test
public void authenticated() {
assertThat(new TestRestTemplate("user", "password").getRestTemplate()