1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
//
// ========================================================================
// Copyright (c) 1995-2016 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.rhttp.gateway;
import java.net.URLEncoder;
import java.util.HashMap;
import java.util.concurrent.atomic.AtomicReference;
import javax.servlet.http.HttpServletResponse;
import junit.framework.TestCase;
import org.eclipse.jetty.client.Address;
import org.eclipse.jetty.client.ContentExchange;
import org.eclipse.jetty.client.HttpClient;
import org.eclipse.jetty.client.HttpExchange;
import org.eclipse.jetty.http.HttpMethods;
import org.eclipse.jetty.io.ByteArrayBuffer;
import org.eclipse.jetty.rhttp.client.JettyClient;
import org.eclipse.jetty.rhttp.client.RHTTPClient;
import org.eclipse.jetty.rhttp.client.RHTTPListener;
import org.eclipse.jetty.rhttp.client.RHTTPRequest;
import org.eclipse.jetty.rhttp.client.RHTTPResponse;
import org.eclipse.jetty.rhttp.gateway.GatewayServer;
import org.eclipse.jetty.rhttp.gateway.StandardGateway;
import org.eclipse.jetty.server.Connector;
import org.eclipse.jetty.server.nio.SelectChannelConnector;
/**
* @version $Revision$ $Date$
*/
public class GatewayTimeoutTest extends TestCase
{
/**
* Tests a forwarded request that lasts longer than the gateway timeout.
* The gateway client will perform 2 long polls before the forwarded request's response is returned.
*
* @throws Exception in case of test exceptions
*/
public void testGatewayTimeout() throws Exception
{
GatewayServer server = new GatewayServer();
Connector connector = new SelectChannelConnector();
server.addConnector(connector);
final long gatewayTimeout = 5000L;
((StandardGateway)server.getGateway()).setGatewayTimeout(gatewayTimeout);
final long externalTimeout = gatewayTimeout * 2;
((StandardGateway)server.getGateway()).setExternalTimeout(externalTimeout);
server.start();
try
{
Address address = new Address("localhost", connector.getLocalPort());
HttpClient httpClient = new HttpClient();
httpClient.setConnectorType(HttpClient.CONNECTOR_SELECT_CHANNEL);
httpClient.start();
try
{
String targetId = "1";
final RHTTPClient client = new JettyClient(httpClient, address, server.getContext().getContextPath()+GatewayServer.DFT_CONNECT_PATH, targetId);
final AtomicReference<Exception> exceptionRef = new AtomicReference<Exception>();
client.addListener(new RHTTPListener()
{
public void onRequest(RHTTPRequest request)
{
try
{
// Wait until gateway timeout expires
Thread.sleep(gatewayTimeout + 1000L);
RHTTPResponse response = new RHTTPResponse(request.getId(), 200, "OK", new HashMap<String, String>(), request.getBody());
client.deliver(response);
}
catch (Exception x)
{
exceptionRef.set(x);
}
}
});
client.connect();
try
{
// Make a request to the gateway and check response
ContentExchange exchange = new ContentExchange(true);
exchange.setMethod(HttpMethods.POST);
exchange.setAddress(address);
exchange.setURI(server.getContext().getContextPath()+GatewayServer.DFT_EXT_PATH + "/" + URLEncoder.encode(targetId, "UTF-8"));
String requestContent = "body";
exchange.setRequestContent(new ByteArrayBuffer(requestContent.getBytes("UTF-8")));
httpClient.send(exchange);
int status = exchange.waitForDone();
assertEquals(HttpExchange.STATUS_COMPLETED, status);
assertEquals(HttpServletResponse.SC_OK, exchange.getResponseStatus());
assertNull(exceptionRef.get());
String responseContent = exchange.getResponseContent();
assertEquals(responseContent, requestContent);
}
finally
{
client.disconnect();
}
}
finally
{
httpClient.stop();
}
}
finally
{
server.stop();
}
}
}
|
