如何使用 mockito 捕获特定类型的列表

Question

有没有办法使用 mockitos ArgumentCaptore 捕获特定类型的列表。这不起作用：

ArgumentCaptor<ArrayList<SomeType>> argument = ArgumentCaptor.forClass(ArrayList.class);

Answer 1

使用@Captor annotation 可以避免嵌套泛型问题：

public class Test{

    @Mock
    private Service service;

    @Captor
    private ArgumentCaptor<ArrayList<SomeType>> captor;

    @Before
    public void init(){
        MockitoAnnotations.initMocks(this);
    }

    @Test 
    public void shouldDoStuffWithListValues() {
        //...
        verify(service).doStuff(captor.capture()));
    }
}

Answer 2

是的，这是一个通用的泛型问题，而不是 mockito 特定的问题。

ArrayList<SomeType> 没有类对象，因此您不能以类型安全的方式将此类对象传递给需要 Class<ArrayList<SomeType>> 的方法。

您可以将对象转换为正确的类型：

Class<ArrayList<SomeType>> listClass =
              (Class<ArrayList<SomeType>>)(Class)ArrayList.class;
ArgumentCaptor<ArrayList<SomeType>> argument = ArgumentCaptor.forClass(listClass);

这会给出一些关于不安全强制转换的警告，当然，如果不检查元素，您的 ArgumentCaptor 无法真正区分 ArrayList<SomeType> 和 ArrayList<AnotherType>。

（正如另一个答案中提到的，虽然这是一个通用的泛型问题，但对于 @Captor 注释的类型安全问题，有一个 Mockito 特定的解决方案。它仍然无法区分 ArrayList<SomeType>和ArrayList<OtherType>。）

编辑：

还请查看tenshi 的评论。您可以将原始代码更改为这个简化版本：

final ArgumentCaptor<List<SomeType>> listCaptor
        = ArgumentCaptor.forClass((Class) List.class);

Answer 3

如果您不害怕旧的 java 风格（非类型安全的泛型）语义，这也可以工作并且很简单：

ArgumentCaptor<List> argument = ArgumentCaptor.forClass(List.class);
verify(subject.method(argument.capture()); // run your code
List<SomeType> list = argument.getValue(); // first captured List, etc.

Answer 4

List<String> mockedList = mock(List.class);

List<String> l = new ArrayList();
l.add("someElement");

mockedList.addAll(l);

ArgumentCaptor<List> argumentCaptor = ArgumentCaptor.forClass(List.class);

verify(mockedList).addAll(argumentCaptor.capture());

List<String> capturedArgument = argumentCaptor.<List<String>>getValue();

assertThat(capturedArgument, hasItem("someElement"));

Answer 5

基于@tenshi 和@pkalinow 的cmets（也感谢@rogerdpack），以下是创建列表参数捕获器的简单解决方案，它也禁用“使用未经检查或不安全的操作”警告 ：

@SuppressWarnings("unchecked")
final ArgumentCaptor<List<SomeType>> someTypeListArgumentCaptor =
    ArgumentCaptor.forClass(List.class);

完整示例 here 和相应的通过 CI 构建和测试运行 here。

我们的团队已经在我们的单元测试中使用了一段时间，这对我们来说似乎是最直接的解决方案。

Answer 6

对于较早版本的junit，可以这样做

Class<Map<String, String>> mapClass = (Class) Map.class;
ArgumentCaptor<Map<String, String>> mapCaptor = ArgumentCaptor.forClass(mapClass);

Answer 7

我在我的 Android 应用中测试活动时遇到了同样的问题。我用ActivityInstrumentationTestCase2 和MockitoAnnotations.initMocks(this); 没用。 我用另一个类分别解决了这个问题。例如：

class CaptorHolder {

        @Captor
        ArgumentCaptor<Callback<AuthResponse>> captor;

        public CaptorHolder() {
            MockitoAnnotations.initMocks(this);
        }
    }

那么，在活动测试方法中：

HubstaffService hubstaffService = mock(HubstaffService.class);
fragment.setHubstaffService(hubstaffService);

CaptorHolder captorHolder = new CaptorHolder();
ArgumentCaptor<Callback<AuthResponse>> captor = captorHolder.captor;

onView(withId(R.id.signInBtn))
        .perform(click());

verify(hubstaffService).authorize(anyString(), anyString(), captor.capture());
Callback<AuthResponse> callback = captor.getValue();

Answer 8

关于这个确切的问题有一个open issue in Mockito's GitHub。

我找到了一个简单的解决方法，它不会强制您在测试中使用注释：

import org.mockito.ArgumentCaptor;
import org.mockito.Captor;
import org.mockito.MockitoAnnotations;

public final class MockitoCaptorExtensions {

    public static <T> ArgumentCaptor<T> captorFor(final CaptorTypeReference<T> argumentTypeReference) {
        return new CaptorContainer<T>().captor;
    }

    public static <T> ArgumentCaptor<T> captorFor(final Class<T> argumentClass) {
        return ArgumentCaptor.forClass(argumentClass);
    }

    public interface CaptorTypeReference<T> {

        static <T> CaptorTypeReference<T> genericType() {
            return new CaptorTypeReference<T>() {
            };
        }

        default T nullOfGenericType() {
            return null;
        }

    }

    private static final class CaptorContainer<T> {

        @Captor
        private ArgumentCaptor<T> captor;

        private CaptorContainer() {
            MockitoAnnotations.initMocks(this);
        }

    }

}

这里发生的是我们创建一个新类 @Captor 注释并将捕获器注入其中。然后我们只需提取捕获者并从我们的静态方法中返回它。

在您的测试中，您可以像这样使用它：

ArgumentCaptor<Supplier<Set<List<Object>>>> fancyCaptor = captorFor(genericType());

或者使用类似于 Jackson 的 TypeReference 的语法：

ArgumentCaptor<Supplier<Set<List<Object>>>> fancyCaptor = captorFor(
    new CaptorTypeReference<Supplier<Set<List<Object>>>>() {
    }
);

它有效，因为 Mockito 实际上不需要任何类型信息（例如，与序列化程序不同）。