使用Mockito模拟泛型类型时的ClassCastException

Question

我有一个带有泛型参数的抽象测试类。

public abstract class AbstractCarTest<G> {
    ...
    @Mock
    protected G carMock;
    ...
}

我已经实现了它的一个具体测试类。

public class TruckTest extends AbstractCarTest<Truck> {
    ...
    when(truckFactory.getTruck(anyString()).return(carMock);
    ...
}

方法签名如下所示

public Truck getTruck(String name);

当运行TruckTest时，我得到一个ClassCastException说

java.lang.ClassCastException: org.mockito.internal.creation.jmock.ClassImposterizer$ClassWithSuperclassToWorkAroundCglibBug$$EnhancerByMockitoWithCGLIB$$... cannot be cast to com.example.Truck

为什么？我有什么办法可以解决这个问题吗？

Answer 1

只是想从@geoand的答案上改进一点。

public abstract class AbstractCarTest<G> {
    ...
    protected G carMock;
    ...

    @Before
    public void setUp() throws Exception {
        carMock= Mockito.mock(getClazz());
    }

    private Class<G> getClazz() {
        return (Class<G>) ((ParameterizedType) getClass().getGenericSuperclass()).getActualTypeArguments()[0];
    }
}


public class TruckTest extends AbstractCarTest<Truck> {
    ...
    when(truckFactory.getTruck(anyString()).return(carMock);
    ...
}

这最终消除了对抽象方法的需求。提供给 getActualTypeArguments（） 的索引将取决于类型参数在声明中的显示位置。在大多数情况下，这将是0，但是如果你有这样的东西：

public abstract class AbstractCarTest<A, B, C, G>

如果您想使用< code>G的< code>Class，您需要提供< code>3的索引。

Answer 2

Mockito似乎不会在不知道类是具体类型的情况下模拟类。

您可以通过执行以下操作来解决问题：

public abstract class AbstractCarTest<G> {
    ...
    protected G carMock;
    ...

    @Before
    public void setUp() throws Exception {
        carMock= Mockito.mock(getClazz());
    }

    protected abstract Class<G> getClazz();
}


public class TruckTest extends AbstractCarTest<Truck> {
    ...
    when(truckFactory.getTruck(anyString()).return(carMock);
    ...

    @Override
    protected Class<Truck> getClazz() {
        return Truck.class;
    }
}

Answer 3

Mockito正在运行时生成您的模拟类。Mocks是通过对给定类进行子类化并覆盖其所有方法来创建的。（因此不嘲笑最终类或方法的限制。）在运行时，所有泛型类型都被擦除并替换为它们的上限类型，在您的AbstractCarTest中，这是Object，因为您没有指定显式上限。因此，Mockito将您的原始类视为：

public abstract class AbstractCarTest {

  @Mock
  protected Object carMock;
}

在运行时，并将创建一个扩展Object而不是您想要的Truck类的mock。（您看不到这一点，因为cglib有一个错误，它无法直接扩展Object。相反，Mockito正在扩展一个名为ClassSusSuperclass ToWorkAroundCglibBug的内部类。）通常，编译器会在编译时发出类型错误，其中泛型类型仍然可用，但在运行时，您会遇到堆污染。

解决方法如下：

public abstract class AbstractCarTest<T> {

  protected abstract T getCarMock();

  // define some base test cases here that use the generic type.
}

public class TruckTest extends AbstractCarTest<Truck> {

  @Mock
  private Truck carMock;

  @Override
  protected Truck getCarMock() { return carMock; }

  // ...
  when(truckFactory.getTruck(anyString()).return(getCarMock());
}

通过在不使用泛型的情况下定义模拟类型，您可以通过getter从抽象基类访问模拟，其中模拟类型被正确定义为Truck。