Question

我正在为我的应用程序编写单元测试，我想知道Mockito框架是否有可能影响传递给返回模拟类的void的方法的对象。例如，调用一个模拟的验证类，该类包含一个返回void但通过作为参数传入的对象跟踪各种更改和元数据的方法。

public GetCartItemsOutput getCartItems(GetCartItemsInput getCartItemsInput) {
    CartItemsFilter cartItemsFilter = new CartItemsFilter();
    validator.validateCartItemsInput(getCartItemsInput, cartItemsFilter); ...

我为我的其他测试模拟了验证器类，但对于这个我需要模拟对cartItemsFilter对象的更改，我不知道该怎么做。

Answer 1

答案是肯定的，你可以，基于测试的需要，基本上有两个级别。

如果您只想测试与模拟对象的交互，可以使用verify()方法来验证是否调用了void方法。

如果您的测试确实需要模拟对象来修改传递给它的参数，则需要实现Answer：

已编辑以显示使用带有void方法的答案的正确形式

doAnswer(new Answer() {
    @Override
    Object answer(InvocationOnMock invocation) {
        Object[] args = invocation.getArguments();
        ((MyClass)args[0]).myClassSetMyField(NEW_VALUE);
        return null; // void method, so return null
    }
}).when(mock).someMethod();

在Java 8+中，上面的内容简化为lambda：

doAnswer(invocation-> {
    Object[] args = invocation.getArguments();
    ((MyClass)args[0]).myClassSetMyField(NEW_VALUE);
    return null;
}).when(mock).someMethod();

Answer 2

加入Kevin Welker的回答，如果你的MyClass是一个自定义类，那么在其中定义一个equals / hashcode方法，以便Mockito可以使用它来完全匹配方法调用。

就我而言，“MyClass”属于第三方API，因此我必须使用“any”方法，如下所示 -

 doAnswer(new Answer() {
    Object answer(InvocationOnMock invocation) {
        Object[] args = invocation.getArguments();
        ((MyClass)args[0]).myClassSetMyField(NEW_VALUE);
        return null; // void method, so return null
    }
}).when(mock).someMethod(any(MyClass.class));

Answer 3

ArgumentCaptor也可以提供帮助。

这是从here找到的代码段：

  ArgumentCaptor<GetCartItemsInput > argument = ArgumentCaptor.forClass(GetCartItemsInput .class);
  verify(mock).getCartItems(argument.capture());
  assertEquals(cartItemsFilter, argument.getValue());

如何模拟影响对象的void返回方法

3 个答案: