如何使用 Mockito 模拟 RestTemplate 的最终新实例

Question

Component
public class MyService {
  private final ObjectMapper objectMapper = new ObjectMapper();
  private final RestTemplate restTemplate = new RestTemplate();

  public void myMethod(){
      

    ------- codes to test above it------ 
     
     HttpEntity<String> httpEntity = new HttpEntity<>(objectMapper
          .writeValueAsString(message), httpHeaders);
      String response = restTemplate
          .postForObject(getUrl(), httpEntity, String.class);

  }

}

我试过@Spy，但它不起作用

@InjectMocks
  private MyService myService;
  
  @Spy
  private RestTemplate restTemplate;


 when(restTemplate.postForObject(
    getUrl(),
    httpEntity,
    String.class
)).thenReturn(getResponse());

Answer 1

如果你想窥探：

@Test
void test(){
    MyService myService = Mockito.spy(MyService.class);
    myService.myMethod();
}

这样就得到了 RestTemplate 的实例。

我个人更喜欢将 RestTemplate 作为依赖项接收，这使得它易于模拟。

例如：

private final RestTemplate restTemplate;

public MyService(RestTemplate restTemplate) {
    this.restTemplate = restTemplate;
}

在你的测试中

//@RunWith(MockitoJUnitRunner.class) //junit4
@ExtendWith(MockitoExtension.class) //junit 5
public class MyServiceTest {

@Mock
private RestTemplate restTemplate;

@InjectMocks
private MyService myService;

@Test
void test(){
    when(restTemplate.postForObject(
            getUrl(),
            httpEntity,
            String.class
    )).thenReturn(getResponse());
    //do whatever you like
   }

}

Answer 2

spring-boot-test 提供了一种通过MockBean和SpyBean注解模拟 bean 的好方法。

尝试使用@SpyBean代替@Spy和@MockBean代替@Mock 。

@MockBean
private RestTemplate restTemplate

在这种情况下，不需要@InjectMocks 。