首页 >Java >java教程 >如何在 Mockito 中部分模拟方法:在保留其他方法的同时模拟特定方法?

如何在 Mockito 中部分模拟方法:在保留其他方法的同时模拟特定方法?

Barbara Streisand
Barbara Streisand原创
2024-12-05 13:26:11758浏览

How to Partially Mock Methods in Mockito: Mocking Specific Methods While Preserving Others?

部分模拟:在 Mockito 中模拟特定方法

测试类时,可能需要仅模拟某些方法,同时保留行为其他人的。在 Mockito 中,这是通过“部分模拟”实现的,其中模拟了一系列方法,而未模拟的方法则按预期执行。

考虑以下 Stock 类,其中包含 getPrice()、getQuantity() 和 getValue( ) 方法:

public class Stock {
    ...
    public double getValue() {
        return getPrice() * getQuantity();
    }
}

在测试用例中,如果我们希望模拟 getPrice() 和 getQuantity() 方法但保留getValue() 要正常执行,我们可以定义一个部分模拟,如下所示:

Stock stock = mock(Stock.class);
when(stock.getPrice()).thenReturn(100.00);
when(stock.getQuantity()).thenReturn(200);
when(stock.getValue()).thenCallRealMethod();

这里, getPrice() 和 getQuantity() 方法有模拟实现,而 getValue() 执行与原始 Stock 中一样

或者,使用“spy”API:

Stock stock = spy(Stock.class);
when(stock.getPrice()).thenReturn(100.00);
when(stock.getQuantity()).thenReturn(200);

所有其他方法调用,在本例中为 getValue(),使用真正的实现。

为了确保不调用真正的方法并且一致地应用模拟逻辑,请考虑以下替代表示法:

doReturn(100.00).when(stock).getPrice();
doReturn(200).when(stock).getQuantity();

但是,需要注意的是,在此示例中,getValue() 方法直接依赖于价格和数量,而不是它们的 getter。因此,模拟 getPrice() 和 getQuantity() 可能仍然无法产生所需的行为。

另一种方法是完全避免模拟,而是依赖 Stock 类的直接实例化:

Stock stock = new Stock(100.00, 200);
double value = stock.getValue();

通过创建真实实例,我们可以确保 getValue() 根据实际价格和数量值按预期执行。

以上是如何在 Mockito 中部分模拟方法:在保留其他方法的同时模拟特定方法?的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn