จะบอกวัตถุจำลอง Mockito ให้คืนสิ่งที่แตกต่างในครั้งต่อไปที่เรียกได้อย่างไร

ดังนั้นฉันกำลังสร้างวัตถุจำลองเป็นตัวแปรแบบคงที่ในระดับชั้นเช่นนี้ ... ในการทดสอบหนึ่งฉันต้องการFoo.someMethod()คืนค่าบางค่าในขณะที่การทดสอบอื่นฉันต้องการให้คืนค่าที่แตกต่างกัน ปัญหาที่ฉันมีคือดูเหมือนว่าฉันต้องสร้าง mocks ขึ้นใหม่เพื่อให้ทำงานได้อย่างถูกต้อง ฉันต้องการหลีกเลี่ยงการสร้าง mocks ขึ้นใหม่และใช้วัตถุเดียวกันในการทดสอบแต่ละครั้ง

class TestClass {

    private static Foo mockFoo;

    @BeforeClass
    public static void setUp() {
        mockFoo = mock(Foo.class);
    }

    @Test
    public void test1() {
        when(mockFoo.someMethod()).thenReturn(0);

        TestObject testObj = new TestObject(mockFoo);

        testObj.bar(); // calls mockFoo.someMethod(), receiving 0 as the value

    }

    @Test
    public void test2() {
        when(mockFoo.someMethod()).thenReturn(1);

        TestObject testObj = new TestObject(mockFoo);

        testObj.bar(); // calls mockFoo.someMethod(), STILL receiving 0 as the value, instead of expected 1.

    }

}

ในการทดสอบครั้งที่สองฉันยังคงได้รับ 0 เป็นค่าเมื่อเรียกว่า testObj.bar () ... วิธีที่ดีที่สุดในการแก้ไขปัญหานี้คืออะไร โปรดทราบว่าฉันรู้ว่าฉันสามารถใช้การจำลองที่แตกต่างกันFooในการทดสอบแต่ละครั้งอย่างไรก็ตามฉันต้องmockFooเชื่อมโยงคำขอจำนวนมากออกไปซึ่งหมายความว่าฉันต้องทำการผูกมัดในการทดสอบแต่ละครั้ง

ประการแรกไม่ได้ทำให้จำลองคงที่ ทำให้เป็นเขตข้อมูลส่วนตัว เพียงแค่ใส่ชั้นตั้งค่าของคุณในไม่ได้@Before @BeforeClassอาจเป็นพวง แต่ราคาถูก

ประการที่สองวิธีที่คุณมีในขณะนี้คือวิธีที่ถูกต้องในการจำลองเพื่อคืนสิ่งต่าง ๆ โดยขึ้นอยู่กับการทดสอบ

คุณสามารถStub การโทรติดต่อกัน (# 10 in 2.8.9 api) ในกรณีนี้คุณจะใช้การเรียกแบบส่งคืนแล้วหลายครั้งหรือหนึ่งการโทรกลับแล้วหนึ่งครั้งพร้อมกับหลายพารามิเตอร์ (varargs)

import static org.junit.Assert.assertEquals;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import org.junit.Before;
import org.junit.Test;

public class TestClass {

    private Foo mockFoo;

    @Before
    public void setup() {
        setupFoo();
    }

    @Test
    public void testFoo() {
        TestObject testObj = new TestObject(mockFoo);

        assertEquals(0, testObj.bar());
        assertEquals(1, testObj.bar());
        assertEquals(-1, testObj.bar());
        assertEquals(-1, testObj.bar());
    }

    private void setupFoo() {
        mockFoo = mock(Foo.class);

        when(mockFoo.someMethod())
            .thenReturn(0)
            .thenReturn(1)
            .thenReturn(-1); //any subsequent call will return -1

        // Or a bit shorter with varargs:
        when(mockFoo.someMethod())
            .thenReturn(0, 1, -1); //any subsequent call will return -1
    }
}

สำหรับทุกคนที่ค้นหาเพื่อส่งคืนบางสิ่งจากนั้นจึงมีข้อยกเว้นการโทรอีกครั้ง:

    when(mockFoo.someMethod())
            .thenReturn(obj1)
            .thenReturn(obj2)
            .thenThrow(new RuntimeException("Fail"));

หรือ

    when(mockFoo.someMethod())
            .thenReturn(obj1, obj2)
            .thenThrow(new RuntimeException("Fail"));

หรือแม้แต่ทำความสะอาด:

when(mockFoo.someMethod()).thenReturn(obj1, obj2);

สำหรับทุกคนที่ใช้สอดแนม ()และ doReturn () แทนเมธอด when ():

สิ่งที่คุณต้องการส่งคืนวัตถุที่แตกต่างกันในการโทรที่แตกต่างกันคือ:

doReturn(obj1).doReturn(obj2).when(this.spyFoo).someMethod();

.

สำหรับ mocks คลาสสิก:

when(this.mockFoo.someMethod()).thenReturn(obj1, obj2);

หรือมีข้อยกเว้นถูกโยน:

when(mockFoo.someMethod())
        .thenReturn(obj1)
        .thenThrow(new IllegalArgumentException())
        .thenReturn(obj2, obj3);