package org.jboss.resteasy.test;
import org.jboss.resteasy.client.jaxrs.ResteasyClient;
import org.jboss.resteasy.client.jaxrs.ResteasyClientBuilder;
import org.jboss.resteasy.client.jaxrs.ResteasyWebTarget;
import org.jboss.resteasy.client.jaxrs.internal.ClientInvocation;
import org.jboss.resteasy.plugins.server.vertx.VertxContainer;
import org.junit.AfterClass;
import org.junit.Assert;
import org.junit.BeforeClass;
import org.junit.Test;
import javax.ws.rs.GET;
import javax.ws.rs.HttpMethod;
import javax.ws.rs.POST;
import javax.ws.rs.Path;
import javax.ws.rs.Produces;
import javax.ws.rs.QueryParam;
import javax.ws.rs.client.Client;
import javax.ws.rs.client.ClientBuilder;
import javax.ws.rs.client.Entity;
import javax.ws.rs.client.WebTarget;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.GenericType;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.UriInfo;
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.Socket;
import java.util.Locale;
import static org.jboss.resteasy.test.TestPortProvider.generateURL;
import static org.jboss.resteasy.test.TestPortProvider.getHost;
import static org.jboss.resteasy.test.TestPortProvider.getPort;
/**
* @author <a href="mailto:bill@burkecentral.com">Bill Burke</a>
* @version $Revision: 1 $
*/
public class VertxTest
{
@Path("/")
public static class Resource
{
@GET
@Path("/test")
@Produces("text/plain")
public String hello()
{
return "hello world";
}
@GET
@Path("empty")
public void empty()
{
}
@GET
@Path("query")
public String query(@QueryParam("param") String value)
{
return value;
}
@GET
@Path("/exception")
@Produces("text/plain")
public String exception()
{
throw new RuntimeException();
}
@GET
@Path("large")
@Produces("text/plain")
public String large()
{
StringBuffer buf = new StringBuffer();
for (int i = 0; i < 1000; i++)
{
buf.append(i);
}
return buf.toString();
}
@GET
@Path("/context")
@Produces("text/plain")
public String context(
@Context io.vertx.core.Context context,
@Context io.vertx.core.Vertx vertx,
@Context io.vertx.core.http.HttpServerRequest req,
@Context io.vertx.core.http.HttpServerResponse resp)
{
if (context != null && vertx != null && req != null && resp != null)
{
return "pass";
} else
{
return "fail";
}
}
@POST
@Path("/post")
@Produces("text/plain")
public String post(String postBody)
{
return postBody;
}
@GET
@Path("/test/absolute")
@Produces("text/plain")
public String absolute(@Context UriInfo info)
{
return "uri: " + info.getRequestUri().toString();
}
}
static Client client;
@BeforeClass
public static void setup() throws Exception
{
VertxContainer.start().getRegistry().addPerRequestResource(Resource.class);
client = ClientBuilder.newClient();
}
@AfterClass
public static void end() throws Exception
{
try
{
client.close();
} catch (Exception e)
{
}
VertxContainer.stop();
}
@Test
public void testBasic() throws Exception
{
WebTarget target = client.target(generateURL("/test"));
String val = target.request().get(String.class);
Assert.assertEquals("hello world", val);
}
@Test
public void testHeadContentLength() throws Exception
{
ResteasyClient client = new ResteasyClientBuilder().build();
ResteasyWebTarget target = client.target(generateURL("/test"));
Response getResponse = target.request().buildGet().invoke();
String val = ClientInvocation.extractResult(new GenericType<String>(String.class), getResponse, null);
Assert.assertEquals("hello world", val);
Assert.assertEquals("chunked", getResponse.getHeaderString("transfer-encoding"));
Response headResponse = target.request().build(HttpMethod.HEAD).invoke();
Assert.assertNull(headResponse.getHeaderString("Content-Length"));
Assert.assertNull(headResponse.getHeaderString("transfer-encoding"));
}
@Test
public void testQuery() throws Exception
{
WebTarget target = client.target(generateURL("/query"));
String val = target.queryParam("param", "val").request().get(String.class);
Assert.assertEquals("val", val);
}
@Test
public void testEmpty() throws Exception
{
WebTarget target = client.target(generateURL("/empty"));
Response response = target.request().get();
try
{
Assert.assertEquals(204, response.getStatus());
} finally
{
response.close();
}
}
@Test
public void testLarge() throws Exception
{
WebTarget target = client.target(generateURL("/large"));
Response response = target.request().get();
try
{
Assert.assertEquals(200, response.getStatus());
StringBuffer buf = new StringBuffer();
for (int i = 0; i < 1000; i++)
{
buf.append(i);
}
String expected = buf.toString();
String have = response.readEntity(String.class);
Assert.assertEquals(expected, have);
} finally
{
response.close();
}
}
@Test
public void testUnhandledException() throws Exception
{
WebTarget target = client.target(generateURL("/exception"));
Response resp = target.request().get();
try
{
Assert.assertEquals(500, resp.getStatus());
} finally
{
resp.close();
}
}
@Test
public void testChannelContext() throws Exception
{
WebTarget target = client.target(generateURL("/context"));
String val = target.request().get(String.class);
Assert.assertEquals("pass", val);
}
@Test
public void testPost()
{
WebTarget target = client.target(generateURL("/post"));
String postBody = "hello world";
String result = (String) target.request().post(Entity.text(postBody), String.class);
Assert.assertEquals(postBody, result);
}
/**
* Per the HTTP spec, we must allow requests like:
* <p>
* <pre>
* GET http://www.example.com/content HTTP/1.1
* Host: www.example.com
* </pre>
* <p>
* <blockquote>
* RFC 2616 5.1.12:
* To allow for transition to absoluteURIs in all requests in future
* versions of HTTP, all HTTP/1.1 servers MUST accept the absoluteURI
* form in requests, even though HTTP/1.1 clients will only generate
* them in requests to proxies.
* </blockquote>
*
* @throws Exception
*/
@Test
public void testAbsoluteURI() throws Exception
{
String uri = generateURL("/test/absolute");
Socket client = new Socket(getHost(), getPort());
PrintWriter out = new PrintWriter(client.getOutputStream(), true);
BufferedReader in = new BufferedReader(new InputStreamReader(client.getInputStream()));
out.printf(Locale.US, "GET %s HTTP/1.1\nHost: %s:%d\n\n", uri, getHost(), getPort());
String statusLine = in.readLine();
String response = in.readLine();
while (!response.startsWith("uri"))
{
response = in.readLine();
}
client.close();
Assert.assertEquals("HTTP/1.1 200 OK", statusLine);
Assert.assertEquals(uri, response.subSequence(5, response.length()));
}
}