我需要在Mockito中验证具有多个参数的方法,但只需要捕获一个参数,其他的我只需要一个简单的匹配器。这可能吗?
例如,如果我有:
@Mock
private Map<K,V> mockedMap;
...
ArgumentCaptor<K> argument = ArgumentCaptor.forClass(K.class);
verify(mockedMap).put(argument.capture(), any(V.class));
在这种情况下,我是否需要为每个参数编写捕获程序,尽管我只需要捕获第一个参数?
在这种情况下,我是否需要为每个参数编写捕获程序,尽管我只需要捕获第一个参数?
durron597的答案是正确的—如果您想捕获其中一个参数,则不需要捕获所有参数。有一点需要澄清:对ArgumentCaptor.capture()
的调用计数为Mockito匹配器,并且在Mockito中,如果您对任何方法参数使用匹配器,则必须对所有参数使用匹配器。
对于方法yourMock.yourMethod(int, int, int)
和ArgumentCaptor<Integer> intCaptor
:
/* good: */ verify(yourMock).yourMethod(2, 3, 4); // eq by default
/* same: */ verify(yourMock).yourMethod(eq(2), eq(3), eq(4));
/* BAD: */ verify(yourMock).yourMethod(intCaptor.capture(), 3, 4);
/* fixed: */ verify(yourMock).yourMethod(intCaptor.capture(), eq(3), eq(4));
也可以:
verify(yourMock).yourMethod(intCaptor.capture(), eq(5), otherIntCaptor.capture());
verify(yourMock).yourMethod(intCaptor.capture(), anyInt(), gt(9000));
当然可以。为什么不呢?
import java.util.Map;
import org.junit.*;
import org.mockito.*;
import static org.mockito.Mockito.*;
import static org.junit.Assert.*;
public class MockitoTest {
@Mock
private Map<Integer, String> mockedMap;
@Before
public void setup() {
MockitoAnnotations.initMocks(this);
}
@Test
public void testCaptor() {
mockedMap.put(5, "Hello World!");
ArgumentCaptor<Integer> argument = ArgumentCaptor.forClass(Integer.class);
verify(mockedMap).put(argument.capture(), any(String.class));
assertEquals(5L, argument.getValue().longValue());
}
}
正确工作并通过。
作为题外话,您几乎从来不想模拟
List
和Map
这样的数据结构,因为正确模拟它们的所有行为是一件非常痛苦的事情,而且如果您添加了一个元素,然后该元素实际上并不存在,那么大多数代码都不会很高兴。在您的情况下,创建部分模拟(使用Mockito.spy
)可能比创建实际模拟更好。