Search in sources :

Example 1 with RootUriTemplateHandler

use of org.springframework.boot.web.client.RootUriTemplateHandler in project spring-boot by spring-projects.

the class RootUriRequestExpectationManager method forRestTemplate.

/**
	 * Return {@link RequestExpectationManager} to be used for binding with the specified
	 * {@link RestTemplate}. If the {@link RestTemplate} is using a
	 * {@link RootUriTemplateHandler} then a {@link RootUriRequestExpectationManager} is
	 * returned, otherwise the source manager is returned unchanged.
	 * @param restTemplate the source REST template
	 * @param expectationManager the source {@link RequestExpectationManager}
	 * @return a {@link RequestExpectationManager} to be bound to the template
	 */
public static RequestExpectationManager forRestTemplate(RestTemplate restTemplate, RequestExpectationManager expectationManager) {
    Assert.notNull(restTemplate, "RestTemplate must not be null");
    UriTemplateHandler templateHandler = restTemplate.getUriTemplateHandler();
    if (templateHandler instanceof RootUriTemplateHandler) {
        return new RootUriRequestExpectationManager(((RootUriTemplateHandler) templateHandler).getRootUri(), expectationManager);
    }
    return expectationManager;
}
Also used : RootUriTemplateHandler(org.springframework.boot.web.client.RootUriTemplateHandler) UriTemplateHandler(org.springframework.web.util.UriTemplateHandler) RootUriTemplateHandler(org.springframework.boot.web.client.RootUriTemplateHandler)

Aggregations

RootUriTemplateHandler (org.springframework.boot.web.client.RootUriTemplateHandler)1 UriTemplateHandler (org.springframework.web.util.UriTemplateHandler)1