Mockito rest 模板始终为 null 用于模拟 ZA8CFDE6331BD59EB2AC9Z6F8911C4B666

Question

When mocking a class which contains a rest template, the rest template is always null, my code is:

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"); 我注意到 object classA 包含 rest 模板，但它是 null。

Answer 1

Mocking 是在依赖项上完成的，而不是在被测的 class 上完成的。 还有一些方法可以在测试方法下模拟 class ，这可以使用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);
   }
}

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

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

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

  classa.method(url);
}