298234 Unit test for jetty-client handling different HTTP error codes

git-svn-id: svn+ssh://dev.eclipse.org/svnroot/rt/org.eclipse.jetty/jetty/trunk@1165 7e9141cc-0065-0410-87d8-b60c137991c4

VERSION.txt

@@ -6,6 +6,7 @@ jetty-7.0.2-SNAPSHOT
  + 297783 Handle HEAD reponses in HttpClient
  + 298144 Unit test for jetty-client connecting to a server that uses Basic Auth
  + 298145 Reorganized test harness to separate the HTTP PUT and HTTP GET test URLs
+ + 298234 Unit test for jetty-client handling different HTTP error codes
  + JETTY-910 Allow request listeners to access session
  + JETTY-1155 HttpConnection.close notifies HttpExchange
  + JETTY-1156 SSL blocking close with JVM Bug busy key fix

jetty-client/src/test/java/org/eclipse/jetty/client/ErrorStatusTest.java

@@ -0,0 +1,185 @@
+// ========================================================================
+// Copyright (c) 2009-2009 Mort Bay Consulting Pty. Ltd.
+// ------------------------------------------------------------------------
+// All rights reserved. This program and the accompanying materials
+// are made available under the terms of the Eclipse Public License v1.0
+// and Apache License v2.0 which accompanies this distribution.
+// The Eclipse Public License is available at 
+// http://www.eclipse.org/legal/epl-v10.html
+// The Apache License v2.0 is available at
+// http://www.opensource.org/licenses/apache2.0.php
+// You may elect to redistribute this code under either of these licenses. 
+// ========================================================================
+
+package org.eclipse.jetty.client;
+
+import java.io.IOException;
+
+import javax.servlet.ServletException;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+import org.eclipse.jetty.http.HttpMethods;
+import org.eclipse.jetty.http.HttpStatus;
+import org.eclipse.jetty.io.ByteArrayBuffer;
+import org.eclipse.jetty.server.Handler;
+import org.eclipse.jetty.server.Request;
+import org.eclipse.jetty.server.Server;
+import org.eclipse.jetty.server.handler.AbstractHandler;
+import org.eclipse.jetty.server.handler.HandlerCollection;
+import org.eclipse.jetty.server.nio.SelectChannelConnector;
+import org.eclipse.jetty.servlet.DefaultServlet;
+import org.eclipse.jetty.servlet.ServletContextHandler;
+import org.eclipse.jetty.servlet.ServletHolder;
+
+public class ErrorStatusTest
+    extends ContentExchangeTest
+{
+    public void testPutBadRequest()
+        throws Exception
+    {
+        doPutFail(HttpStatus.BAD_REQUEST_400);
+    }
+    
+    public void testPutUnauthorized()
+        throws Exception
+    {
+        doPutFail(HttpStatus.UNAUTHORIZED_401);
+    }
+    
+    public void testPutForbidden()
+        throws Exception
+    {
+        doPutFail(HttpStatus.FORBIDDEN_403);
+    }
+    
+    public void testPutNotFound()
+        throws Exception
+    {
+        doPutFail(HttpStatus.NOT_FOUND_404);    
+    }
+    
+    public void testPutServerError()
+        throws Exception
+    {
+        doPutFail(HttpStatus.INTERNAL_SERVER_ERROR_500);    
+    }
+    
+    public void testGetBadRequest()
+        throws Exception
+    {
+        doGetFail(HttpStatus.BAD_REQUEST_400);
+    }
+    
+    public void testGetUnauthorized()
+        throws Exception
+    {
+        doGetFail(HttpStatus.UNAUTHORIZED_401);
+    }
+    
+    public void testGetNotFound()
+        throws Exception
+    {
+        doGetFail(HttpStatus.NOT_FOUND_404);    
+    }
+    
+    public void testGetServerError()
+        throws Exception
+    {
+        doGetFail(HttpStatus.INTERNAL_SERVER_ERROR_500);    
+    }
+    
+    protected void doPutFail(int status)
+        throws Exception
+    {
+        System.err.println(getName());
+    
+        startClient(getRealm());
+
+        ContentExchange putExchange = new ContentExchange();
+        putExchange.setURL(getBaseUrl() + "output.txt");
+        putExchange.setMethod(HttpMethods.PUT);
+        putExchange.setRequestHeader("X-Response-Status",Integer.toString(status));
+        putExchange.setRequestContent(new ByteArrayBuffer(getContent().getBytes()));
+        
+        getClient().send(putExchange);
+        int state = putExchange.waitForDone();
+                
+        int responseStatus = putExchange.getResponseStatus();
+        
+        stopClient();
+
+        assertEquals(status, responseStatus);            
+    }
+    
+    protected void doGetFail(int status)
+        throws Exception
+    {
+        System.err.println(getName());
+    
+        startClient(getRealm());
+
+        ContentExchange getExchange = new ContentExchange();
+        getExchange.setURL(getBaseUrl() + "input.txt");
+        getExchange.setMethod(HttpMethods.GET);
+        getExchange.setRequestHeader("X-Response-Status",Integer.toString(status));
+       
+        getClient().send(getExchange);
+        int state = getExchange.waitForDone();
+        
+        String content;
+        int responseStatus = getExchange.getResponseStatus();
+        if (responseStatus == HttpStatus.OK_200) {
+            content = getExchange.getResponseContent();
+        }
+        
+        stopClient();
+
+        assertEquals(status, responseStatus);
+    }
+    
+    protected void configureServer(Server server)
+        throws Exception
+    {
+        setProtocol("http");
+    
+        SelectChannelConnector connector = new SelectChannelConnector();
+        server.addConnector(connector);
+        
+        ServletContextHandler root = new ServletContextHandler();
+        root.setContextPath("/");
+        root.setResourceBase(getBasePath());
+        ServletHolder servletHolder = new ServletHolder( new DefaultServlet() );
+        servletHolder.setInitParameter( "gzip", "true" );
+        root.addServlet( servletHolder, "/*" );    
+
+        Handler status = new StatusHandler();
+        Handler puthdl = new PutHandler(getBasePath());
+        
+        HandlerCollection handlers = new HandlerCollection();
+        handlers.setHandlers(new Handler[]{status, puthdl, root});
+        server.setHandler( handlers ); 
+    }
+    
+    protected static class StatusHandler extends AbstractHandler {
+        public void handle(String target, Request baseRequest,
+                HttpServletRequest request, HttpServletResponse response)
+            throws IOException, ServletException
+        {
+            if (baseRequest.isHandled())
+                return;
+
+            int statusValue = 0;
+            String statusHeader = request.getHeader("X-Response-Status");
+            if (statusHeader != null)
+            {
+                statusValue = Integer.parseInt(statusHeader);
+            }
+            if (statusValue != 0)
+            {
+                response.setStatus(statusValue);
+                baseRequest.setHandled(true);
+            }
+        }
+    }
+}