Mockito.when是一種Mockito框架中的方法,它可以用于模擬返回值和異常。該方法接受一個方法調(diào)用作為參數(shù),然后可以使用鏈式調(diào)用來模擬方法的返回值或拋出一個異常。
一、Mockito.when的基本語法
Mockito.when的語法格式如下:
when(mock.method()).thenReturn(value);
// or
when(mock.method()).thenThrow(exception);
在這里,mock是被模擬的對象,method是被模擬的方法,value是模擬的返回值,exception是模擬的異常。
使用Mockito.when時,必須先創(chuàng)建一個mock對象,并指定它模擬哪個類或接口。例如:
SomeInterface mock = Mockito.mock(SomeInterface.class);
這將創(chuàng)建一個模擬SomeInterface接口的mock對象。然后,就可以使用Mockito.when來模擬該接口的方法了。
二、Mockito.when的鏈式調(diào)用
使用Mockito.when的另一個強大功能是鏈式調(diào)用。鏈式調(diào)用允許我們按方法的順序指定模擬的返回值或異常。
下面是一個簡單的例子,它演示了如何使用Mockito.when和鏈式調(diào)用來模擬一個方法的返回值:
when(mock.getValue()).thenReturn("hello").thenReturn("world");
String first = mock.getValue(); // returns "hello"
String second = mock.getValue(); // returns "world"
在這個例子中,我們使用了兩個thenReturn方法來指定兩個不同的返回值。我們首先指定返回值為"hello",然后指定返回值為"world"。
當我們使用mock.getValue()方法時,Mockito會注意到我們使用了兩個thenReturn方法,并依次返回它們。第一次調(diào)用時,返回值為"hello",第二次調(diào)用時,返回值為"world"。
三、Mockito.when的參數(shù)匹配器
Mockito.when還提供了參數(shù)匹配器的功能。參數(shù)匹配器允許您指定對于某些參數(shù),Mockito應(yīng)該如何模擬方法的行為。
下面是一個例子,它演示了如何使用參數(shù)匹配器:
when(mock.someMethod(anyInt(), eq("test"))).thenReturn("result");
String result = mock.someMethod(1, "test"); // returns "result"
在這個例子中,我們指定了一個參數(shù)匹配器anyInt()和一個參數(shù)匹配器eq("test")。anyInt()匹配任何整數(shù)值,eq("test")匹配"test"字符串。
當我們調(diào)用mock.someMethod(1, "test")方法時,Mockito會首先檢查第一個參數(shù)是否為任何整數(shù)值(由anyInt() 匹配),然后檢查第二個參數(shù)是否等于"test"字符串(由eq("test")匹配),如果兩個參數(shù)都匹配,則返回"result"字符串。
四、Mockito.when的異常模擬
Mockito.when還可以用于模擬方法拋出異常的情況。你可以使用Mockito.when和thenThrow方法來模擬異常:
when(mock.someMethod()).thenThrow(new Exception());
mock.someMethod(); // throws Exception
在這個例子中,我們使用Mockito.when和thenThrow方法來模擬模擬方法拋出異常。在我們調(diào)用mock.someMethod()方法的時候,它將拋出一個Exception異常。
五、總結(jié)
在本文中,我們學習了Mockito.when的基本語法、鏈式調(diào)用、參數(shù)匹配器和異常模擬。Mockito.when是Mockito框架中最常用的方法之一,它允許我們使用簡單、有效的語法模擬方法的行為。
下面是學習Mockito.when的完整示例代碼:
public class MockExample {
public static void main(String[] args) {
// 創(chuàng)建Mock對象
SomeInterface mock = Mockito.mock(SomeInterface.class);
// 模擬返回值
when(mock.getValue()).thenReturn("hello").thenReturn("world");
String first = mock.getValue(); // returns "hello"
String second = mock.getValue(); // returns "world"
// 參數(shù)匹配器
when(mock.someMethod(anyInt(), eq("test"))).thenReturn("result");
String result = mock.someMethod(1, "test"); // returns "result"
// 異常模擬
when(mock.someMethod()).thenThrow(new Exception());
try {
mock.someMethod(); // throws Exception
} catch (Exception e) {
// Handle Exception
}
}
}
public interface SomeInterface {
String getValue();
String someMethod(int num, String str) throws Exception;
}