java - 使用 Mockito 进行泛型方法的 ClassCastException

Question

我正在使用mockito来测试通用方法。但是当我运行 junit-test 时，我得到了 ClassCastException。

被测试的方法如下所示:

public ExampleClass {
    public <E> E randomObject(List<E> list) {
            E e = list.get(0);
            return e;
    }
}

模拟看起来像这样:

ExampleClass exampleMock = mock(ExampleClass.class);
List listMock = mock(List.class);
when(exampleMock.randomObject(Matchers<List<String>any())).thenReturn("Hello");
when(exampleMock.randomObject(Matchers.List<Integer>any())).thenReturn(20);

异常出现在when-method的第二个定义处。看起来该方法只接受一种不可更改的类型。但为什么会这样呢？当我在纯 java 中使用具有两种不同类型的泛型方法时，不会出现异常。

有人可以帮忙吗？

Answer 1

tl;博士

将代码拆分为多个测试方法。或者链接 thenReturn API。

长话

您可能知道 Java 泛型是通过类型删除实现的，这意味着您在代码中看到的大多数泛型仅存在于源代码中，而不存在于已编译的字节码中。

例如此处签名如下

<E> E randomObject(List<E> list)

编译为

Object randomObject(List list)

这就是mockito看到的签名。对于匹配器来说也是如此:

when(exampleMock.randomObject(Matchers.<List<String>>any())).thenReturn("Hello");
when(exampleMock.randomObject(Matchers.<List<Integer>>any())).thenReturn(20);

变成了

when(exampleMock.randomObject(Matchers.any())).thenReturn("Hello");
when(exampleMock.randomObject(Matchers.any())).thenReturn(20);

Mockito 发现这是具有相同匹配器的相同调用。虽然它的代码读起来像两个不同的 stub ，但 mockito 不知道这一点，只能假设开发人员想要覆盖第一个 stub 。

对于在某些 @Before 方法中声明默认 stub 并且需要在某些 @Test 方法中覆盖该 stub 的场景，需要这种模拟行为.

在这种情况下，代码应该被拆分，或者如果是更复杂场景的一部分， stub 应该使用链 API，例如

when(exampleMock.randomObject(Matchers.anyList()))
       .thenReturn("Hello")
       .thenReturn(20);

_请注意，匹配器已更改为 anyList()，这可能对编译时间更加友好。

测试可以使用@SuppressWarnings("unchecked")进行注释。