EasyMock测试Servlet端实例

下面介绍两个类Login 和 LoginTest，其中LoginTest是用EasyMock对Login的servlet的测试类

Login.java

package servlet.manage.users;

import java.io.IOException;
import java.io.PrintWriter;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import delegate.common.message.MessInfo;
import delegate.logic.manage.users.MLogin;

/**
 * @Project Name : Login
 *
 * @author  Zou Shasha
 *
 * This class is a servlet client for user login by which the UserID and password are verified.
**/

@SuppressWarnings("serial")
public class Login extends HttpServlet {

    /**
     * Constructor of the object.
     */
    public Login() {
        super();
    }
    /**
     * The doGet method of the servlet. <br>
     *
     * This method is called when a form has its tag value method equals to get.
     * @param request the request send by the client to the server
     * @param response the response send by the server to the client
     * @throws ServletException if an error occurred
     * @throws IOException if an error occurred
     */
    public void doGet(HttpServletRequest request, HttpServletResponse response)
        throws ServletException, IOException {
        //Gets a RequestDispatcher object and forwards a request from a servlet to "menu.jsp" on the server.
        request.getRequestDispatcher("menu.jsp").forward(request, response);
        //response.sendRedirect("menu.jsp");
    }

    /**
     * The doPost method of the servlet. <br>
     * This method is called when a form has its tag value method equals to post.
     * And gets the values of User ID and password and veri
     * @param request the request send by the client to the server
     * @param response the response send by the server to the client
     * @throws ServletException if an error occurred
     * @throws IOException if an error occurred
     */
    public void doPost(HttpServletRequest request, HttpServletResponse response)
        throws ServletException, IOException {

        //Sets the content type of the response for text or html  being sent to the client
        response.setContentType("text/html;charset=utf-8");

        //Overrides UTF-8 used in the body of this request.
        request.setCharacterEncoding("utf-8");

        //Returns a PrintWriter object that can send character text to the client.
        PrintWriter out = response.getWriter();

        //Returns the value of UserID as a String
        String UserID = request.getParameter("UserID");

        //Returns the value of Password as a String
        String Password = request.getParameter("Password");

        //Create a new instance of MLogin class
        MLogin oper = new MLogin();

        //Verifies the User ID and the password  and returns the result as a String
        String identify = oper.loginCheck(UserID, Password);

        //The User ID and the password are legal
        if (identify.equals("true")) {

            //Returns the current HttpSession associated with this request or, if there is no current session and create is true, returns a new session.
            HttpSession session = request.getSession(true);

            //Binds the value of UserID and Password to this session, using UserID and Password String.
            session.setAttribute("UserID", UserID);
            session.setAttribute("Password", Password);
            //Goes to the menu page.
            out.println("<script>");
            out.println("window.location.href='../menu/home'");
            out.println("</script>");
        } else if (identify.equals("UserIDError")) {
            //Pops up alarm in case of non-registered user ID and goes to the login page.

            String message1 =   new String(new MessInfo().getMessInfo("user_login_5").getBytes("iso-8859-1"), "UTF-8");
            out.println("<script>");
            out.println("alert('" + message1 + "')");
            out.println("</script>");
            out.println("<script>");
            out.println("window.location.href='../../../Login.jsp'");
            out.println("</script>");
        } else if (identify.equals("PasswordError")) {
            //Pops up alarm in case of invalid password and goes to the login page.

            String message2 =   new String(new MessInfo().getMessInfo("user_login_6").getBytes("iso-8859-1"), "UTF-8");
            out.println("<script>");
            out.println("alert('" + message2 + "')" );
            out.println("</script>");
            out.println("<script>");
            out.println("window.location.href='../../../Login.jsp'");
            out.println("</script>");
        } else {
            //Pops up alarm in case of database access failure and goes to the login page.
            String message3 =   new String(new MessInfo().getMessInfo("user_login_7").getBytes("iso-8859-1"), "UTF-8");
            out.println("<script>");
            out.println("alert('" + message3 + "')");
            out.println("</script>");
            out.println("<script>");
            out.println("window.location.href='../../../Login.jsp'");
            out.println("</script>");
        }
    }
}

 

LoginTest.java

package servlet.manage.users;

import static org.junit.Assert.*;

import java.io.IOException;
import java.io.PrintWriter;
import java.io.StringWriter;
import javax.servlet.RequestDispatcher;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.easymock.EasyMock;

import delegate.common.message.MessInfo;

/**

 * @Project Name : LoginTest
 * @author  Liu Chen
 *          Zou Shasha
 *
 * This class is the test class for the Login class.
 *
 */
public class LoginTest {

    private Login servlet;
    private HttpServletRequest mockRequest;
    private HttpServletResponse mockResponse;

    public LoginTest() {
    }
    @Before
    public void setUp() {

        servlet = new Login();

        //创建request和response的Mock
        mockRequest = (HttpServletRequest)EasyMock.createMock(HttpServletRequest.class);
        mockResponse = (HttpServletResponse) EasyMock.createMock(HttpServletResponse.class);
    }

    @After
    public void tearDown() {

    }

    @Test
    public void testDoGetHttpServletRequestHttpServletResponse() throws ServletException, IOException {

        RequestDispatcher dispatcher = (RequestDispatcher) EasyMock.createMock(RequestDispatcher.class);

        EasyMock.expect(mockRequest.getRequestDispatcher("menu.jsp")).andReturn(dispatcher);

        dispatcher.forward(mockRequest, mockResponse);

        //回放
        EasyMock.replay(mockRequest);
        EasyMock.replay(mockResponse);
        EasyMock.replay(dispatcher);

        servlet.doGet(mockRequest, mockResponse);

        EasyMock.verify(mockRequest);
        EasyMock.verify(mockResponse);
        EasyMock.verify(dispatcher);
    }

    @Test
    public void testLoginSuccess() throws IOException {

        //Sets the content type of the response for text or html  being sent to the client
        mockResponse.setContentType("text/html;charset=utf-8");

        //Overrides UTF-8 used in the body of this request.
        mockRequest.setCharacterEncoding("utf-8");

        //录制request和response的动作
        EasyMock.expect(mockRequest.getParameter("UserID")).andReturn("123");
        EasyMock.expect(mockRequest.getParameter("Password")).andReturn("123");

        StringWriter output = new StringWriter();
        PrintWriter contentWriter = new PrintWriter(output);
        EasyMock.expect(mockResponse.getWriter()).andReturn(contentWriter);

        HttpSession session = (HttpSession) EasyMock.createMock(HttpSession.class);
        EasyMock.expect(mockRequest.getSession(true)).andReturn(session);
        session.setAttribute("UserID", "123");
        session.setAttribute("Password","123");

        //回放
        EasyMock.replay(session);
        EasyMock.replay(mockRequest);
        EasyMock.replay(mockResponse);

        //开始测试Servlet的doPost方法
        try {
            servlet.doPost(mockRequest, mockResponse);
        } catch (ServletException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        EasyMock.verify(session);
        EasyMock.verify(mockRequest);
        EasyMock.verify(mockResponse);
       // System.out.println(output.toString());
        assertEquals("<script>\r\nwindow.location.href='../menu/home'\r\n</script>\r\n", output.toString());

    }
    @Test
    public void testLoginUserIDError() throws IOException {
        //Sets the content type of the response for text or html  being sent to the client
        mockResponse.setContentType("text/html;charset=utf-8");

        //Overrides UTF-8 used in the body of this request.
        mockRequest.setCharacterEncoding("utf-8");

        //录制request和response的动作
      //  mockRequest.getParameter("UserID");
      //  EasyMock.expectLastCall().andReturn("123");//设置前一方法被调用时的返回值
        EasyMock.expect(mockRequest.getParameter("UserID")).andReturn("aaa");
        EasyMock.expect(mockRequest.getParameter("Password")).andReturn("123");

        StringWriter output = new StringWriter();
        PrintWriter contentWriter = new PrintWriter(output);
        EasyMock.expect(mockResponse.getWriter()).andReturn(contentWriter);
        //回放
        
        EasyMock.replay(mockRequest);
        EasyMock.replay(mockResponse);

        try {
            servlet.doPost(mockRequest, mockResponse);
        } catch (ServletException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        EasyMock.verify(mockRequest);
        EasyMock.verify(mockResponse);
       // System.out.println(output.toString());
        
        String message = new String(new MessInfo().getMessInfo("user_login_5").getBytes("iso-8859-1"), "UTF-8");

        assertEquals("<script>\r\nalert('"+message+"')\r\n</script>\r\n<script>\r\nwindow.location.href='../../../Login.jsp'\r\n</script>\r\n", output.toString());
    }

    @Test
    public void testLoginPwdError() throws IOException {
      //Sets the content type of the response for text or html  being sent to the client
        mockResponse.setContentType("text/html;charset=utf-8");

        //Overrides UTF-8 used in the body of this request.
        mockRequest.setCharacterEncoding("utf-8");

        //录制request和response的动作
      //  mockRequest.getParameter("UserID");
      //  EasyMock.expectLastCall().andReturn("123");//设置前一方法被调用时的返回值
        EasyMock.expect(mockRequest.getParameter("UserID")).andReturn("123");
        EasyMock.expect(mockRequest.getParameter("Password")).andReturn("aaa");

        StringWriter output = new StringWriter();
        PrintWriter contentWriter = new PrintWriter(output);
        EasyMock.expect(mockResponse.getWriter()).andReturn(contentWriter);
        //回放

        EasyMock.replay(mockRequest);
        EasyMock.replay(mockResponse);

        try {
            servlet.doPost(mockRequest, mockResponse);
        } catch (ServletException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        EasyMock.verify(mockRequest);
        EasyMock.verify(mockResponse);
        String message = new String(new MessInfo().getMessInfo("user_login_6").getBytes("iso-8859-1"), "UTF-8");
        assertEquals("<script>\r\nalert('"+message+"')\r\n</script>\r\n<script>\r\nwindow.location.href='../../../Login.jsp'\r\n</script>\r\n", output.toString());
    }
}

 

 

另外，大多数人在写源程序的时候都有用到request.setAttribute("list",list);其中list表示业务逻辑的运行后的封装结果，这个在EasyMock中可用mockRequest.setAttribute(EasyMock.eq("list"), (List)EasyMock.anyObject())

或者mockRequest.setAttribute(EasyMock.eq("list"), EasyMock.isA(List.class)) 模拟。