Mockito when same parameter with different result?

2019-12-16 17:38发布

问题:

I have a builder class that has build() function, I'm passing a service for it.

I have a test class for the builder which has two methods

Is there a way to test the same with the same parameter twice but returning a different value?

@Before
public void initTest() {
    MockitoAnnotations.initMocks(this);
}

@Test
private void test1{
    Mockito.when(defaultBuilder.build(service)).thenReturn(createObject());
}

@Test
private void test2(){  
    Mockito.when(defaultBuilder.build(service)).thenReturn(createDifferentObject());
}   

The problem is I get the createObject() object from second test, even though I'm expecting createDifferentObject().

回答1:

    public class Test {

    private MyObject myobject;
    private MyObject myDifferentObject;

    @Before
    public void initTest() {
        MockitoAnnotations.initMocks(this);
    myobject=createObject();
    myDifferentObject=createDifferentObject();
    }

    @Test
    private void test1{
        Mockito.when(defaultBuilder.build(service)).thenReturn(myobject);
    }

    @Test
    private void test2(){  
        Mockito.when(defaultBuilder.build(service)).thenReturn(myDifferentObject);
    }     

    }

//should work.


回答2:

You can reset your mock before setting second value:

https://static.javadoc.io/org.mockito/mockito-core/2.25.1/org/mockito/Mockito.html#resetting_mocks

However, it is regarded as a sign of poor test concept.