如何在Mockito中正确匹配varargs

我一直在尝试使用Mockito模拟具有vararg参数的方法：

interface A {
  B b(int x, int y, C... c);
}

A a = mock(A.class);
B b = mock(B.class);

when(a.b(anyInt(), anyInt(), any(C[].class))).thenReturn(b);
assertEquals(b, a.b(1, 2));

这不起作用，但是如果我改为这样做：

when(a.b(anyInt(), anyInt())).thenReturn(b);
assertEquals(b, a.b(1, 2));

尽管我在对方法进行存根时已经完全省略了varargs参数，但这仍然有效。

有什么线索吗？

Mockito 1.8.1引入了anyVararg（）匹配器：

when(a.b(anyInt(), anyInt(), Matchers.<String>anyVararg())).thenReturn(b);

另请参阅历史记录：https : //code.google.com/archive/p/mockito/issues/62

弃用后编辑新语法：

when(a.b(anyInt(), anyInt(), ArgumentMatchers.<String>any())).thenReturn(b);

一个未公开的功能：如果要开发与vararg参数匹配的自定义Matcher，则需要实现它 org.mockito.internal.matchers.VarargMatcher以使其正常工作。这是一个空的标记接口，如果没有Matcher调用带有varargs的方法，则Mockito将无法正确比较参数。

例如：

class MyVarargMatcher extends ArgumentMatcher<C[]> implements VarargMatcher {
    @Override public boolean matches(Object varargArgument) {
        return /* does it match? */ true;
    }
}

when(a.b(anyInt(), anyInt(), argThat(new MyVarargMatcher()))).thenReturn(b);

基于Eli Levine的答案，这是一个更通用的解决方案：

import org.hamcrest.Description;
import org.hamcrest.Matcher;
import org.mockito.ArgumentMatcher;
import org.mockito.internal.matchers.VarargMatcher;

import static org.mockito.Matchers.argThat;

public class VarArgMatcher<T> extends ArgumentMatcher<T[]> implements VarargMatcher {

    public static <T> T[] varArgThat(Matcher<T[]> hamcrestMatcher) {
        argThat(new VarArgMatcher(hamcrestMatcher));
        return null;
    }

    private final Matcher<T[]> hamcrestMatcher;

    private VarArgMatcher(Matcher<T[]> hamcrestMatcher) {
        this.hamcrestMatcher = hamcrestMatcher;
    }

    @Override
    public boolean matches(Object o) {
        return hamcrestMatcher.matches(o);
    }

    @Override
    public void describeTo(Description description) {
        description.appendText("has varargs: ").appendDescriptionOf(hamcrestMatcher);
    }

}

然后，您可以将其与hamcrest的数组匹配器一起使用：

verify(a).b(VarArgMatcher.varArgThat(
            org.hamcrest.collection.IsArrayContaining.hasItemInArray("Test")));

（显然，静态导入将使其更具可读性。）

我一直在使用Peter Westmacott的答案中的代码，但是使用Mockito 2.2.15，您现在可以执行以下操作：

verify(a).method(100L, arg1, arg2, arg3)

arg1, arg2, arg3varargs 在哪里。

基于topchef的答案，

对于2.0.31-beta，我必须使用Mockito.anyVararg而不是Matchers.anyVararrg：

when(a.b(anyInt(), anyInt(), Mockito.<String>anyVararg())).thenReturn(b);

在我的情况下，我要捕获其参数的方法的签名为：

    public byte[] write(byte ... data) throws IOException;

在这种情况下，您应该显式转换为字节数组：

       when(spi.write((byte[])anyVararg())).thenReturn(someValue);

我正在使用模仿版本 1.10.19

您还可以遍历参数：

Object[] args = invocation.getArguments(); 
for( int argNo = 0; argNo < args.length; ++argNo) { 
    // ... do something with args[argNo] 
}

例如，检查其类型并将其正确转换，添加到列表或其他内容。

改编@topchef的答案，

Mockito.when(a.b(Mockito.anyInt(), Mockito.anyInt(), Mockito.any())).thenReturn(b);

根据Mockito 2.23.4的Java文档，Mockito.any（）“匹配任何内容，包括null和varargs。”

您可以通过传递ArgumentCaptor捕获，然后使用“ getAllValues”将变量作为列表检索来完成此操作，请参阅：https ://stackoverflow.com/a/55621731/11342928