是否有一种干净的方法来用泛型参数模拟类?假设我必须模拟一个类Foo<T>,我需要传递到一个方法,期望Foo<Bar>。我可以很容易地做到以下几点:
Foo mockFoo = mock(Foo.class);
when(mockFoo.getValue).thenReturn(new Bar());
假设getValue()返回泛型类型t,但当我后来将它传递给一个期望Foo<Bar>的方法时,它将有小猫。选角是唯一的方法吗?
是否有一种干净的方法来用泛型参数模拟类?假设我必须模拟一个类Foo<T>,我需要传递到一个方法,期望Foo<Bar>。我可以很容易地做到以下几点:
Foo mockFoo = mock(Foo.class);
when(mockFoo.getValue).thenReturn(new Bar());
假设getValue()返回泛型类型t,但当我后来将它传递给一个期望Foo<Bar>的方法时,它将有小猫。选角是唯一的方法吗?
当前回答
创建一个测试工具方法。特别有用,如果你需要它不止一次。
@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>();
}
});
我同意人们不应该在类或方法中压制警告,因为这样会忽略其他意外压制的警告。但是恕我直言,对于只影响一行代码的警告,不发出警告是完全合理的。
@SuppressWarnings("unchecked")
Foo<Bar> mockFoo = mock(Foo.class);
所以你得到了这个:
Foo mockFoo = mock(Foo.class);
解决方法,从我最不喜欢的开始到最喜欢的:
使用@SuppressWarnings("unchecked")注释。这并不能真正解决问题,但你将不再收到警告。
@SuppressWarnings("unchecked")
Foo mockFoo = mock(Foo.class);
when(mockFoo.getValue).thenReturn(new Bar());
把它。不幸的是,它仍然给出了警告。所以你也需要在这里使用注释:
@SuppressWarnings("unchecked")
Foo<Bar> mockFoo = (Foo<Bar>) mock(Foo.class);
when(mockFoo.getValue).thenReturn(new Bar());
使用@Mock注释。不会有任何警告。这里,什么时候可以加入到实际测试中。
@Mock
public Foo<Bar> fooMock;
使用@MockBean注释。这将直接创建一个模拟bean。没有警告。
@MockBean
public Foo<Bar> fooMock;
为什么不用间谍
var mock = spy(new Foo<Bar>());
when(mockFoo.getValue()).thenReturn(new Bar());
(在我看来)最简单和最易读的方法是使用方法级注入。
这将导致在测试方法中拥有所有测试数据。这将使您的测试类保持干净,因为没有“浮动”mock。
@ExtendWith(MockitoExtension.class)
public class SomeClassTest {
@Test
void someTestMethod(@Mock Foo<Bar> fooMock) {
// do something with your mock
}
}