java - 对于模拟对象，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);
}