java - 使用 Mockito 验证无参数和私有(private)方法

Question

这个问题已经有答案了:

How do I test a class that has private methods, fields or inner classes? (59 个回答)

已关闭 3 年前。

我是 JUnit 和 Mockito 的新手，下面的类定义了一个需要测试的方法execute，我想知道是否可以模拟这种类型的类？请阐明一些观点并分享您的想法。

public class MyRule implements SuperRule{    
  private OrderList orderList;
  private ScheduleList scheduleList;

  public MyRule (OrderList orderList, ScheduleList scheduleList) {
    this.orderList = orderList;
    this.scheduleList = scheduleList;
  }

  @Override
  public void execute() {
    createWeeklyClassificationRule(orderList);      
    handle(scheduleList);

  }      
  private void createWeeklyClassificationRule(OrderList orderList) {
   //......
  }      
  private void handle(ScheduleList scheduleList) { 
    //......      
  }
}

Answer 1

您可以模拟 scheduleList 和 orderList 并使用 verify 来确保从它们调用正确的方法。

public class MyRuleTest
{
  private MyRule myRule;
  private ScheduleList scheduleListMock;
  private OrderList orderListMock;

  @Before
  public void setUp() throws Exception
  {
    scheduleListMock = mock(ScheduleList.class);
    orderListMock = mock(OrderList.class);

    myRule = new MyRule(orderListMock, scheduleListMock);
  }

  @Test
  public void testExecute()
  {
    myRule.execute();
    verify(scheduleListMock).foo();
    verify(orderListMock).bar();
  }

...

您只需将 foo 和 bar 替换为您希望调用的任何方法即可。

如果您正在测试 MyRule，您不应该 mock 它。一个好的经验法则是不要模拟正在测试的类。