如何在JUnit测试中重复捕获循环中的参数？

Question

考虑以下情况......

public boolean someMethod(List<SomeType> list) {
    for(SomeType someObject: list) {
        if(aMethod(someObject.getId())) {
            return false;
        }
    }
    return true;
}

我想重复捕获someObject.getId()。如果我使用这样的东西......

...

@Captor
ArgumentCaptor<Long> longArgumentCaptor; //Assuming ID is Long
...

when(aMethod(longArgumentCaptor.capture()).thenReturn(true);

assertThat(x.someMethod(Arrays.asList(new SomeType(65L),new SomeType(734L)))).isTrue();

assertThat(longArgumentCaptor.getValue()).isEqualTo(734L); //<--
...

ArgumentCaptor仅捕获最后一个值。

但是，我想验证每个值。要做到这一点，我需要存储它们。如何捕获每次迭代？

Answer 1

ArgumentCaptor有一个getAllValues方法正好用于此用例 - 它会将所有调用中的值集合返回给mock方法。