作者：xuefeng 文章來源：本站原創(chuàng) 點(diǎn)擊數(shù)：61 更新時(shí)間：2006-2-25
單元測(cè)試是XP極力推薦的測(cè)試驅(qū)動(dòng)開發(fā)模式，是保證軟件質(zhì)量的重要方法。盡管如此，對(duì)許多類的單元測(cè)試仍然是極其困難的，例如，對(duì)數(shù)據(jù)庫(kù)操作的類進(jìn)行測(cè)試，如果不準(zhǔn)備好數(shù)據(jù)庫(kù)環(huán)境以及相關(guān)測(cè)試數(shù)據(jù)，是很難進(jìn)行單元測(cè)試的；再例如，對(duì)需要運(yùn)行在容器內(nèi)的Servlet或EJB組件，脫離了容器也難于測(cè)試。
幸運(yùn)的是，Mock Object可以用來模擬一些我們需要的類，這些對(duì)象被稱之為模仿對(duì)象，在單元測(cè)試中它們特別有價(jià)值。
Mock Object用于模仿真實(shí)對(duì)象的方法調(diào)用，從而使得測(cè)試不需要真正的依賴對(duì)象。Mock Object只為某個(gè)特定的測(cè)試用例的場(chǎng)景提供剛好滿足需要的最少功能。它們還可以模仿錯(cuò)誤的條件，例如拋出指定的異常等。
目前，有許多可用的Mock類庫(kù)可供我們選擇。一些Mock庫(kù)提供了常見的模仿對(duì)象，例如：HttpServletRequest，而另一些Mock庫(kù)則提供了動(dòng)態(tài)生成模仿對(duì)象的功能，本文將討論使用EasyMock動(dòng)態(tài)生成模仿對(duì)象以便應(yīng)用于單元測(cè)試。
到目前為止，EasyMock提供了1.2版本和2.0版本，2.0版本僅支持Java SE 5.0，本例中，我們選擇EasyMock 1.2 for Java 1.3版本進(jìn)行測(cè)試，可以從
http://www.easymock.org下載合適的版本。
我們首先來看一個(gè)用戶驗(yàn)證的LoginServlet類：
/**
* LoginServlet.java
* Author: Liao Xue Feng,
www.crackj2ee.com*/
package com.crackj2ee.test.mock;
import java.io.*;
import javax.servlet.*;
import javax.servlet.http.*;
public class LoginServlet extends HttpServlet {
protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
String username = request.getParameter("username");
String password = request.getParameter("password");
// check username & password:
if("admin".equals(username) && "123456".equals(password)) {
ServletContext context = getServletContext();
RequestDispatcher dispatcher = context.getNamedDispatcher("dispatcher");
dispatcher.forward(request, response);
}
else {
throw new RuntimeException("Login failed.");
}
}
}
這個(gè)Servlet實(shí)現(xiàn)簡(jiǎn)單的用戶驗(yàn)證的功能，若用戶名和口令匹配“admin”和“123456”，則請(qǐng)求被轉(zhuǎn)發(fā)到指定的dispatcher上，否則，直接拋出RuntimeException。
為了測(cè)試doPost()方法，我們需要模擬HttpServletRequest，ServletContext和RequestDispatcher對(duì)象，以便脫離J2EE容器來測(cè)試這個(gè)Servlet。
我們建立TestCase，名為L(zhǎng)oginServletTest：
public class LoginServletTest extends TestCase {
}
我們首先測(cè)試當(dāng)用戶名和口令驗(yàn)證失敗的情形，演示如何使用EasyMock來模擬HttpServletRequest對(duì)象：
public void testLoginFailed() throws Exception {
MockControl mc = MockControl.createControl(HttpServletRequest.class);
HttpServletRequest request = (HttpServletRequest)mc.getMock();
// set Mock Object behavior:
request.getParameter("username");
mc.setReturnValue("admin", 1);
request.getParameter("password");
mc.setReturnValue("1234", 1);
// ok, all behaviors are set!
mc.replay();
// now start test:
LoginServlet servlet = new LoginServlet();
try {
servlet.doPost(request, null);
fail("Not caught exception!");
}
catch(RuntimeException re) {
assertEquals("Login failed.", re.getMessage());
}
// verify:
mc.verify();
}
仔細(xì)觀察測(cè)試代碼，使用EasyMock來創(chuàng)建一個(gè)Mock對(duì)象需要首先創(chuàng)建一個(gè)MockControl：
MockControl mc = MockControl.createControl(HttpServletRequest.class);
然后，即可獲得MockControl創(chuàng)建的Mock對(duì)象：
HttpServletRequest request = (HttpServletRequest)mc.getMock();
下一步，我們需要“錄制”Mock對(duì)象的預(yù)期行為。在LoginServlet中，先后調(diào)用了request.getParameter("username")和request.getParameter("password")兩個(gè)方法，因此，需要在MockControl中設(shè)置這兩次調(diào)用后的指定返回值。我們期望返回的值為“admin”和“1234”：
request.getParameter("username"); // 期望下面的測(cè)試將調(diào)用此方法，參數(shù)為"username"
mc.setReturnValue("admin", 1); // 期望返回值為"admin"，僅調(diào)用1次
request.getParameter("password"); // 期望下面的測(cè)試將調(diào)用此方法，參數(shù)為" password"
mc.setReturnValue("1234", 1); // 期望返回值為"1234"，僅調(diào)用1次
緊接著，調(diào)用mc.replay()，表示Mock對(duì)象“錄制”完畢，可以開始按照我們?cè)O(shè)定的方式運(yùn)行，我們對(duì)LoginServlet進(jìn)行測(cè)試，并預(yù)期會(huì)產(chǎn)生一個(gè)RuntimeException：
LoginServlet servlet = new LoginServlet();
try {
servlet.doPost(request, null);
fail("Not caught exception!");
}
catch(RuntimeException re) {
assertEquals("Login failed.", re.getMessage());
}
由于本次測(cè)試的目的是檢查當(dāng)用戶名和口令驗(yàn)證失敗后，LoginServlet是否會(huì)拋出RuntimeException，因此，response對(duì)象對(duì)測(cè)試沒有影響，我們不需要模擬它，僅僅傳入null即可。
最后，調(diào)用mc.verify()檢查Mock對(duì)象是否按照預(yù)期的方法調(diào)用正常運(yùn)行了。
運(yùn)行JUnit，測(cè)試通過！表示我們的Mock對(duì)象正確工作了！
下一步，我們來測(cè)試當(dāng)用戶名和口令匹配時(shí)，LoginServlet應(yīng)當(dāng)把請(qǐng)求轉(zhuǎn)發(fā)給指定的RequestDispatcher。在這個(gè)測(cè)試用例中，我們除了需要HttpServletRequest Mock對(duì)象外，還需要模擬ServletContext和RequestDispatcher對(duì)象：
MockControl requestCtrl = MockControl.createControl(HttpServletRequest.class);
HttpServletRequest requestObj = (HttpServletRequest)requestCtrl.getMock();
MockControl contextCtrl = MockControl.createControl(ServletContext.class);
final ServletContext contextObj = (ServletContext)contextCtrl.getMock();
MockControl dispatcherCtrl = MockControl.createControl(RequestDispatcher.class);
RequestDispatcher dispatcherObj = (RequestDispatcher)dispatcherCtrl.getMock();
按照doPost()的語句順序，我們?cè)O(shè)定Mock對(duì)象指定的行為：
requestObj.getParameter("username");
requestCtrl.setReturnValue("admin", 1);
requestObj.getParameter("password");
requestCtrl.setReturnValue("123456", 1);
contextObj.getNamedDispatcher("dispatcher");
contextCtrl.setReturnValue(dispatcherObj, 1);
dispatcherObj.forward(requestObj, null);
dispatcherCtrl.setVoidCallable(1);
requestCtrl.replay();
contextCtrl.replay();
dispatcherCtrl.replay();
然后，測(cè)試doPost()方法，這里，為了讓getServletContext()方法返回我們創(chuàng)建的ServletContext Mock對(duì)象，我們定義一個(gè)匿名類并覆寫getServletContext()方法：
LoginServlet servlet = new LoginServlet() {
public ServletContext getServletContext() {
return contextObj;
}
};
servlet.doPost(requestObj, null);
最后，檢查所有Mock對(duì)象的狀態(tài)：
requestCtrl.verify();
contextCtrl.verify();
dispatcherCtrl.verify();
運(yùn)行JUnit，測(cè)試通過！
倘若LoginServlet的代碼有誤，例如，將context.getNamedDispatcher("dispatcher")誤寫為 context.getNamedDispatcher("dispatcher2")，則測(cè)試失敗，JUnit報(bào)告：
junit.framework.AssertionFailedError:
Unexpected method call getNamedDispatcher("dispatcher2"):
getNamedDispatcher("dispatcher2"): expected: 0, actual: 1
getNamedDispatcher("dispatcher"): expected: 1, actual: 0
at ...
完整的LoginServletTest代碼如下： /**
* LoginServletTest.java
* Author: Liao Xue Feng,
www.crackj2ee.com*/
package com.crackj2ee.test.mock; import javax.servlet.*;
import javax.servlet.http.*; import org.easymock.*;
import junit.framework.TestCase; public class LoginServletTest extends TestCase { public void testLoginFailed() throws Exception {
MockControl mc = MockControl.createControl(HttpServletRequest.class);
HttpServletRequest request = (HttpServletRequest)mc.getMock();
// set Mock Object behavior:
request.getParameter("username");
mc.setReturnValue("admin", 1);
request.getParameter("password");
mc.setReturnValue("1234", 1);
// ok, all behaviors are set!
mc.replay();
// now start test:
LoginServlet servlet = new LoginServlet();
try {
servlet.doPost(request, null);
fail("Not caught exception!");
}
catch(RuntimeException re) {
assertEquals("Login failed.", re.getMessage());
}
// verify:
mc.verify();
} public void testLoginOK() throws Exception {
// create mock:
MockControl requestCtrl = MockControl.createControl(HttpServletRequest.class);
HttpServletRequest requestObj = (HttpServletRequest)requestCtrl.getMock();
MockControl contextCtrl = MockControl.createControl(ServletContext.class);
final ServletContext contextObj = (ServletContext)contextCtrl.getMock();
MockControl dispatcherCtrl = MockControl.createControl(RequestDispatcher.class);
RequestDispatcher dispatcherObj = (RequestDispatcher)dispatcherCtrl.getMock();
// set behavior:
requestObj.getParameter("username");
requestCtrl.setReturnValue("admin", 1);
requestObj.getParameter("password");
requestCtrl.setReturnValue("123456", 1);
contextObj.getNamedDispatcher("dispatcher");
contextCtrl.setReturnValue(dispatcherObj, 1);
dispatcherObj.forward(requestObj, null);
dispatcherCtrl.setVoidCallable(1);
// done!
requestCtrl.replay();
contextCtrl.replay();
dispatcherCtrl.replay();
// test:
LoginServlet servlet = new LoginServlet() {
public ServletContext getServletContext() {
return contextObj;
}
};
servlet.doPost(requestObj, null);
// verify:
requestCtrl.verify();
contextCtrl.verify();
dispatcherCtrl.verify();
}
}
總結(jié)：
雖然EasyMock可以用來模仿依賴對(duì)象，但是，它只能動(dòng)態(tài)模仿接口，無法模仿具體類。這一限制正好要求我們遵循“針對(duì)接口編程”的原則：如果不針對(duì)接口，則測(cè)試難于進(jìn)行。應(yīng)當(dāng)把單元測(cè)試看作是運(yùn)行時(shí)代碼的最好運(yùn)用，如果代碼在單元測(cè)試中難于應(yīng)用，則它在真實(shí)環(huán)境中也將難于應(yīng)用?？傊?，創(chuàng)建盡可能容易測(cè)試的代碼就是創(chuàng)建高質(zhì)量的代碼。
