验证方法在方法内部被调用，而不在该方法内部调用其他方法

Question

public class ServiceClass {
    @Autowired
    ConnClient connClient;
    public void abc() {
         result = connClient.query(arg1, arg2); // verify called once.
         allDocsReturned = result.getResults(); // throws NPE
    }
}

我需要验证是否至少调用了一次 connClient.query(arg1,arg2)。所以在测试用例文件中我调用方法 abc()

@InjectMocks
ServiceClass serviceClass;
@Mock
ConnClient connClient;
@Test
public void testABC() {
  serviceClass.abc();
  verify(connClient, times(1)).query(arg1,arg2);
}

我得到 NullPointerException 因为上面调用的 result.getResults() 行因为结果为空，并且测试用例失败。 请帮忙。

Answer 1

您的课程 ConnClient 是一个模拟，这就是为什么调用 connClient.query(arg1, arg2) 将返回 null 广告，因此在取消引用结果时会引发 NPE。

这种情况下可以指定方法调用的返回值，即

doReturn(someObject).when(connClient).query(arg1,arg2);

someObject 允许作为模拟对象（或者，您可以返回具体值）。

一旦完成，result 将不再为空，也不会抛出 NPE。

请注意，如果稍后取消引用 result，您还必须模拟 getResults() 调用。

编辑：

@InjectMocks
ServiceClass serviceClass;
@Mock
ConnClient connClient;
@Mock
Result result;
@Test
public void testABC() {
    doReturn(result).when(connClient).query(arg1,arg2);       
    serviceClass.abc();
    verify(connClient, times(1)).query(arg1,arg2);
}

现在结果不会为空，但是，allDocsReturned 仍然为空，但不会导致带有您发布的代码的 NPE

Answer 2

您需要模拟对connClient.query 和ResultSet 的调用：

@InjectMocks
ServiceClass serviceClass;
@Mock
ConnClient connClient;
@Test
public void testABC() {
  ResultSet rs = mock(ResultSet.class);
  when(rs.getResults()).thenReturn(//here return the object of type of your variable allDocsReturned);
  when(connClient.query(arg1,arg2)).thenReturn(rs);
  serviceClass.abc();
  verify(connClient, times(1)).query(arg1,arg2);
}