如何使用 Mockito 模拟超类中的特定方法调用？

使用 Mockito 模拟超类中的特定方法调用

在测试场景中，可能需要仅模拟超类中的特定方法调用，而保留他人原有的行为。这可以使用 Mockito 灵活的模拟功能来实现。

考虑以下类层次结构：

<code class="java">class BaseService {

    public void save() {...}
}

class ChildService extends BaseService {

    public void save() {
        // Custom logic
        super.save();
    }
}</code>

在此示例中，我们只想模拟 ChildService 中对 super.save() 的调用同时允许执行 BaseService 中原始的 save() 实现。

为了实现这一点，我们可以使用间谍函数创建 ChildService 的代理：

<code class="java">ChildService classToTest = Mockito.spy(new ChildService());</code>

接下来，我们需要在 ChildService 的上下文中存根 super.save() 的行为。但是，由于 super 引用了 ChildService 中的 BaseService 类，因此在使用 Mockito.doNothing() 之前，我们需要将 classToTest 强制转换为 BaseService。

<code class="java">Mockito.doNothing().when((BaseService)classToTest).validate();</code>

通过对 BaseService 的 validate() 方法进行存根，我们可以有效地防止它在测试期间被调用。因此，只有 ChildService 中的 super.save() 调用会被模拟拦截。

为了完成测试，我们现在可以在 classToTest 上调用 save() 并断言 ChildService 中的预期逻辑已执行：

<code class="java">classToTest.save();
verify(classToTest).load();</code>

这种方法允许我们有选择地模拟超类中的特定方法调用，同时保留继承方法的功能。在修改类层次结构或重构代码不可行的情况下，它被证明是有价值的。

以上是如何使用 Mockito 模拟超类中的特定方法调用？的详细内容。更多信息请关注PHP中文网其他相关文章！