juplo.de Git - percentcodec/blob - cachecontrol/src/test/java/de/halbekunst/juplo/cachecontrol/HttpTestCase.java

   1 package de.halbekunst.juplo.cachecontrol;
   2
   3 import com.meterware.httpunit.WebResponse;
   4 import com.meterware.servletunit.InvocationContext;
   5 import com.meterware.servletunit.ServletRunner;
   6 import com.meterware.servletunit.ServletUnitClient;
   7 import java.io.IOException;
   8 import java.io.PrintWriter;
   9 import java.util.Enumeration;
  10 import java.util.HashMap;
  11 import java.util.Map;
  12 import javax.servlet.ServletOutputStream;
  13 import javax.servlet.http.HttpServletRequest;
  14 import javax.servlet.http.HttpServletResponse;
  15 import javax.servlet.http.HttpServletResponseWrapper;
  16 import org.junit.Before;
  17 import org.slf4j.Logger;
  18 import org.slf4j.LoggerFactory;
  19
  20 /**
  21  *
  22  * @author kai
  23  */
  24 public abstract class HttpTestCase {
  25   private final static Logger log = LoggerFactory.getLogger(HttpTestCase.class);
  26
  27   private ServletRunner sr;
  28   ServletUnitClient client;
  29   int buffer = 2048;
  30
  31
  32   @Before
  33   public void init() throws Exception {
  34     sr = new ServletRunner(ParameterGuessingTest.class.getResourceAsStream("/web.xml"));
  35     client = sr.newClient();
  36   }
  37
  38   protected WebResponse executeRequest(String uri) throws Exception {
  39     log.debug("---------- GET: {}", uri);
  40     InvocationContext invocation = client.newInvocation(uri);
  41     HttpServletRequest request = invocation.getRequest();
  42     log.debug("Request - {}: {}", request.getMethod(), request.getProtocol());
  43     Enumeration<String> headers = request.getHeaderNames();
  44     while (headers.hasMoreElements()) {
  45       String header = headers.nextElement();
  46       Enumeration<String> values = request.getHeaders(header);
  47       while (values.hasMoreElements())
  48         log.debug("Request - {}: {}", header, values.nextElement());
  49     }
  50     log.debug("Invocing service method.");
  51
  52     /**
  53      * We cannot call invocation.service(), because we have to wrap the
  54      * response. Therefore this was coppied from InvocationContextImpl.
  55      */
  56     TestHttpServletResponse wrappedResponse = new TestHttpServletResponse(invocation.getResponse());
  57     if (invocation.isFilterActive()) {
  58       invocation.getFilter().doFilter(invocation.getRequest(), wrappedResponse, invocation.getFilterChain());
  59     }
  60     else {
  61       invocation.getServlet().service(invocation.getRequest(), wrappedResponse);
  62     }
  63
  64     WebResponse response = client.getResponse(invocation);
  65     log.debug("Response - {}: {}", response.getResponseCode(), response.getResponseMessage());
  66     log.debug("Response - {}, {} bytes", response.getContentType(), wrappedResponse.getCount());
  67     for (String header : response.getHeaderFieldNames()) {
  68       for (String value : response.getHeaderFields(header)) {
  69         log.debug("Response - {}: {}", header, value);
  70       }
  71     }
  72     return response;
  73   }
  74
  75
  76   class TestHttpServletResponse extends HttpServletResponseWrapper {
  77
  78     private CountingServletOutputStream out;
  79     private HttpServletResponse response;
  80     private ServletOutputStream stream;
  81     private PrintWriter writer;
  82     private boolean committed = false;
  83
  84
  85     TestHttpServletResponse(HttpServletResponse response) {
  86       super(response);
  87       this.response = response;
  88     }
  89
  90
  91     public long getCount() {
  92       if (out == null)
  93         return -1l;
  94       else
  95         return out.count;
  96     }
  97
  98
  99     @Override
 100     public void flushBuffer() throws IOException {
 101       committed = true;
 102       super.flushBuffer();
 103     }
 104
 105     @Override
 106     public int getBufferSize() {
 107       return buffer;
 108     }
 109
 110     @Override
 111     public boolean isCommitted() {
 112       return committed;
 113     }
 114
 115     @Override
 116     public void reset() {
 117       if (committed)
 118         throw new IllegalStateException("call to reset() after response has been commited!");
 119       if (out != null)
 120         out.count = 0;
 121       super.reset();
 122     }
 123
 124     @Override
 125     public void resetBuffer() {
 126       if (committed)
 127         throw new IllegalStateException("call to resetBuffer() after response has been commited!");
 128       if (out != null)
 129         out.count = 0;
 130       super.resetBuffer();
 131     }
 132
 133     @Override
 134     public void setBufferSize(int size) {
 135       if (out != null && out.count > 0)
 136         throw new IllegalStateException("call to setBuffer() after content has been written!");
 137       buffer = size;
 138     }
 139
 140     @Override
 141     public ServletOutputStream getOutputStream() throws IOException {
 142
 143       if (writer != null)
 144         throw new IllegalStateException("ServletOutputStream and PrintWriter cannot be requested both!");
 145
 146       if (stream == null) {
 147         out = new CountingServletOutputStream(response.getOutputStream());
 148         stream = out;
 149       }
 150
 151       return stream;
 152     }
 153
 154     @Override
 155     public PrintWriter getWriter() throws IOException {
 156
 157       if (stream != null)
 158         throw new IllegalStateException("ServletOutputStream and PrintWriter cannot be requested both!");
 159
 160       if (writer == null) {
 161         out = new CountingServletOutputStream(response.getOutputStream());
 162         writer = new PrintWriter(out);
 163       }
 164
 165       return writer;
 166     }
 167
 168
 169     class CountingServletOutputStream extends ServletOutputStream {
 170
 171       private ServletOutputStream out;
 172       long count = 0l;
 173
 174
 175       CountingServletOutputStream(ServletOutputStream out) {
 176         this.out = out;
 177       }
 178
 179
 180       @Override
 181       public void write(int i) throws IOException {
 182         count++;
 183         /** Simulate commit, when count is getting bigger then buffer */
 184         if (count == buffer + 1) {
 185           log.debug("simulating commit because buffer overflow! buffer: {}, count: {}", buffer, count);
 186           committed = true;
 187         }
 188         out.write(i);
 189       }
 190     }
 191   }
 192 }
 193