Mockito: doAnswer Vs thenReturn

ฉันใช้ Mockito สำหรับการทดสอบหน่วยบริการในภายหลัง ฉันกำลังสับสนเมื่อจะใช้VSdoAnswerthenReturn

ใครสามารถช่วยฉันโดยละเอียด? จนถึงตอนนี้ฉันได้ลองใช้thenReturnแล้ว

คุณควรใช้thenReturnหรือdoReturnเมื่อคุณทราบค่าส่งคืนในเวลาที่คุณเยาะเย้ยการเรียกใช้เมธอด ค่าที่กำหนดนี้จะถูกส่งกลับเมื่อคุณเรียกใช้เมธอดจำลอง

thenReturn(T value) ตั้งค่าการส่งคืนที่จะส่งคืนเมื่อมีการเรียกใช้เมธอด

@Test
public void test_return() throws Exception {
    Dummy dummy = mock(Dummy.class);
    int returnValue = 5;

    // choose your preferred way
    when(dummy.stringLength("dummy")).thenReturn(returnValue);
    doReturn(returnValue).when(dummy).stringLength("dummy");
}

Answer ถูกใช้เมื่อคุณต้องการดำเนินการเพิ่มเติมเมื่อมีการเรียกใช้เมธอดจำลองเช่นเมื่อคุณต้องการคำนวณค่าที่ส่งคืนตามพารามิเตอร์ของการเรียกใช้เมธอดนี้

ใช้เมื่อคุณต้องการต้นขั้ววิธีโมฆะกับทั่วไปdoAnswer()Answer

คำตอบระบุการดำเนินการที่ดำเนินการและค่าส่งคืนที่ส่งคืนเมื่อคุณโต้ตอบกับการจำลอง

@Test
public void test_answer() throws Exception {
    Dummy dummy = mock(Dummy.class);
    Answer<Integer> answer = new Answer<Integer>() {
        public Integer answer(InvocationOnMock invocation) throws Throwable {
            String string = invocation.getArgumentAt(0, String.class);
            return string.length() * 2;
        }
    };

    // choose your preferred way
    when(dummy.stringLength("dummy")).thenAnswer(answer);
    doAnswer(answer).when(dummy).stringLength("dummy");
}

doAnswerและthenReturnทำสิ่งเดียวกันหาก:

1. คุณกำลังใช้ Mock ไม่ใช่ Spy
2. เมธอดที่คุณกำลังขีดฆ่ากำลังส่งคืนค่าไม่ใช่วิธีโมฆะ

มาล้อเลียน BookService นี้

public interface BookService {
    String getAuthor();
    void queryBookTitle(BookServiceCallback callback);
}

คุณสามารถต้นขั้ว getAuthor () โดยใช้และdoAnswerthenReturn

BookService service = mock(BookService.class);
when(service.getAuthor()).thenReturn("Joshua");
// or..
doAnswer(new Answer() {
    @Override
    public Object answer(InvocationOnMock invocation) throws Throwable {
        return "Joshua";
    }
}).when(service).getAuthor();

โปรดทราบว่าเมื่อใช้doAnswerคุณจะไม่สามารถส่งต่อวิธีการwhenได้

// Will throw UnfinishedStubbingException
doAnswer(invocation -> "Joshua").when(service.getAuthor());

แล้วคุณจะใช้doAnswerแทนเมื่อthenReturnไหร่? ฉันนึกถึงการใช้งานสองกรณี:

1. เมื่อคุณต้องการ "ต้นขั้ว" วิธีโมฆะ

การใช้ doAnswer คุณสามารถดำเนินการเพิ่มเติมบางอย่างเมื่อเรียกใช้เมธอด ตัวอย่างเช่นทริกเกอร์การเรียกกลับบน queryBookTitle

BookServiceCallback callback = new BookServiceCallback() {
    @Override
    public void onSuccess(String bookTitle) {
        assertEquals("Effective Java", bookTitle);
    }
};
doAnswer(new Answer() {
    @Override
    public Object answer(InvocationOnMock invocation) throws Throwable {
        BookServiceCallback callback = (BookServiceCallback) invocation.getArguments()[0];
        callback.onSuccess("Effective Java");
        // return null because queryBookTitle is void
        return null;
    }
}).when(service).queryBookTitle(callback);
service.queryBookTitle(callback);

2. เมื่อคุณใช้ Spy แทน Mock

เมื่อใช้ when-thenReturn on Spy Mockito จะเรียกวิธีการจริงจากนั้นจะทำให้คำตอบของคุณสะดุด สิ่งนี้อาจทำให้เกิดปัญหาได้หากคุณไม่ต้องการเรียกใช้วิธีจริงเช่นในตัวอย่างนี้:

List list = new LinkedList();
List spy = spy(list);
// Will throw java.lang.IndexOutOfBoundsException: Index: 0, Size: 0
when(spy.get(0)).thenReturn("java");
assertEquals("java", spy.get(0));

การใช้ doAnswer ทำให้เราสามารถตัดมันได้อย่างปลอดภัย

List list = new LinkedList();
List spy = spy(list);
doAnswer(invocation -> "java").when(spy).get(0);
assertEquals("java", spy.get(0));

doReturnที่จริงถ้าคุณไม่ต้องการที่จะทำดำเนินการเพิ่มเติมเมื่อวิธีการอุทธรณ์คุณก็สามารถใช้

List list = new LinkedList();
List spy = spy(list);
doReturn("java").when(spy).get(0);
assertEquals("java", spy.get(0));