单元测试整理（六）—— 使用EasyMock和JUnit进行单元测试

EasyMock是Apache许可下发布的Java开源测试框架，它可以和jUnit很好的继承在一起。该框架允许为测试驱动开发（TDD）或行为驱动开发（BDD）创建测试双重对象1。使用EasyMock只需导入相应的jar包即可，本篇用到的所有jar包和代码都可以在我的Github下载。

在这里我们用一个进行用户验证的servlet代码作为被测代码，这段代码来自我之前看过的一篇EasyMock教程2。代码逻辑很简单常用，下面我们用JUnit+EasyMock写出它的单元测试代码。

// loginServlet.java
import java.io.*;
import javax.servlet.*;
import javax.servlet.http.*;

public class loginServlet extends HttpServlet {
protected void doPost(HttpServletRequest req, HttpServletResponse res)
throws ServletException, IOException{
String username = req.getParameter("username");
String password = req.getParameter("password");
if("admin".equals(username) && "123456".equals(password)) {
ServletContext context = getServletContext();
RequestDispatcher dispatcher = context.getNamedDispatcher("dispatcher");
dispatcher.forward(req, res);
}else {
throw new RuntimeException("Login failed.");
}
}
}

下面一步一步地说一下单元测试代码的构建：

1，导入需要的jar包，配置JUnit。jar包在我的Github里有，就是servlet和easymock的包。JUnit在 项目右键->Build Path->Configure Build Path->Libraries->Add Library->JUnit，选择JUnit4然后Finish。

2，import下面这些，如果报错说明第一步有问题，解决之

3，为了测试doPost()方法，我们需要模拟HttpServletRequest，ServletContext和RequestDispatcher对象，以便脱离J2EE容器来测试这个Servlet。我们建立TestCase，名为loginServletTest

4，定义我们需要用到的变量

5，setUp建立所需环境，主要就是mock我们要用到的但是不是自己实现的类和接口，（这话不严谨，先这么理解吧）或者说创建mock对象。

6，tearDown释放资源，可以不写

7，编写测试方法

7.1，设置期望得到的值，或者说“录制”mock对象的预期行为

7.2，激活mock，即replay

7.3，调用被测代码的方法进行测试

7.4，检查mock对象的状态

// loginServletTest.java
//静态导入这样可以直接使用expect等方法
import static org.easymock.EasyMock.expect;
import static org.easymock.EasyMock.replay;
import static org.easymock.EasyMock.verify;
import javax.servlet.*;
import javax.servlet.http.*;
import org.easymock.*;
import junit.framework.*;
import org.junit.*;
import org.junit.Test;
public class loginServletTest extends TestCase {
private HttpServletRequest request;
private loginServlet servlet;
private ServletContext context;
private RequestDispatcher dispatcher;
@Before
//setup用于环境的建立，在其中mock需要用到的接口
public void setUp() throws Exception{
request = EasyMock.createMock(HttpServletRequest.class);
context = EasyMock.createMock(ServletContext.class);
dispatcher = EasyMock.createMock(RequestDispatcher.class);
servlet = new loginServlet();
}
@After
//好像没什么要释放的，可以不写，这里为了完整性
public void tearDown() {

}
@Test
public void testLoginFailed() throws Exception {
//设置期望得到的值
expect(request.getParameter("username")).andReturn("admin");
expect(request.getParameter("password")).andReturn("1234");
//setup完成激活mock
replay(request);
try {
servlet.doPost(request, null);
fail("Not caught exception!");
}
catch(RuntimeException re) {
assertEquals("Login failed.", re.getMessage());
}
//检查mock对象的状态
verify(request);
}
@Test
public void testLoginOK() throws Exception {
//设置期望得到的值，最好按照doPost的语句顺序来
expect(request.getParameter("username")).andReturn("admin");
expect(request.getParameter("password")).andReturn("123456");
expect(context.getNamedDispatcher("dispatcher")).andReturn(dispatcher);
dispatcher.forward(request, null);
//setup完成激活mock
replay(request);
replay(context);
replay(dispatcher);
/*为了让getServletContext()方法返回我们创建的ServletContext Mock对象，
我们定义一个匿名类并覆写getServletContext()方法*/
loginServlet servlet = new loginServlet() {
public ServletContext getServletContext() {
return context;
}
};
//测试doPost方法
servlet.doPost(request, null);
//检查mock对象的状态
verify(request);
verify(context);
verify(dispatcher);
}
}

下一章介绍Spring MVC

维基百科 ↩
廖雪峰的博客 ↩