Mockito验证在我的情况下一次调用一个函数

Question

我正在使用Mockito编写我的测试用例。我有一个简单的类，其中包含countPerson(boolean)我要测试的功能：

public class School {
  //School is a singleton class.

  public void countPerson(boolean includeTeacher) {
       if (includeTeacher) {
          countIncludeTeacher();
          return;
       }
       countOnlyStudents();
  }

  public void countIncludeTeacher() {...}
  public void countOnlyStudents() {...}
}

在单元测试中，我要测试countPerson(boolean)功能：

public class SchoolTest{
   private School mSchool;
   @Before
   public void setUp(){
      mSchool = School.getInstance();
   }
   @Test 
   public void testCountPerson() {
       mSchool.countPerson(true);
       //How to test/verify countIncludeTeacher() is invoked once?
   }
}

countIncludeTeacher()在我的测试案例中，如何使用Mockito进行检查/验证？

Answer 1

您将不得不使用间谍。这里的问题是您要验证是否在
真实
对象而非模拟对象上调用了方法。您不能在此处使用模拟，因为它将对类中的所有方法进行存根，因此countPerson缺省情况下也对存根不执行任何操作。

@Test 
public void testCountPerson() {
    School school = School.getInstance();
    School spySchool = Mockito.spy(school);
    spySchool.countPerson(true);
    verify(spySchool).countIncludeTeacher();
}

但是，请注意，使用间谍程序时应格外小心，因为除非经过特殊处理，否则实际方法都会被调用。引用Mockito Javadoc：

真正的间谍应该 小心谨慎 地使用，例如在处理遗留代码时。