给servlet写单元测试的总结

转自: http://blog.csdn.net/cz_hyf/archive/2007/02/10/1507211.aspx

servlet的测试一般来说需要容器的支持，不是像通常的java类的junit测试一样简单，下面通过对HelloWorld代码的测试阐述了几种servlet测试方法。被测试的HelloWorld类的代码如下：

/**
 * 被测试的servlet
 */
import java.io.IOException; 
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
 
public class HelloWorld extends HttpServlet{
 
 public void saveToSession(HttpServletRequest request) {
         request.getSession().setAttribute("testAttribute",request.getParameter("testparam"));
}
 
 public void doGet(HttpServletRequest request,HttpServletResponse response) throws IOException{
         String username=request.getParameter("username");
         response.getWriter().write(username+":Hello World!");        
 }
 
 public boolean authenticate(){         
        return true; 
 }

}

以HelloWorld为例，我总结了Servlet的多种测试方法如下：
一.使用HttpUnit测试

import com.meterware.httpunit.GetMethodWebRequest;
import com.meterware.httpunit.WebRequest;
import com.meterware.httpunit.WebResponse;
import com.meterware.servletunit.InvocationContext;
import com.meterware.servletunit.ServletRunner;
import com.meterware.servletunit.ServletUnitClient;
import junit.framework.Assert;
import junit.framework.TestCase;
 
public class HttpUnitTestHelloWorld extends TestCase {
 
 protected void setUp() throws Exception {
  super.setUp();
 }
 
 protected void tearDown() throws Exception {
  super.tearDown();
 }
 
 public void testHelloWorld() {

		try {

			// 创建Servlet的运行环境
			ServletRunner sr = new ServletRunner();

			// 向环境中注册Servlet
			sr.registerServlet("HelloWorld", HelloWorld.class.getName());

			// 创建访问Servlet的客户端
			ServletUnitClient sc = sr.newClient();

			// 发送请求
			WebRequest request = new GetMethodWebRequest(
					"http://localhost/HelloWorld");
			request.setParameter("username", "testuser");

			InvocationContext ic = sc.newInvocation(request);

			HelloWorld is = (HelloWorld) ic.getServlet();

			// 测试servlet的某个方法
			Assert.assertTrue(is.authenticate());

			// 获得模拟服务器的信息
			WebResponse response = sc.getResponse(request);

			// 断言
			Assert.assertTrue(response.getText().equals("testuser:Hello World!"));

		} catch (Exception e) {
			e.printStackTrace();
		}
	}
 
}

上述例子其实是junit的一个测试例子，在其中使用了httpunit模拟的servlet环境,使用上述方法测试servlet可以脱离容器，容易把该测试写入ant或maven脚本，让测试进行。
httpunit网址：
http://httpunit.sourceforge.net/
使用该种方法测试的弱点就是：如果要使用request(response)的setCharercterEncoding方法时，测试会出现一些问题,而且httpunit在测试servlet行为时，采用的是完全模拟浏览器，有时测试比较难写。

二 使用cactus测试

/**
 * cactus测试servlet的例子
 * 必须要有tomcat的支持
 */
import junit.framework.Test;
import junit.framework.TestSuite;
import org.apache.cactus.ServletTestCase; 
import org.apache.cactus.WebRequest;
import org.apache.cactus.WebResponse;
public class CactusHelloWorld extends ServletTestCase{
  
     HelloWorld servlet;
     public CactusHelloWorld(String theName) {
         super(theName);
     }
 
     protected void setUp() throws Exception {
         super.setUp();
         servlet = new HelloWorld();
     }
 
     protected void tearDown() throws Exception {
         super.tearDown();
     }
 
     /**
      * 测试方法测试参数在此设置
     * @param webrequest
      */
 
     public void beginSaveToSessionOK(WebRequest request) {
         request.addParameter("testparam", "it works!");
     }
     
     /**
      * 测试方法测试参数在此设置
      * 
      * @param webrequest
      */
 
     public void beginDoGet(WebRequest request) {
         request.addParameter("username", "testuser");
     }
 
     /**
      * 调用servlet的测试方法
      *  
      */

     public void testSaveToSessionOK() {
         servlet.saveToSession(request);
         assertEquals("it works!", session.getAttribute("testAttribute"));
     }
 
     public void testDoGet() {
         try {
             servlet.doGet(request, response);
         } catch (Exception e) {
             e.printStackTrace();
         }
     }
 
     /**
      * 此方法可以判断测试方法的输出，会传递测试方法的reponse给end***,并且格式化为cactus
      * 的WebResponse或者可以跟httpunit集成，格式化为httpunit的response
      * 
      * @param response
      */

     public void endDoGet(WebResponse response) {
         String content;         
         content = response.getText();
         assertEquals("testuser:Hello World!", content);
     }
}
 

cactus具备丰富灵活的测试功能，如要测试doGet方法，分为beginDoGet(模拟测试参数设置)、DoGet(执行测试)、endDoGet(状态结果验证)相比httpunit来说，写测试更为容易，测试servlet更为专业,流程更为清晰，但是cactus需要容器支持，使得测试不可以自动进行，但是如果使用一个嵌入式的容器，测试就可以自动了。
cactus是一个servlet和jsp的测试框架:
http://jakarta.apache.org/cactus/getting_started.html
三 使用Jetty作为嵌入式容器测试servlet.

/**
 * 一个关于嵌入式jetty测试的例子，jetty作为stubs的一个例子
 */
package com.easyjf.testexample;
import org.mortbay.jetty.Connector;
import org.mortbay.jetty.Server;
import org.mortbay.jetty.bio.SocketConnector;
import org.mortbay.jetty.servlet.ServletHandler; 
import com.meterware.httpunit.WebClient;
import com.meterware.httpunit.WebConversation;
import com.meterware.httpunit.WebResponse; 
import junit.framework.Assert;
import junit.framework.TestCase;
 
public class JettySampleTest extends TestCase {
 
 Server server;
 protected void setUp() throws Exception {
      //通过代码设置并启动一个服务器，该服务器是servlet的测试容器
      super.setUp();
      server = new Server();
      Connector connector=new SocketConnector();
      connector.setPort(80);
      server.setConnectors(new Connector[]{connector});
      ServletHandler handler=new ServletHandler();
      server.setHandler(handler);
      handler.addServletWithMapping("HelloWorld", "/");
      server.start();
 }
 
 protected void tearDown() throws Exception {
  super.tearDown();
  server.stop();
 }
 
 public void testHellWorld() {
  try {
   WebConversation wc = new WebConversation();
   WebResponse web = wc.getResponse("http://127.0.0.1/HelloWorld");
   String result=web.getText();
   Assert.assertEquals(result,"it works!");
  } catch (Exception e) {
   e.printStackTrace();
  }
 }
}
 

可以发现，jetty可以充当一个servlet的容器，方便的是,jetty支持嵌入式服务，即可以通过代码来启动，所以要写自动测试的例子很方便，可以结合httpunit或者cactus进行servlet测试。

jetty主页：http://docs.codehaus.org/display/JETTY/Embedding+Jetty

四 使用mock对象，此处使用easymock

 
import java.io.PrintWriter;
import java.io.Writer;
import javax.servlet.ServletConfig;
import javax.servlet.ServletContext;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse; 
import junit.framework.Assert;
import junit.framework.TestCase;
import static org.easymock.EasyMock.*;
public class MockTestServlet extends TestCase {
 
    public void testService() throws Exception {

        System.out.println("service");

        HttpServletRequest request = createMock(HttpServletRequest.class);

        HttpServletResponse response = createMock(HttpServletResponse.class);

        //Creating the ServletConfig mock here

        ServletConfig servletConfig = createMock(ServletConfig.class);

        //Creating the ServletContext mock here

        ServletContext servletContext = createMock(ServletContext.class);
        
        //Create the target object  
      
        HelloWorld4 instance = new HelloWorld();

        //初始化servlet,一般由容器承担，一般调用servletConfig作为参数初始化，此处模拟容器行为

        instance.init(servletConfig);
 
        //在某些方法被调用时设置期望的返回值，如下这样就不会去实际调用servletConfig的getServletContext方法，而是直接返回
 
        //servletContext,由于servletConfig是mock出来的，所以可以完全控制。

        expect(servletConfig.getServletContext()).andReturn(servletContext).anyTimes();

        expect(request.getParameter("username")).andReturn("testuser");

        PrintWriter pw=new PrintWriter(System.out,true);

        expect(response.getWriter()).andReturn(pw).anyTimes();
        
        //以上均是录制，下面为重放，该种机制为easymock测试机制，要理解请看easymock测试的一些资料
        replay(request);
        replay(response);
        replay(servletConfig);
        replay(servletContext);
 
        instance.doGet(request, response);

        pw.flush();        
 
        //验证结果是否预期，如果预期，则会在pw上写出testuser.
        verify(request);
        verify(response);
        verify(servletConfig);
        verify(servletContext);
   }
}
 

mock测试注重行为，mock对象其实都是模拟的对象，方法一般直接给出一个返回值，没有具体的对象逻辑，mock对象是用来帮助测试要测试的类的。比如要测试servlet的内部行为，又不想要容器等环境，就可以采用mock测试。

easymock是mock测试的一个框架：http://www.easymock.org/