如何在方法中验证创建的对象的方法调用？

2024-03-02 12:05:01

## 验证在方法内部创建的对象中的方法调用

场景

在测试中，你可能需要验证在一个方法内部创建的对象中的方法是否被调用。例如，你有一个 Foo 类，它在 foo 方法中创建了一个 Bar 对象并调用 someMethod 方法。你要验证在调用 foo 之后，someMethod 被调用了一次。

解决方法

使用 Mockito，你可以通过以下步骤验证这一行为：

1. 创建 Foo 类模拟对象

Foo foo = Mockito.mock(Foo.class);

2. 捕获内部创建的对象

使用 ArgumentCaptor 来捕获内部创建的对象：

ArgumentCaptor<Bar> barCaptor = ArgumentCaptor.forClass(Bar.class);

3. 验证 Foo 方法

验证 foo 方法被调用了一次：

Mockito.verify(foo, times(1)).foo();

4. 验证内部对象的方法

从 barCaptor 中获取内部对象，然后对其进行验证：

Bar bar = barCaptor.getValue();
Mockito.verify(bar, times(1)).someMethod();

代码示例

完整的测试代码如下：

@Test
public void testVerifyMethodCalledOnObjectCreatedWithinMethod() {
    Foo foo = Mockito.mock(Foo.class);
    ArgumentCaptor<Bar> barCaptor = ArgumentCaptor.forClass(Bar.class);

    Mockito.verify(foo, times(1)).foo();

    Bar bar = barCaptor.getValue();
    Mockito.verify(bar, times(1)).someMethod();
}