单元测试整理（六）—— 使用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

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

猜你喜欢