Використання Mockito для макетування класів із загальними параметрами

Чи існує чистий метод глузування з класом із загальними параметрами? Скажіть, я маю знущатися над класом, Foo<T>який мені потрібно передати в метод, який очікує на a Foo<Bar>. Я можу зробити наступне досить легко:

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

Припускаючи, що getValue()повертає загальний тип T. Але це матимуть кошенят, коли я згодом передам це в очікуванні Foo<Bar>. Чи є кастинг єдиним засобом для цього?

Я думаю, що вам потрібно це зробити, але це не повинно бути дуже погано:

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

Ще один спосіб цього - @Mockзамість цього використовувати анотацію. Працює не у всіх випадках, але виглядає набагато сексуальніше :)

Ось приклад:

@RunWith(MockitoJUnitRunner.class)
public class FooTests {

    @Mock
    public Foo<Bar> fooMock;

    @Test
    public void testFoo() {
        when(fooMock.getValue()).thenReturn(new Bar());
    }
}

MockitoJUnitRunnerІніціалізує поля анотовані з @Mock.

Ви завжди можете створити проміжний клас / інтерфейс, який би задовольняв загальний тип, який ви хочете вказати. Наприклад, якщо Foo був інтерфейсом, ви можете створити наступний інтерфейс у своєму тестовому класі.

private interface FooBar extends Foo<Bar>
{
}

У ситуаціях, коли Foo - це не остаточний клас, ви можете просто розширити клас наступним кодом і зробити те саме:

public class FooBar extends Foo<Bar>
{
}

Тоді ви можете споживати будь-який із наведених вище прикладів із наступним кодом:

Foo<Bar> mockFoo = mock(FooBar.class);
when(mockFoo.getValue()).thenReturn(new 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);
}

Я погоджуюся, що не слід придушувати попередження в класах чи методах, оскільки можна було не помітити інших, випадково придушених попереджень. Але IMHO абсолютно розумно придушити попередження, яке стосується лише одного рядка коду.

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

Ось цікавий випадок: метод отримує загальну колекцію та повертає загальну колекцію того ж базового типу. Наприклад:

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

Цей метод можна знущатись, поєднуючи Mockito anyCollectionOf matcher та 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>();
         }
     });