mockito给定大小的anyList

Question

我正在用mockito验证一个方法已被调用.方法:

public void createButtons(final List<Button> buttonsConfiguration) {...}

由于传递哪个列表无关紧要,因此我验证该方法的调用方式如下:

verify(mock).createButtons(Matchers.anyListOf(Button.class));

但是,它的大小List很重要.因此,List列表必须具有X元素并不重要.

这有可能吗？

Answer 1

一种方法是使用Captor

ArgumentCaptor<List> captor = ArgumentCaptor.forClass(List.class);
verify(mock).createButtons(captor.capture());
assertEquals(x, captor.getValue().size()); // or if expecting multiple lists:
assertEquals(x, captor.getValues().size());

有关文档,请参见http://docs.mockito.googlecode.com/hg/org/mockito/Mockito.html#15.

您还可以使用自定义参数匹配器.文档显示了一个完全符合您要求的示例:

http://docs.mockito.googlecode.com/hg/org/mockito/ArgumentMatcher.html

 class IsListOfTwoElements extends ArgumentMatcher<List> {
     public boolean matches(Object list) {
         return ((List) list).size() == 2;
     }
 }

 List mock = mock(List.class);
 when(mock.addAll(argThat(new IsListOfTwoElements()))).thenReturn(true);
 mock.addAll(Arrays.asList("one", "two"));
 verify(mock).addAll(argThat(new IsListOfTwoElements()));

例如,您可以添加构造函数,以便指定所需的列表大小等.

Answer 2

使用 Mockito 3.x 和 2.1+，您可以使用 Java 8 lambda 表达式：

verify(mock).createButtons(argThat(list -> list.size() == 5));

与 Mockito 2.1 及以下类似：

verify(mock).createButtons(argThat(list -> ((List) list).size() == 5));

要检查空性，它甚至更容易：

verify(mock).createButtons(argThat(List::isEmpty));

Answer 3

Hamcrest提供了一种更简单的方法.

verify(mock).addAll(argThat(IsCollectionWithSize.hasSize(4)));