Small updates to ProxyServletTest.

This commit is contained in:
Simone Bordet 2015-02-12 14:27:09 +01:00
parent c449f1fa20
commit 968ed9f0ae
1 changed files with 16 additions and 2 deletions

View File

@ -173,7 +173,10 @@ public class ProxyServletTest
private HttpClient prepareClient() throws Exception private HttpClient prepareClient() throws Exception
{ {
QueuedThreadPool clientPool = new QueuedThreadPool();
clientPool.setName("client");
HttpClient result = new HttpClient(); HttpClient result = new HttpClient();
result.setExecutor(clientPool);
result.getProxyConfiguration().getProxies().add(new HttpProxy("localhost", proxyConnector.getLocalPort())); result.getProxyConfiguration().getProxies().add(new HttpProxy("localhost", proxyConnector.getLocalPort()));
result.start(); result.start();
return result; return result;
@ -305,8 +308,19 @@ public class ProxyServletTest
@Override @Override
protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException
{ {
if (req.getHeader("Via") != null) try
resp.addHeader(PROXIED_HEADER, "true"); {
// Give some time to the proxy to
// upload the content to the server.
Thread.sleep(1000);
if (req.getHeader("Via") != null)
resp.addHeader(PROXIED_HEADER, "true");
}
catch (InterruptedException x)
{
throw new InterruptedIOException();
}
} }
}); });
startProxy(); startProxy();