Mock 方法是單元測試中常見的一種技術,它的主要作用是模擬一些在應用中不容易構造或者比較復雜的對象,從而把測試與測試邊界以外的對象隔離開。
編寫自定義的 Mock 對象需要額外的編碼工作,同時也可能引入錯誤。EasyMock 提供了根據指定接口動態構建 Mock 對象的方法,避免了手工編寫 Mock 對象。本文將向您展示如何使用 EasyMock 進行單元測試,并對 EasyMock 的原理進行分析。
單元測試是對應用中的某一個模塊的功能進行驗證。在單元測試中,我們常遇到的問題是應用中其它的協同模塊尚未開發完成,或 者被測試模塊需要和一些不容易構造、比較復雜的對象進行交互。另外,由于不能肯定其它模塊的正確性,我們也無法確定測試中發現的問題是由哪個模塊引起的。
Mock 對象能夠模擬其它協同模塊的行為,被測試模塊通過與 Mock 對象協作,可以獲得一個孤立的測試環境。此外,使用 Mock 對象還可以模擬在應用中不容易構造(如 HttpServletRequest 必須在 Servlet 容器中才能構造出來)和比較復雜的對象(如 JDBC 中的 ResultSet 對象),從而使測試順利進行。
手動的構造 Mock 對象會給開發人員帶來額外的編碼量,而且這些為創建 Mock 對象而編寫的代碼很有可能引入錯誤。目前,有許多開源項目對動態構建 Mock 對象提供了支持,這些項目能夠根據現有的接口或類動態生成,這樣不僅能避免額外的編碼工作,同時也降低了引入錯誤的可能。
EasyMock 是一套用于通過簡單的方法對于給定的接口生成 Mock 對象的類庫。它提供對接口的模擬,能夠通過錄制、回放、檢查三步來完成大體的測試過程,可以驗證方法的調用種類、次數、順序,可以令 Mock 對象返回指定的值或拋出指定異常。通過 EasyMock,我們可以方便的構造 Mock 對象從而使單元測試順利進行。
EasyMock 是采用 MIT license 的一個開源項目,您可以在 Sourceforge 上下載到相關的 zip 文件。目前您可以下載的 EasyMock 最新版本是2.3,它需要運行在 Java 5.0 平臺上。如果您的應用運行在 Java 1.3 或 1.4 平臺上,您可以選擇 EasyMock1.2。在解壓縮 zip 包后,您可以找到 easymock.jar 這個文件。如果您使用 Eclipse 作為 IDE,把 easymock.jar 添加到項目的 Libraries 里就可以使用了(如下圖所示)。此外,由于我們的測試用例運行在 JUnit 環境中,因此您還需要 JUnit.jar(版本3.8.1以上)。
通過 EasyMock,我們可以為指定的接口動態的創建 Mock 對象,并利用 Mock 對象來模擬協同模塊或是領域對象,從而使單元測試順利進行。這個過程大致可以劃分為以下幾個步驟:
- 使用 EasyMock 生成 Mock 對象;
- 設定 Mock 對象的預期行為和輸出;
- 將 Mock 對象切換到 Replay 狀態;
- 調用 Mock 對象方法進行單元測試;
- 對 Mock 對象的行為進行驗證。
接下來,我們將對以上的幾個步驟逐一進行說明。除了以上的基本步驟外,EasyMock 還對特殊的 Mock 對象類型、特定的參數匹配方式等功能提供了支持,我們將在之后的章節中進行說明。
根據指定的接口或類,EasyMock 能夠動態的創建 Mock 對象(EasyMock 默認只支持為接口生成 Mock 對象,如果需要為類生成 Mock 對象,在 EasyMock 的主頁上有擴展包可以實現此功能),我們以
ResultSet
接口為例說明EasyMock的功能。
java.sql.ResultSet
是每一個 Java 開發人員都非常熟悉的接口:
public interface java.sql.ResultSet { ...... public abstract java.lang.String getString(int arg0) throws java.sql.SQLException; public abstract double getDouble(int arg0) throws java.sql.SQLException; ...... } |
?
通常,構建一個真實的
RecordSet
對象需要經過一個復雜的過程:在開發過程中,開發人員通常會編寫一個
DBUtility
類來獲取數據庫連接
Connection
,并利用
Connection
創建一個
Statement
。執行一個
Statement
可以獲取到一個或多個
ResultSet
對象。這樣的構造過程復雜并且依賴于數據庫的正確運行。數據庫或是數據庫交互模塊出現問題,都會影響單元測試的結果。
我們可以使用 EasyMock 動態構建
ResultSet
接口的 Mock 對象來解決這個問題。一些簡單的測試用例只需要一個 Mock 對象,這時,我們可以用以下的方法來創建 Mock 對象:
ResultSet mockResultSet = createMock(ResultSet.class); |
?
其中
createMock
是
org.easymock.EasyMock
類所提供的靜態方法,你可以通過 static import 將其引入(注:static import 是 java 5.0 所提供的新特性)。
如果需要在相對復雜的測試用例中使用多個 Mock 對象,EasyMock 提供了另外一種生成和管理 Mock 對象的機制:
IMocksControl control = EasyMock.createControl(); java.sql.Connection mockConnection = control.createMock(Connection.class); java.sql.Statement mockStatement = control.createMock(Statement.class); java.sql.ResultSet mockResultSet = control.createMock(ResultSet.class); |
?
EasyMock
類的
createControl
方法能創建一個接口
IMocksControl
的對象,該對象能創建并管理多個 Mock 對象。如果需要在測試中使用多個 Mock 對象,我們推薦您使用這一機制,因為它在多個 Mock 對象的管理上提供了相對便捷的方法。
如果您要模擬的是一個具體類而非接口,那么您需要下載擴展包 EasyMock Class Extension 2.2.2。在對具體類進行模擬時,您只要用
org.easymock.classextension.EasyMock
類中的靜態方法代替
org.easymock.EasyMock
類中的靜態方法即可。
在一個完整的測試過程中,一個 Mock 對象將會經歷兩個狀態:Record 狀態和 Replay 狀態。Mock 對象一經創建,它的狀態就被置為 Record。在 Record 狀態,用戶可以設定 Mock 對象的預期行為和輸出,這些對象行為被錄制下來,保存在 Mock 對象中。
添加 Mock 對象行為的過程通常可以分為以下3步:
- 對 Mock 對象的特定方法作出調用;
-
通過
org.easymock.EasyMock
提供的靜態方法expectLastCall
獲取上一次方法調用所對應的 IExpectationSetters 實例; -
通過
IExpectationSetters
實例設定 Mock 對象的預期輸出。
設定預期返回值
Mock 對象的行為可以簡單的理解為 Mock 對象方法的調用和方法調用所產生的輸出。在 EasyMock 2.3 中,對 Mock 對象行為的添加和設置是通過接口
IExpectationSetters
來實現的。Mock 對象方法的調用可能產生兩種類型的輸出:(1)產生返回值;(2)拋出異常。接口
IExpectationSetters
提供了多種設定預期輸出的方法,其中和設定返回值相對應的是 andReturn 方法:
IExpectationSetters<T> andReturn(T value); |
?
我們仍然用
ResultSet
接口的 Mock 對象為例,如果希望方法
mockResult.getString(1)
的返回值為 "My return value",那么你可以使用以下的語句:
mockResultSet.getString(1); expectLastCall().andReturn("My return value"); |
?
以上的語句表示
mockResultSet
的
getString
方法被調用一次,這次調用的返回值是 "My return value"。有時,我們希望某個方法的調用總是返回一個相同的值,為了避免每次調用都為 Mock 對象的行為進行一次設定,我們可以用設置默認返回值的方法:
void andStubReturn(Object value); |
?
假設我們創建了
Statement
和
ResultSet
接口的 Mock 對象 mockStatement 和 mockResultSet,在測試過程中,我們希望 mockStatement 對象的
executeQuery
方法總是返回 mockResultSet,我們可以使用如下的語句
mockStatement.executeQuery("SELECT * FROM sales_order_table"); expectLastCall().andStubReturn(mockResultSet); |
?
EasyMock 在對參數值進行匹配時,默認采用
Object.equals()
方法。因此,如果我們以
"select * from sales_order_table"
作為參數,預期方法將不會被調用。如果您希望上例中的 SQL 語句能不區分大小寫,可以用特殊的參數匹配器來解決這個問題,我們將在 "在 EasyMock 中使用參數匹配器" 一章對此進行說明。
設定預期異常拋出
對象行為的預期輸出除了可能是返回值外,還有可能是拋出異常。
IExpectationSetters
提供了設定預期拋出異常的方法:
IExpectationSetters<T> andThrow(Throwable throwable); |
?
和設定默認返回值類似,
IExpectationSetters
接口也提供了設定拋出默認異常的函數:
void andStubThrow(Throwable throwable); |
?
設定預期方法調用次數
通過以上的函數,您可以對 Mock 對象特定行為的預期輸出進行設定。除了對預期輸出進行設定,
IExpectationSetters
接口還允許用戶對方法的調用次數作出限制。在
IExpectationSetters
所提供的這一類方法中,常用的一種是
times
方法:
IExpectationSetters<T>times(int count); |
?
該方法可以 Mock 對象方法的調用次數進行確切的設定。假設我們希望 mockResultSet 的
getString
方法在測試過程中被調用3次,期間的返回值都是 "My return value",我們可以用如下語句:
mockResultSet.getString(1); expectLastCall().andReturn("My return value").times(3); |
?
注意到
andReturn
和
andThrow
方法的返回值依然是一個
IExpectationSetters
實例,因此我們可以在此基礎上繼續調用
times
方法。
除了設定確定的調用次數,
IExpectationSetters
還提供了另外幾種設定非準確調用次數的方法:
times(int minTimes, int maxTimes)
:該方法最少被調用 minTimes 次,最多被調用 maxTimes 次。
atLeastOnce()
:該方法至少被調用一次。
anyTimes()
:該方法可以被調用任意次。
某些方法的返回值類型是 void,對于這一類方法,我們無需設定返回值,只要設置調用次數就可以了。以
ResultSet
接口的
close
方法為例,假設在測試過程中,該方法被調用3至5次:
mockResultSet.close(); expectLastCall().times(3, 5); |
?
為了簡化書寫,EasyMock 還提供了另一種設定 Mock 對象行為的語句模式。對于上例,您還可以將它寫成:
expect(mockResult.close()).times(3, 5); |
?
這個語句和上例中的語句功能是完全相同的。
在生成 Mock 對象和設定 Mock 對象行為兩個階段,Mock 對象的狀態都是 Record 。在這個階段,Mock 對象會記錄用戶對預期行為和輸出的設定。
在使用 Mock 對象進行實際的測試前,我們需要將 Mock 對象的狀態切換為 Replay。在 Replay 狀態,Mock 對象能夠根據設定對特定的方法調用作出預期的響應。將 Mock 對象切換成 Replay 狀態有兩種方式,您需要根據 Mock 對象的生成方式進行選擇。如果 Mock 對象是通過
org.easymock.EasyMock
類提供的靜態方法 createMock 生成的(第1節中介紹的第一種 Mock 對象生成方法),那么
EasyMock
類提供了相應的 replay 方法用于將 Mock 對象切換為 Replay 狀態:
replay(mockResultSet); |
?
如果 Mock 對象是通過
IMocksControl
接口提供的
createMock
方法生成的(第1節中介紹的第二種Mock對象生成方法),那么您依舊可以通過
IMocksControl
接口對它所創建的所有 Mock 對象進行切換:
control.replay(); |
?
以上的語句能將在第1節中生成的 mockConnection、mockStatement 和 mockResultSet 等3個 Mock 對象都切換成 Replay 狀態。
為了更好的說明 EasyMock 的功能,我們引入 src.zip 中的示例來解釋 Mock 對象在實際測試階段的作用。其中所有的示例代碼都可以在 src.zip 中找到。如果您使用的 IDE 是 Eclipse,在導入 src.zip 之后您可以看到 Workspace 中增加的 project(如下圖所示)。
下面是示例代碼中的一個接口
SalesOrder
,它的實現類
SalesOrderImpl
的主要功能是從數據庫中讀取一個 Sales Order 的 Region 和 Total Price,并根據讀取的數據計算該 Sales Order 的 Price Level(完整的實現代碼都可以在 src.zip 中找到):
public interface SalesOrder { …… public void loadDataFromDB(ResultSet resultSet) throws SQLException; public String getPriceLevel(); } |
?
其實現類
SalesOrderImpl
中對
loadDataFromDB
的實現如下:
public class SalesOrderImpl implements SalesOrder { ...... public void loadDataFromDB(ResultSet resultSet) throws SQLException { orderNumber = resultSet.getString(1); region = resultSet.getString(2); totalPrice = resultSet.getDouble(3); } ...... } |
?
方法
loadDataFromDB
讀取了
ResultSet
對象包含的數據。當我們將之前定義的 Mock 對象調整為 Replay 狀態,并將該對象作為參數傳入,那么 Mock 對象的方法將會返回預先定義的預期返回值。完整的 TestCase 如下:
public class SalesOrderTestCase extends TestCase { public void testSalesOrder() { IMocksControl control = EasyMock.createControl(); ...... ResultSet mockResultSet = control.createMock(ResultSet.class); try { ...... mockResultSet.next(); expectLastCall().andReturn(true).times(3); expectLastCall().andReturn(false).times(1); mockResultSet.getString(1); expectLastCall().andReturn("DEMO_ORDER_001").times(1); expectLastCall().andReturn("DEMO_ORDER_002").times(1); expectLastCall().andReturn("DEMO_ORDER_003").times(1); mockResultSet.getString(2); expectLastCall().andReturn("Asia Pacific").times(1); expectLastCall().andReturn("Europe").times(1); expectLastCall().andReturn("America").times(1); mockResultSet.getDouble(3); expectLastCall().andReturn(350.0).times(1); expectLastCall().andReturn(1350.0).times(1); expectLastCall().andReturn(5350.0).times(1); control.replay(); ...... int i = 0; String[] priceLevels = { "Level_A", "Level_C", "Level_E" }; while (mockResultSet.next()) { SalesOrder order = new SalesOrderImpl(); order.loadDataFromDB(mockResultSet); assertEquals(order.getPriceLevel(), priceLevels[i]); i++; } control.verify(); } catch (Exception e) { e.printStackTrace(); } } } |
?
在這個示例中,我們首先創建了
ResultSet
的 Mock 對象 moResultSet,并記錄該 Mock 對象的預期行為。之后我們調用了
control.replay()
,將 Mock 對象的狀態置為 Replay 狀態。 在實際的測試階段,Sales Order 對象的
loadDataFromDB
方法調用了 mockResultSet 對象的
getString
和
getDouble
方法讀取 mockResultSet 中的數據。Sales Order 對象根據讀取的數據計算出 Price Level,并和預期輸出進行比較。
在利用 Mock 對象進行實際的測試過程之后,我們還有一件事情沒有做:對 Mock 對象的方法調用的次數進行驗證。
為了驗證指定的方法調用真的完成了,我們需要調用
verify
方法進行驗證。和
replay
方法類似,您需要根據 Mock 對象的生成方式來選用不同的驗證方式。如果 Mock 對象是由
org.easymock.EasyMock
類提供的
createMock
靜態方法生成的,那么我們同樣采用
EasyMock
類的靜態方法
verify
進行驗證:
verify(mockResultSet); |
?
如果Mock對象是有
IMocksControl
接口所提供的
createMock
方法生成的,那么采用該接口提供的
verify
方法,例如第1節中的
IMocksControl
實例 control:
control.verify(); |
?
將對 control 實例所生成的 Mock 對象 mockConnection、mockStatement 和 mockResultSet 等進行驗證。如果將上例中
expectLastCall().andReturn(false).times(1)
的預期次數修改為2,在 Eclipse 中將可以看到:
為了避免生成過多的 Mock 對象,EasyMock 允許對原有 Mock 對象進行重用。要對 Mock 對象重新初始化,我們可以采用 reset 方法。和 replay 和 verify 方法類似,EasyMock 提供了兩種 reset 方式:(1)如果 Mock 對象是由
org.easymock.EasyMock
類中的靜態方法
createMock
生成的,那么該 Mock 對象的可以用
EasyMock
類的靜態方法
reset
重新初始化;(2)如果 Mock 方法是由
IMocksControl
實例的
createMock
方法生成的,那么該
IMocksControl
實例方法
reset
的調用將會把所有該實例創建的 Mock 對象重新初始化。
在重新初始化之后,Mock 對象的狀態將被置為 Record 狀態。
在使用 Mock 對象進行實際的測試過程中,EasyMock 會根據方法名和參數來匹配一個預期方法的調用。EasyMock 對參數的匹配默認使用
equals()
方法進行比較。這可能會引起一些問題。例如在上一章節中創建的mockStatement對象:
mockStatement.executeQuery("SELECT * FROM sales_order_table"); expectLastCall().andStubReturn(mockResultSet); |
?
在實際的調用中,我們可能會遇到 SQL 語句中某些關鍵字大小寫的問題,例如將 SELECT 寫成 Select,這時在實際的測試中,EasyMock 所采用的默認匹配器將認為這兩個參數不匹配,從而造成 Mock 對象的預期方法不被調用。EasyMock 提供了靈活的參數匹配方式來解決這個問題。如果您對 mockStatement 具體執行的語句并不關注,并希望所有輸入的字符串都能匹配這一方法調用,您可以用
org.easymock.EasyMock
類所提供的
anyObject
方法來代替參數中的 SQL 語句:
mockStatement.executeQuery( anyObject() ); expectLastCall().andStubReturn(mockResultSet); |
?
anyObject
方法表示任意輸入值都與預期值相匹配。除了
anyObject
以外,EasyMock還提供了多個預先定義的參數匹配器,其中比較常用的一些有:
-
aryEq(X value)
:通過Arrays.equals()
進行匹配,適用于數組對象; -
isNull()
:當輸入值為Null時匹配; -
notNull()
:當輸入值不為Null時匹配; -
same(X value)
:當輸入值和預期值是同一個對象時匹配; -
lt(X value), leq(X value), geq(X value), gt(X value)
:當輸入值小于、小等于、大等于、大于預期值時匹配,適用于數值類型; -
startsWith(String prefix), contains(String substring), endsWith(String suffix)
:當輸入值以預期值開頭、包含預期值、以預期值結尾時匹配,適用于String類型; -
matches(String regex)
:當輸入值與正則表達式匹配時匹配,適用于String類型。
預定義的參數匹配器可能無法滿足一些復雜的情況,這時你需要定義自己的參數匹配器。在上一節中,我們希望能有一個匹配器對 SQL 中關鍵字的大小寫不敏感,使用
anyObject
其實并不是一個好的選擇。對此,我們可以定義自己的參數匹配器 SQLEquals。
要定義新的參數匹配器,需要實現
org.easymock.IArgumentMatcher
接口。其中,
matches(Object actual)
方法應當實現輸入值和預期值的匹配邏輯,而在
appendTo(StringBuffer buffer)
方法中,你可以添加當匹配失敗時需要顯示的信息。以下是 SQLEquals 實現的部分代碼(完整的代碼可以在 src.zip 中找到):
public class SQLEquals implements IArgumentMatcher { private String expectedSQL = null; public SQLEquals(String expectedSQL) { this.expectedSQL = expectedSQL; } ...... public boolean matches(Object actualSQL) { if (actualSQL == null && expectedSQL == null) return true; else if (actualSQL instanceof String) return expectedSQL.equalsIgnoreCase((String) actualSQL); else return false; } } |
?
在實現了
IArgumentMatcher
接口之后,我們需要寫一個靜態方法將它包裝一下。這個靜態方法的實現需要將 SQLEquals 的一個對象通過
reportMatcher
方法報告給EasyMock:
public static String sqlEquals(String in) { reportMatcher(new SQLEquals(in)); return in; } |
?
這樣,我們自定義的 sqlEquals 匹配器就可以使用了。我們可以將上例中的
executeQuery
方法設定修改如下:
mockStatement.executeQuery(sqlEquals("SELECT * FROM sales_order_table")); expectLastCall().andStubReturn(mockResultSet); |
?
在使用
executeQuery("select * from sales_order_table")
進行方法調用時,該預期行為將被匹配。
到目前為止,我們所創建的 Mock 對象都屬于 EasyMock 默認的 Mock 對象類型,它對預期方法的調用次序不敏感,對非預期的方法調用拋出 AssertionError。除了這種默認的 Mock 類型以外,EasyMock 還提供了一些特殊的 Mock 類型用于支持不同的需求。
如果 Mock 對象是通過
EasyMock.createMock()
或是
IMocksControl.createMock()
所創建的,那么在進行 verify 驗證時,方法的調用順序是不進行檢查的。如果要創建方法調用的先后次序敏感的 Mock 對象(Strick Mock),應該使用
EasyMock.createStrickMock()
來創建,例如:
ResultSet strickMockResultSet = createStrickMock(ResultSet.class); |
?
類似于 createMock,我們同樣可以用
IMocksControl
實例來創建一個 Strick Mock 對象:
IMocksControl control = EasyMock.createStrictControl(); ResultSet strickMockResultSet = control.createMock(ResultSet.class); |
?
使用
createMock()
創建的 Mock 對象對非預期的方法調用默認的行為是拋出 AssertionError,如果需要一個默認返回0,null 或 false 等"無效值"的 "Nice Mock" 對象,可以通過
EasyMock
類提供的
createNiceMock()
方法創建。類似的,你也可以用
IMocksControl
實例來創建一個 Nice Mock 對象。
EasyMock 是如何為一個特定的接口動態創建 Mock 對象,并記錄 Mock 對象預期行為的呢?其實,EasyMock 后臺處理的主要原理是利用
java.lang.reflect.Proxy
為指定的接口創建一個動態代理,這個動態代理,就是我們在編碼中用到的 Mock 對象。EasyMock 還為這個動態代理提供了一個
InvocationHandler
接口的實現,這個實現類的主要功能就是將動態代理的預期行為記錄在某個映射表中和在實際調用時從這個映射表中取出預期輸出。下圖是 EasyMock 中主要的功能類:
和開發人員聯系最緊密的是
EasyMock
類,這個類提供了
createMock、replay、verify
等方法以及所有預定義的參數匹配器。
我們知道 Mock 對象有兩種創建方式:一種是通過
EasyMock
類提供的
createMock
方法創建,另一種是通過
EasyMock
類的
createControl
方法得到一個
IMocksControl
實例,再由這個
IMocksControl
實例創建 Mock 對象。其實,無論通過哪種方法獲得 Mock 對象,EasyMock 都會生成一個
IMocksControl
的實例,只不過第一種方式中的
IMocksControl
的實例對開發人員不可見而已。這個
IMocksControl
的實例,其實就是
MocksControl
類的一個對象。
MocksControl
類提供了
andReturn、andThrow、times、createMock
等方法。
MocksControl
類中包含了兩個重要的成員變量,分別是接口
IMocksBehavior
和
IMocksControlState
的實例。其中,
IMocksBehavior
的實現類
MocksBehavior
是 EasyMock 的核心類,它保存著一個
ExpectedInvocationAndResult
對象的一個列表,而
ExpectedInvocationAndResult
對象中包含著 Mock 對象方法調用和預期結果的映射。
MocksBehavior
類提供了
addExpected
和
addActual
方法用于添加預期行為和實際調用。
MocksControl
類中包含的另一個成員變量是
IMocksControlState
實例。
IMocksControlState
擁有兩個不同的實現類:
RecordState
和
ReplayState
。顧名思義,
RecordState
是 Mock 對象在 Record 狀態時的支持類,它提供了
invoke
方法在 Record 狀態下的實現。此外,它還提供了
andReturn、andThrow、times
等方法的實現。
ReplayState
是 Mock 對象在 Replay 狀態下的支持類,它提供了
invoke
方法在 Replay 狀態下的實現。在 ReplayState 中,
andReturn、andThrow、times
等方法的實現都是拋出IllegalStateException,因為在 Replay 階段,開發人員不應該再調用這些方法。
當我們調用
MocksControl
的
createMock
方法時,該方法首先會生成一個
JavaProxyFactory
類的對象。
JavaProxyFactory
是接口
IProxyFactory
的實現類,它的主要功能就是通過
java.lang.reflect.Proxy
對指定的接口創建動態代理實例,也就是開發人員在外部看到的 Mock 對象。
在創建動態代理的同時,應當提供
InvocationHandler
的實現類。
MockInvocationHandler
實現了這個接口,它的
invoke
方法主要的功能是根據 Mock 對象狀態的不同而分別調用
RecordState
的
invoke
實現或是
ReplayState
的
invoke
實現。
下圖是創建 Mock 對象的時序圖:
當
EasyMock
類的
createMock
方法被調用時,它首先創建一個
MocksControl
對象,并調用該對象的
createMock
方法創建一個
JavaProxyFactory
對象和一個
MockInvocationHandler
對象。
JavaProxyFactory
對象將
MockInvocationHandler
對象作為參數,通過
java.lang.reflect.Proxy
類的
newProxyInstance
靜態方法創建一個動態代理。
記錄 Mock 的預期行為可以分為兩個階段:預期方法的調用和預期輸出的設定。在外部程序中獲得的 Mock 對象,其實就是由
JavaProxyFactory
創建的指定接口的動態代理,所有外部程序對接口方法的調用,都會指向
InvocationHandler
實現類的
invoke
方法。在 EasyMock 中,這個實現類是
MockInvocationHandler
。下圖是調用預期方法的時序圖:
當
MockInvocationHandler
的
invoke
方法被調用時,它首先通過
reportLastControl
靜態方法將 Mock 對象對應的
MocksControl
對象報告給
LastControl
類,
LastControl
類將該對象保存在一個 ThreadLocal 變量中。接著,
MockInvocationHandler
將創建一個 Invocation 對象,這個對象將保存預期調用的 Mock 對象、方法和預期參數。
在記錄 Mock 對象預期行為時,Mock 對象的狀態是 Record 狀態,因此
RecordState
對象的
invoke
方法將被調用。這個方法首先調用
LastControl
的
pullMatchers
方法獲取參數匹配器。如果您還記得自定義參數匹配器的過程,應該能想起參數匹配器被調用時會將實現類的實例報告給 EasyMock,而這個實例最終保存在
LastControl
中。如果沒有指定參數匹配器,默認的匹配器將會返回給
RecordState
。
根據
Invocation
對象和參數匹配器,
RecordState
將創建一個
ExpectedInvocation
對象并保存下來。
在對預期方法進行調用之后,我們可以對該方法的預期輸出進行設定。我們以
expectLastCall().andReturn(X value).times(int times) |
?
為例說明。如果
times
方法未被顯式的調用,EasyMock 會默認作為
times(1)
處理。下圖是設定預期輸出的時序圖:
在預期方法被調用時,Mock 對象對應的
MocksControl
對象引用已經記錄在
LastControl
中,
expectLastCall
方法通過調用
LastControl
的
lastControl
方法可以獲得這個引用。
MocksControl
對象的
andReturn
方法在 Mock 對象 Record 狀態下會調用
RecordState
的
andReturn
方法,將設定的預期輸出以
Result
對象的形式記錄下來,保存在
RecordState
的 lastResult 變量中。
當
MocksControl
的
times
方法被調用時,它會檢查
RecordState
的 lastResult 變量是否為空。如果不為空,則將 lastResult 和預期方法被調用時創建的
ExpectedInvocation
對象一起,作為參數傳遞給
MocksBehavior
的
addExpected
方法。
MocksBehavior
的
addExpected
方法將這些信息保存在數據列表中。
EasyMock
類的
replay
方法可以將 Mock 對象切換到 Replay 狀態。在 Replay 狀態下,Mock 對象將根據之前的設定返回預期輸出。下圖是 Replay 狀態下 Mock 對象方法調用的時序圖:
在 Replay 狀態下,
MockInvocationHandler
會調用
ReplayState
的
invoke
方法。該方法會把 Mock 對象通過
MocksBehavior
的
addActual
方法添加到實際調用列表中,該列表在
verify
方法被調用時將被用到。同時,
addActual
方法會根據實際方法調用與預期方法調用進行匹配,返回對應的
Result
對象。調用
Result
對象的
answer
方法就可以獲取該方法調用的輸出。
如果您需要在單元測試中構建 Mock 對象來模擬協同模塊或一些復雜對象,EasyMock 是一個可以選用的優秀框架。EasyMock 提供了簡便的方法創建 Mock 對象:通過定義 Mock 對象的預期行為和輸出,你可以設定該 Mock 對象在實際測試中被調用方法的返回值、異常拋出和被調用次數。通過創建一個可以替代現有對象的 Mock 對象,EasyMock 使得開發人員在測試時無需編寫自定義的 Mock 對象,從而避免了額外的編碼工作和因此引入錯誤的機會。
更多文章、技術交流、商務合作、聯系博主
微信掃碼或搜索:z360901061

微信掃一掃加我為好友
QQ號聯系: 360901061
您的支持是博主寫作最大的動力,如果您喜歡我的文章,感覺我的文章對您有幫助,請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧,狠狠點擊下面給點支持吧,站長非常感激您!手機微信長按不能支付解決辦法:請將微信支付二維碼保存到相冊,切換到微信,然后點擊微信右上角掃一掃功能,選擇支付二維碼完成支付。
【本文對您有幫助就好】元
