mock resttemplate将服务测试为restFul客户端

Question

我有一个服务类,用春天写的,有一些方法.其中一个充当了如下的宁静消费者:

.....
        HttpEntity request = new HttpEntity<>(getHeadersForRequest());
        RestTemplate restTemplate = new RestTemplate();
        String url = ENDPOINT_URL.concat(ENDPOINT_API1);

        UriComponentsBuilder builder = UriComponentsBuilder.fromHttpUrl(url)
                .queryParam("param1", parameter1);
        ReportModel infoModel = null;
        try{
            infoModel = restTemplate.exchange(builder.toUriString(), HttpMethod.GET, request, ReportModel.class).getBody();
        }catch (HttpClientErrorException | HttpServerErrorException e){
            e.printStackTrace();
        }

我想用来Mockito模拟我的服务,但每个与restful服务器实例交互的方法都是一个新的RestTemplate.我要创建一个静态类来将它注入我的服务中？

Answer 1

依赖注入的一个好处是能够轻松地模拟您的依赖项.在您的情况下,创建RestTemplatebean 会容易得多:

@Bean
public RestTemplate restTemplate() {
    return new RestTemplate();
}

而不是new RestTemplate()在您的客户端使用,您应该使用:

@Autowired
private RestTemplate restTemplate;

对于使用Mockito进行单元测试,您必须模拟RestTemplate,例如使用:

@RunWith(MockitoJUnitRunner.class)
public class ClientTest {
    @InjectMocks
    private Client client;
    @Mock
    private RestTemplate restTemplate;
}

在这种情况下,Mockito将模拟并注入RestTemplate你的bean Client.如果你不喜欢通过反射进行RestTemplate模拟和注入,你总是可以使用单独的构造函数或setter来注入模拟.

现在您可以编写如下测试:

client.doStuff();
verify(restTemplate).exchange(anyString(), eq(HttpMethod.GET), any(HttpModel.class), eq(ReportModel.class));

你可能想要测试更多,但它给你一个基本的想法.