Як сказати макетному об'єкту Mockito повернути щось інше наступного разу, коли він буде викликаний?

Отже, я створюю об'єкт макету як статичну змінну на рівні класу так… В одному тесті я хочу Foo.someMethod()повернути певне значення, а в іншому тесті - я хочу повернути інше значення. Проблема, яка у мене виникає, полягає в тому, що, здається, мені потрібно відновити макети, щоб змусити це правильно працювати. Я хотів би уникнути відновлення макетів і просто використовувати ті самі об’єкти в кожному тесті.

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, тобто я повинен був би робити ланцюжок у кожному тесті.

Перш за все не робіть макет статичним. Зробіть це приватним полем. Просто покладіть свій клас setUp в @Beforenot @BeforeClass. Це може бути запущена купа, але це дешево.

По-друге, спосіб, який ви зараз маєте, - це правильний спосіб змусити повернути щось інше залежно від тесту.

Ви також можете зупинити послідовні дзвінки (№10 в 2.8.9 api). У цьому випадку ви б використовували кілька викликів thenReturn або один callReturn з декількома параметрами (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();

.

Для класичних макетів:

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

або за винятком викидання:

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