对于模拟对象，Mockito 休息模板始终为空

Question

当模拟一个包含休息模板的类时，休息模板总是空的，我的代码是：

public class ClassA {

    private final RestTemplate restTemplate;

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

    public byte[] method(String url) {
       // do some restTemplate.getForObject()
    }
}

   @RunWith(SpringRunner.class)
   @SpringBootTest(classes = {Application.class})
public class TestClass {

    @Mock
    private ClassA classa

     public void test1() {

      Mockito.doReturn(byeArray).when(classA).method("url");
     }
}

检查行时 Mockito.doReturn(byeArray).when(classA).method("url");我注意到对象 classA 包含其余模板，但它是空的。

Answer 1

模拟是在依赖项上完成的，而不是在被测类上。还有一些方法可以模拟被测类的方法，可以使用Spy 来完成。

在你目前的情况下，应该是这样的。

@RunWith(SpringRunner.class)
@SpringBootTest(classes = {Application.class})
public class TestClass {

    @InjectMock
    private ClassA classa

    @Mock
    private RestTemplate restTemplate

    @Test
    public void test1() {
      Mockito.doReturn(byeArray).when(restTemplate).method(url);
   }
}

被测类ClassA 应使用@InjectMocks 注释，依赖项应使用@Mock 模拟。

另外需要注意的是，在您的测试用例中，您应该调用需要测试的ClassA 方法。

@Test
public void test1() {
  Mockito.doReturn(byeArray).when(restTemplate).doSomething(url);

  classa.method(url);
}