java - Mockito - 内部方法调用

Question

我有一个名为 Availability.java 的类，它有两个方法。

 public Long getStockLevelStage() {
     //some logic
      getStockLevelLimit();
    }

    public Long getStockLevelLimit() {
      String primaryOnlineArea = classificationFeatureHelper.getFirstFeatureName(productModel, FEATURE_CODE_PRODUCT_ONLINE_AREA_PRIMARY, language);
................
      return new Long();
    }

我正在编写一个单元测试类AvailabilityTest.java。

@RunWith(MockitoJUnitRunner.class)
public class AvailabilityTest {
  @InjectMocks
  private Availability availability = new Availability();

  @Test
  public void testGetStockLevelStage() {
    availability.getStockLevelStage();
  }
}

当我调用 availability.getStockLevelStage() 方法时，它会调用 getStockLevelLimit() 方法。是否可以模拟内部方法调用？

在这种情况下，我不希望在 getStockLevelStage() 执行时执行 getStockLevelLimit()。

请帮忙。

Answer 1

试试这个:

@RunWith(MockitoJUnitRunner.class)
public class AvailabilityTest {
    @InjectMocks
    @Spy
    private Availability availability = new Availability();

    @Test
    public void testGetStockLevelStage() {
       Mockito.doReturn(expectedLong).when(availability).getStockLevelLimit();
       availability.getStockLevelStage();
    }
}

这是我在 Mockito Spying 上写的一篇文章如果您需要进一步阅读。