如何使用Mockito捕获变量参数?

时间:2013-11-08 04:21:45

标签: java mocking mockito

有一种方法可变参数:

class A {
    public void setNames(String... names) {}
}

现在我想用mockito来模拟它,并捕获传递给它的名字。但我无法找到一种方法来捕获任意数量的名字,我只能这样得到它们:

ArgumentCaptor<String> captor1 = ArgumentCaptor.fromClass(String.class);
ArgumentCaptor<String> captor2 = ArgumentCaptor.fromClass(String.class);

A mock = Mockito.mock(A.class);
mock.setNames("Jeff", "Mike");
Mockito.verity(mock).setNames(captor1.capture(), captor2.capture());
String name1 = captor1.getValue(); // Jeff
String name2 = captor2.getValue(); // Mike

如果我传递三个名字,它就无法使用,我必须定义一个captor3来捕获第三个名字。

如何解决?

2 个答案:

答案 0 :(得分:34)

Mockito 1.10.5引入了此功能。

对于问题中的代码示例,以下是捕获varargs的一种方法:

    ArgumentCaptor<String> varArgs = ArgumentCaptor.forClass(String.class);
    A mock = Mockito.mock(A.class);
    mock.setNames("Jeff", "Mike", "John");
    Mockito.verify(mock).setNames(varArgs.capture());

    //Results may be validated thus:
    List<String> expected = Arrays.asList("Jeff", "Mike", "John");
    assertEquals(expected, varArgs.getAllValues());

有关详细信息,请参阅ArgumentCaptor javadoc

答案 1 :(得分:3)

截至今天(2013年11月7日),它似乎已得到解决,但尚未公布,需要进行一些额外的工作。有关详细信息,请参阅groups threadissue tracker