使用Mockito进行泛型方法的ClassCastException

时间:2015-03-01 22:13:51

标签: java mockito generic-method

我正在使用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中使用两种不同类型的泛型方法时,不会出现异常。

有人可以帮忙吗?

1 个答案:

答案 0 :(得分:2)

<强> TL;博士

以多种测试方法拆分代码。 或者链接thenReturn API。

长篇故事

您可能已经意识到使用类型擦除实现的Java泛型,这意味着您在代码中看到的大多数泛型只出现在源代码中,而不是出现在已编译的字节码中。

E.g。以下签名

<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通过相同的匹配器发现同一个电话。虽然代码看起来像两个不同的存根,但mockito并不知道,并且只能假设开发人员想要来覆盖这个第一个存根。

对于在某些@Before方法中声明默认存根且在某些@Test方法中需要覆盖存根的情况,需要此模拟行为。

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

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

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

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