Jetty+JUnit实现单元测试

    这两天在研究jetty和junit联合进行stub方式测试的东西，小有所得，参考的书籍是《junit in action》，不过此书出的比较早，用的jetty版本也比较低，对于现在最新版的6.1.26，变化太大，书中的例子已不好用，花了很多功夫转换成最新的版本。

  

    Jetty是一个轻量级的web容器，其明显的优点就是可以嵌入到java程序中，从程序中启动，定制handler去处理请求，这个特点可以用在单元测试中，可以以stub的方式进行模拟的测试，不必启动真实的服务器。 首先给出一个简单的示例，说明如何嵌入及响应请求。

 

Server server = new Server(8080); //指定一个index.html作为响应的结果 ResourceHandler resourceHandler = new ResourceHandler(); resourceHandler.setWelcomeFiles(new String[]{"index.html"}); resourceHandler.setResourceBase(".");//指定文件相对的路径 Log.info("servering " + resourceHandler.getBaseResource()); HandlerList handlerList = new HandlerList(); handlerList.setHandlers(new Handler[]{resourceHandler}); server.setHandler(handlerList); server.start();

 

index.html里只有一句：hello,theoffspring

 

启动这段程序后，在浏览器中输入http://127.0.0.1:8080/，即可看到index.html里面的内容，是不是很简单？

 

再贴一段程序来发起一个http请求，基于刚才的例子。

HttpClient httpClient = new HttpClient(); httpClient.start(); ContentExchange contentExchange = new ContentExchange(); httpClient.setConnectorType(HttpClient.CONNECTOR_SELECT_CHANNEL); contentExchange.setURL("http://127.0.0.1:8080/"); httpClient.send(contentExchange); contentExchange.waitForDone(); System.err.println("Response status: "+contentExchange.getResponseContent());

   这是以程序的方式去请求一个地址，结果和你在浏览器中输入网址一个效果，试了一下，访问真实的url的时候，比如网易首页，会被重定向，返回的status为302，穿墙能力显然不够强，不过这也用不着。此处contentExchange.waitForDone()表明请求的方式是同步的，另有一种异步的方式：

HttpClient client = new HttpClient(); client.setConnectorType(HttpClient.CONNECTOR_SELECT_CHANNEL); try { client.start(); } catch (Exception e) { throw new ServletException(e); } // create the exchange object, which lets you define where you want to go // and what you want to do once you get a response ContentExchange exchange = new ContentExchange() { // define the callback method to process the response when you get it back protected void onResponseComplete() throws IOException { super.onResponseComplete(); String responseContent = this.getResponseContent(); // do something with the response content ... } }; exchange.setMethod("GET"); exchange.setURL("http://www.example.com/"); // start the exchange client.send(exchange);

 

它需要你去实现一个ContentExchange，定义响应返回时如何进行处理，要异步，这是必然的，很好理解。

 

行了，基础知识讲完了，下面演示如何结合junit写一个单元测试，目的是先定义一个请求处理器，对于某个特定的url，进行某种内容的响应返回，然后单元测试一下，是不是和期望的结果一致。

 

因为启动和停止服务器需要一定的时间，这两个操作写到每个TestCase中的setup和teardown里显然不是一种明智的办法，因为太耗时间，所以咱们让它继承一个TestSetup类，这样，一组测试只执行一遍启动和停止的操作。

 

源代码：

package junitbook.coarse.try1; import java.io.IOException; import java.io.OutputStream; import junit.extensions.TestSetup; import junit.framework.Test; import org.mortbay.jetty.*; import org.mortbay.jetty.handler.AbstractHandler; import org.mortbay.jetty.handler.ContextHandler; import org.mortbay.log.Log; import org.mortbay.util.ByteArrayISO8859Writer; import javax.servlet.ServletException; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; public class TestWebClientSetup1 extends TestSetup { protected static Server server; public TestWebClientSetup1(Test suite) { super(suite); } protected void setUp() throws Exception { Log.info("it's starting server..."); Server server = new Server(8080); ContextHandler context = new ContextHandler("/testGetContentOk"); context.setHandler(new TestGetContentOkHandler()); context.setResourceBase("."); server.setHandler(context); Log.info("servering " + context.getBaseResource()); server.start(); } protected void tearDown() throws Exception { Log.info("it's stopping server..."); server.stop(); } private class TestGetContentOkHandler extends AbstractHandler { public void handle(String s, HttpServletRequest request, HttpServletResponse response, int i) throws IOException, ServletException { OutputStream out = response.getOutputStream(); ByteArrayISO8859Writer writer = new ByteArrayISO8859Writer(); //这里处理客户端请求时，返回一个字符串 writer.write("It works"); writer.flush(); response.setContentLength(writer.size()); writer.writeTo(out); out.flush(); //转换成jetty中的request,然后才可以调setHandled Request baseRequest = request instanceof Request ? (Request) request : HttpConnection.getCurrentConnection().getRequest(); baseRequest.setHandled(true);//告诉系统，处理完毕 } } }

 

测试用例：

package junitbook.coarse.try1; import junit.framework.Test; import junit.framework.TestCase; import junit.framework.TestSuite; import java.net.URL; public class TestWebClient1 extends TestCase{ public static Test suite(){ TestSuite suite=new TestSuite(); suite.addTestSuite(TestWebClient1.class); return new TestWebClientSetup1(suite); } public void testGetContentOk() throws Exception{ WebClient wc=new WebClient(); String result=wc.getContent(new URL("http://127.0.0.1:8080/testGetContentOk")); assertEquals("It works",result); } }

注意，WebClient是哪里来的，自定义的，用来处理发出请求的逻辑，一个方便类而已。

package junitbook.coarse.try1; import java.net.URL; import java.net.HttpURLConnection; import java.io.InputStream; import java.io.IOException; public class WebClient { public String getContent(URL url) { StringBuffer content = new StringBuffer(); try { HttpURLConnection connection = (HttpURLConnection) url.openConnection(); connection.setDoInput(true); InputStream is = connection.getInputStream(); byte[] buffer = new byte[2048]; int count; while (-1 != (count = is.read(buffer))) { content.append(new String(buffer, 0, count)); } } catch (IOException e) { return null; } return content.toString(); } public String getContent(String url) throws Exception{ return getContent(new URL(url)); } }

全部代码就是这些，这个例子当然很简单，但足以说明stub方式的运行机制，有时候，为了测试你的业务方法，启动一个真实的服务器如weblogic很费时间，或者你的程序依赖别人的模块，而别人的模块还没有写好，这时stub方式就派上用场了，你embed一个jetty进去，定义一个handler，返回你的业务方法的处理结果，就可以专注于你的功能的测试了。如果你连用jetty也嫌heavy了，那就用自定义的HttpURLConnection去实现吧，呵呵，那个用起来太麻烦，强烈不推荐。