การใช้ Mockito เพื่อจำลองคลาสด้วยพารามิเตอร์ทั่วไป

มีวิธีทำความสะอาดแบบจำลองคลาสด้วยพารามิเตอร์ทั่วไปหรือไม่? ว่าฉันมีจะเยาะเย้ยชั้นเรียนซึ่งผมต้องผ่านเข้าไปในวิธีการที่คาดว่าจะเป็นFoo<T> 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 และคำตอบ

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