在TestRestTemplate中启用Cookie和重定向

Question

如何将TestRestTemplate.HttpClientOption.ENABLE_REDIRECTS和TestRestTemplate.HttpClientOption.ENABLE_COOKIES添加到spring-boots TestRestTemplate？

我正在使用spring-boot，因此我自动为我配置了TestRestTemplate。

我可以在创建之前使用RestTemplateBuilder来定制此bean。问题是我不能看到如何添加这些选项：

@Bean 
    public RestTemplateBuilder restTemplateBuilder() { 
     return new RestTemplateBuilder() 
       .errorHandler(new ResponseErrorHandler() { 
        ... 
       }); 
    } 

的documentation有一些构造函数接受这些选项，但问题是，已经为我创建的豆。

Answer 1

您可以创建一个TestRestTemplate，并使用@Bean注释将其呈现给Spring。

例如：

@Bean 
@Primary 
public TestRestTemplate testRestTemplate() { 
    RestTemplate restTemplate = new RestTemplateBuilder() 
      .errorHandler(new ResponseErrorHandler() { 
       @Override 
       public boolean hasError(ClientHttpResponse response) throws IOException { 
        return false; 
       } 

       @Override 
       public void handleError(ClientHttpResponse response) throws IOException { 

       } 
      }).build(); 

    return new TestRestTemplate(restTemplate, user, password, TestRestTemplate.HttpClientOption.ENABLE_REDIRECTS, TestRestTemplate.HttpClientOption.ENABLE_COOKIES); 
} 

或者，如果您不需要定制RestTemplate然后使用下面的构造函数（它在内部实例为您RestTemplate）：

@Bean 
@Primary 
public TestRestTemplate testRestTemplate() { 
    return new TestRestTemplate(TestRestTemplate.HttpClientOption.ENABLE_REDIRECTS, TestRestTemplate.HttpClientOption.ENABLE_COOKIES); 
} 

更新1来解决此评论：

当我运行我的测试，现在我得到以下错误org.apache.http.ProtocolException: Target host is not specified

Spring提供的TestRestTemplate被配置为解决相对于http://localhost:${local.server.port}路径。因此，当您用自己的实例替换Spring提供的实例时，您需要提供完整的地址（包括主机和端口）或者配置LocalHostUriTemplateHandler（您可以在org.springframework.boot.test.context.SpringBootTestContextCustomizer.TestRestTemplateFactory中看到此代码）来配置您自己的TestRestTemplate。下面是后一种方法的一个例子：

@Bean 
@Primary 
public TestRestTemplate testRestTemplate(ApplicationContext applicationContext) { 
    RestTemplate restTemplate = new RestTemplateBuilder() 
      .errorHandler(new ResponseErrorHandler() { 
       @Override 
       public boolean hasError(ClientHttpResponse response) throws IOException { 
        return false; 
       } 

       @Override 
       public void handleError(ClientHttpResponse response) throws IOException { 

       } 
      }).build(); 

    TestRestTemplate testRestTemplate = 
      new TestRestTemplate(restTemplate, user, password, TestRestTemplate.HttpClientOption 
        .ENABLE_REDIRECTS, TestRestTemplate.HttpClientOption.ENABLE_COOKIES); 

    // let this testRestTemplate resolve paths relative to http://localhost:${local.server.port} 
    LocalHostUriTemplateHandler handler = 
      new LocalHostUriTemplateHandler(applicationContext.getEnvironment(), "http"); 
    testRestTemplate.setUriTemplateHandler(handler); 

    return testRestTemplate; 
} 

有了这个bean配置以下测试案例使用了定制的TestRestTemplate并成功调用本地主机上的春天启动的应用程序：

@RunWith(SpringRunner.class) 
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT) 
public class RestTemplateTest { 

    @Autowired 
    private TestRestTemplate restTemplate; 

    @Test 
    public void test() { 
     ResponseEntity<String> forEntity = this.restTemplate.getForEntity("/some/endpoint", String.class); 
     System.out.println(forEntity.getBody()); 
    } 
}