如何告诉一个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.

    }

}

在第二个测试中，当testObj.bar()被调用时，我仍然收到0作为值…解决这个问题的最好方法是什么?请注意，我知道我可以在每个测试中使用不同的Foo模拟，但是，我必须从mockFoo中链接多个请求，这意味着我必须在每个测试中进行链接。

当前回答

你也可以Stub连续的调用(2.8.9 api中的#10)。在这种情况下，你可以使用多个thenReturn调用，或者一个thenReturn调用和多个参数(可变参数)。

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
    }
}

2011-05-19 21:32:59

其他回答

你也可以Stub连续的调用(2.8.9 api中的#10)。在这种情况下，你可以使用多个thenReturn调用，或者一个thenReturn调用和多个参数(可变参数)。

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
    }
}

2011-05-19 21:32:59

或者，更简洁:

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

2018-10-24 11:23:46

首先，不要让模拟是静态的。将其设置为私有字段。只要把你的setUp类放在@Before而不是@BeforeClass。它可能会运行很多次，但很便宜。

其次，您现在拥有它的方式是让mock根据测试返回不同内容的正确方法。

2010-11-18 15:45:52

对于所有搜索返回一些东西，然后另一个调用抛出异常:

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

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

2018-05-24 12:09:04

对于使用spy()和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);

2018-03-07 16:41:43

如何告诉一个Mockito模拟对象返回不同的东西，下次它被调用?

推荐文章

最新文章

标签