是否有一种干净的方法来用泛型参数模拟类?假设我必须模拟一个类Foo<T>,我需要传递到一个方法,期望Foo<Bar>。我可以很容易地做到以下几点:

Foo mockFoo = mock(Foo.class);
when(mockFoo.getValue).thenReturn(new Bar());

假设getValue()返回泛型类型t,但当我后来将它传递给一个期望Foo<Bar>的方法时,它将有小猫。选角是唯一的方法吗?


当前回答

JUnit5:在测试类上使用@ExtendWith(MockitoExtension.class),然后添加这个字段:

@Mock
Foo<Bar> barMock;

其他回答

JUnit5:在测试类上使用@ExtendWith(MockitoExtension.class),然后添加这个字段:

@Mock
Foo<Bar> barMock;

我同意人们不应该在类或方法中压制警告,因为这样会忽略其他意外压制的警告。但是恕我直言,对于只影响一行代码的警告,不发出警告是完全合理的。

@SuppressWarnings("unchecked")
Foo<Bar> mockFoo = mock(Foo.class);

创建一个测试工具方法。特别有用,如果你需要它不止一次。

@Test
public void testMyTest() {
    // ...
    Foo<Bar> mockFooBar = mockFoo();
    when(mockFooBar.getValue).thenReturn(new Bar());

    Foo<Baz> mockFooBaz = mockFoo();
    when(mockFooBaz.getValue).thenReturn(new Baz());

    Foo<Qux> mockFooQux = mockFoo();
    when(mockFooQux.getValue).thenReturn(new Qux());
    // ...
}

@SuppressWarnings("unchecked") // still needed :( but just once :)
private <T> Foo<T> mockFoo() {
    return mock(Foo.class);
}

这里有一个有趣的例子:方法接收泛型集合并返回相同基类型的泛型集合。例如:

Collection<? extends Assertion> map(Collection<? extends Assertion> assertions);

这个方法可以通过组合Mockito anyCollectionOf匹配器和Answer来模拟。

when(mockedObject.map(anyCollectionOf(Assertion.class))).thenAnswer(
     new Answer<Collection<Assertion>>() {
         @Override
         public Collection<Assertion> answer(InvocationOnMock invocation) throws Throwable {
             return new ArrayList<Assertion>();
         }
     });

您总是可以创建一个中间类/接口来满足您想要指定的泛型类型。例如,如果Foo是一个接口,您可以在测试类中创建以下接口。

private interface FooBar extends Foo<Bar>
{
}

在Foo是一个非final类的情况下,你可以用下面的代码扩展这个类,并做同样的事情:

public class FooBar extends Foo<Bar>
{
}

然后你可以使用下面的代码来使用上面的例子:

Foo<Bar> mockFoo = mock(FooBar.class);
when(mockFoo.getValue()).thenReturn(new Bar());