servlet 单元测试

写了个Servlet的测试用例，初学单元测试，大家帮我看看。

关键字: servlet 单元测试 mock easymock2 junit4

用的JUnit是4.4版，EasyMock2.3。

这是Servlet：

package net.javayuan.servlet;

 import java.io.IOException;

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

 /**
 * @(#)LoginServlet.java 2008-6-4 下午11:39:05
 *
 * @author Yuan
 * LoginServlet
 */
 public class LoginServlet extends HttpServlet {

     /**
     *
     */
     private static final long serialVersionUID = -7068817837964223972L;

     @Override
     public void doPost(HttpServletRequest request, HttpServletResponse response)
     throws IOException,ServletException{
         String username = request.getParameter("username").trim();
         String password = request.getParameter("password");
         if ("yuan".equals(username) && "123456".equals(password))
             response.sendRedirect("welcome.jsp");
         else
             response.sendRedirect("error.jsp");
    }
}

package net.javayuan.servlet;

import java.io.IOException;

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

/**
 * @(#)LoginServlet.java 2008-6-4 下午11:39:05
 * 
 * @author Yuan
 * LoginServlet
 */
public class LoginServlet extends HttpServlet {

	/**
	 * 
	 */
	private static final long serialVersionUID = -7068817837964223972L;

	@Override
	public void doPost(HttpServletRequest request, HttpServletResponse response)
	throws IOException,ServletException{
		String username = request.getParameter("username").trim();
		String password = request.getParameter("password");
		if ("yuan".equals(username) && "123456".equals(password))
			response.sendRedirect("welcome.jsp");
		else
			response.sendRedirect("error.jsp");
	}
}

这是Test Case：

package net.javayuan.servlet;

 import static org.easymock.EasyMock.*;

 import java.io.IOException;

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

 import javax.servlet.ServletException;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 /**
 * @(#)LoginServlet.java 2008-6-5 上午09:19:42
 *
 * @author Yuan
 * LoginServlet Test Case
 */
 public class LoginServletTest {

     private LoginServlet servlet;
     private HttpServletRequest mockRequest;
     private HttpServletResponse mockResponse;

     @Before
     public void setUp(){

         servlet = new LoginServlet();

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

     @After
     public void tearDown(){

         //为了验证指定的调用行为确实发生了，要调用verify(mock)进行验证。
        verify(mockRequest);
        verify(mockResponse);
    }

     /**
     * 测试doPost方法
     * @throws IOException
     * @throws ServletException
     * */
     @Test
     public void testDoPost() throws IOException, ServletException{

         //录制request和response的动作
         mockRequest.getParameter("username");
         expectLastCall().andReturn("yuan");//设置前一方法被调用时的返回值

         mockRequest.getParameter("password");
         expectLastCall().andReturn("123456");

         mockResponse.sendRedirect("welcome.jsp");

         //回放
        replay(mockRequest);
        replay(mockResponse);

         //开始测试Servlet的doPost方法
        servlet.doPost(mockRequest, mockResponse);
    }

}

package net.javayuan.servlet;

import static org.easymock.EasyMock.*;

import java.io.IOException;

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

import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
/**
 * @(#)LoginServlet.java 2008-6-5 上午09:19:42
 * 
 * @author Yuan
 * LoginServlet Test Case
 */
public class LoginServletTest {

	private LoginServlet servlet;
	private HttpServletRequest mockRequest;
	private HttpServletResponse mockResponse;
	
	@Before
	public void setUp(){
		
		servlet = new LoginServlet();
		
		//创建request和response的Mock
		mockRequest = createMock(HttpServletRequest.class);
		mockResponse = createMock(HttpServletResponse.class);
	}
	
	@After
	public void tearDown(){
		
		//为了验证指定的调用行为确实发生了，要调用verify(mock)进行验证。
		verify(mockRequest);
		verify(mockResponse);
	}

	/**
	 * 测试doPost方法
	 * @throws IOException
	 * @throws ServletException
	 * */
	@Test
	public void testDoPost() throws IOException, ServletException{
		
		//录制request和response的动作
		mockRequest.getParameter("username");
		expectLastCall().andReturn("yuan");//设置前一方法被调用时的返回值
		
		mockRequest.getParameter("password");
		expectLastCall().andReturn("123456");
		
		mockResponse.sendRedirect("welcome.jsp");
		
		//回放
		replay(mockRequest);
		replay(mockResponse);
		
		//开始测试Servlet的doPost方法
		servlet.doPost(mockRequest, mockResponse);
	}
	
}