該系列文章翻譯自https://www.baeldung.com/mockito-series

接下來(lái)我們將以MyList類為例進(jìn)行介紹

public class MyList extends AbstractList<String> {
 
    @Override
    public String get(final int index) {
        return null;
    }
    @Override
    public int size() {
        return 1;
    }
}

When/Then常見用法常見用法

1.方法一：when().thenReturn()模擬方法的返回

MyList listMock = Mockito.mock(MyList.class);
when(listMock.add(anyString())).thenReturn(false);
 
boolean added = listMock.add(randomAlphabetic(6));
assertThat(added, is(false));

2.方法二：doReturn().when()模擬方法的返回

MyList listMock = Mockito.mock(MyList.class);
doReturn(false).when(listMock).add(anyString());
 
boolean added = listMock.add(randomAlphabetic(6));
assertThat(added, is(false));

3.when().thenThrow()模擬異常(方法返回類型非void)

@Test(expected = IllegalStateException.class)
public void givenMethodIsConfiguredToThrowException_whenCallingMethod_thenExceptionIsThrown() {
    MyList listMock = Mockito.mock(MyList.class);
    when(listMock.add(anyString())).thenThrow(IllegalStateException.class);
 
    listMock.add(randomAlphabetic(6));
}

4.doThrow().when()模擬異常(方法返回類型為void)

MyList listMock = Mockito.mock(MyList.class);
doThrow(NullPointerException.class).when(listMock).clear();
 
listMock.clear();

5.模擬方法的多次調(diào)用

在下面的例子中，第二次調(diào)用add方法會(huì)拋出IllegalStateException

MyList listMock = Mockito.mock(MyList.class);
when(listMock.add(anyString()))
  .thenReturn(false)
  .thenThrow(IllegalStateException.class);
 
listMock.add(randomAlphabetic(6));
listMock.add(randomAlphabetic(6)); // will throw the exception

6.spy對(duì)象

MyList instance = new MyList();
MyList spy = Mockito.spy(instance);
 
doThrow(NullPointerException.class).when(spy).size();
spy.size(); // will throw the exception

7.使用thenCallRealMethod()調(diào)用mock對(duì)象的真實(shí)方法

MyList listMock = Mockito.mock(MyList.class);
when(listMock.size()).thenCallRealMethod();
 
assertThat(listMock.size(), equalTo(1));

8.doAnswer().when()設(shè)置默認(rèn)返回

MyList listMock = Mockito.mock(MyList.class);
doAnswer(invocation -> "Always the same").when(listMock).get(anyInt());
 
String element = listMock.get(1);
assertThat(element, is(equalTo("Always the same")));