给定大小的模拟列表

Question

我正在用嘲笑验证方法已被调用。方法：

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()));

例如，您还可以添加一个构造函数，以便可以指定所需的列表大小，等等。