package org.jboss.resteasy.test.form;
import org.jboss.resteasy.annotations.Form;
import org.jboss.resteasy.client.ClientRequest;
import org.jboss.resteasy.client.ClientResponse;
import org.jboss.resteasy.client.ProxyFactory;
import org.jboss.resteasy.logging.Logger;
import org.jboss.resteasy.specimpl.MultivaluedMapImpl;
import org.jboss.resteasy.test.BaseResourceTest;
import org.jboss.resteasy.util.GenericType;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import javax.servlet.http.HttpServletResponse;
import javax.ws.rs.GET;
import javax.ws.rs.HeaderParam;
import javax.ws.rs.POST;
import javax.ws.rs.Path;
import javax.ws.rs.Produces;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.MultivaluedMap;
import java.io.BufferedInputStream;
import java.io.InputStream;
import java.net.URLDecoder;
import java.util.HashMap;
import java.util.Map;
import static org.jboss.resteasy.test.TestPortProvider.generateBaseUrl;
import static org.jboss.resteasy.test.TestPortProvider.generateURL;
/**
* A TestFormResource.
*
* @author <a href="ryan@damnhandy.com">Ryan J. McDonough</a>
* @version $Revision:$
*/
public class TestFormResource extends BaseResourceTest
{
private static final String SHORT_VALUE_FIELD = "shortValue";
private static final String INTEGER_VALUE_FIELD = "integerValue";
private static final String LONG_VALUE_FIELD = "longValue";
private static final String DOUBLE_VALUE_FIELD = "doubleValue";
private static final String NAME_FIELD = "name";
private static final String BOOLEAN_VALUE_FIELD = "booleanValue";
private static final String TEST_URI = generateURL("/form/42?query=42");
@SuppressWarnings("unused")
private static final Logger logger = Logger.getLogger(TestFormResource.class);
@Path("/form/{id}")
public interface FormClientProxy
{
@Produces(MediaType.APPLICATION_FORM_URLENCODED)
@POST
MultivaluedMap<String, String> post(@Form ClientForm form);
@Produces(MediaType.APPLICATION_FORM_URLENCODED)
@POST
String postString(@Form ClientForm form);
}
public static class ClientForm2
{
@HeaderParam("custom-header")
protected String foo;
public String getFoo()
{
return foo;
}
public void setFoo(String foo)
{
this.foo = foo;
}
}
@Path("/myform")
public interface MyFormProxy
{
@POST
public void post(@Form ClientForm2 form2);
}
@Path("/myform")
public static class MyFormResource
{
@GET
@Path("/server")
@Produces("application/x-www-form-urlencoded")
public MultivaluedMap<String, String> retrieveServername()
{
MultivaluedMap<String, String> serverMap = new MultivaluedMapImpl<String, String>();
serverMap.add("servername", "srv1");
serverMap.add("servername", "srv2");
return serverMap;
}
@POST
public void post()
{
}
}
/**
* FIXME Comment this
*
* @throws java.lang.Exception
*/
@Before
public void setUp() throws Exception
{
addPerRequestResource(FormResource.class);
addPerRequestResource(MyFormResource.class);
}
/**
* RESTEASY-261
*
* @throws Exception
*/
@Test
public void testMultiValueParam() throws Exception
{
ClientRequest request = new ClientRequest(generateURL("/myform/server"));
ClientResponse<MultivaluedMap<String, String>> response = request.get(new GenericType<MultivaluedMap<String, String>>()
{
});
int status = response.getStatus();
Assert.assertEquals(200, status);
boolean sv1 = false;
boolean sv2 = false;
MultivaluedMap<String, String> form = response.getEntity();
Assert.assertEquals(2, form.get("servername").size());
for (String str : form.get("servername"))
{
if (str.equals("srv1")) sv1 = true;
else if (str.equals("srv2")) sv2 = true;
}
Assert.assertTrue(sv1);
Assert.assertTrue(sv2);
}
/**
* RESTEASY-691
*
* @throws Exception
*/
@Test
public void testProxy691() throws Exception
{
MyFormProxy proxy = ProxyFactory.create(MyFormProxy.class, generateBaseUrl());
proxy.post(null);
}
@Test
public void testProxy() throws Exception
{
FormClientProxy proxy = ProxyFactory.create(FormClientProxy.class, generateBaseUrl());
ClientForm form = new ClientForm();
form.setBooleanValue(true);
form.setName("This is My Name");
form.setDoubleValue(123.45);
form.setLongValue(566780L);
form.setIntegerValue(3);
form.setShortValue((short) 12345);
form.setHeaderParam(42);
form.setQueryParam(42);
form.setId(42);
MultivaluedMap<String, String> rtn = proxy.post(form);
Assert.assertEquals(rtn.getFirst(BOOLEAN_VALUE_FIELD), "true");
Assert.assertEquals(rtn.getFirst(NAME_FIELD), "This is My Name");
Assert.assertEquals(rtn.getFirst(DOUBLE_VALUE_FIELD), "123.45");
Assert.assertEquals(rtn.getFirst(LONG_VALUE_FIELD), "566780");
Assert.assertEquals(rtn.getFirst(INTEGER_VALUE_FIELD), "3");
Assert.assertEquals(rtn.getFirst(SHORT_VALUE_FIELD), "12345");
String str = proxy.postString(form);
System.out.println("STR: " + str);
String[] params = str.split("&");
Map<String, String> map = new HashMap<String, String>();
for (int i = 0; i < params.length; i++)
{
int index = params[i].indexOf('=');
String key = params[i].substring(0, index).trim();
String value = params[i].substring(index + 1).trim().replace('+', ' ');
map.put(key, value);
}
Assert.assertEquals(map.get(BOOLEAN_VALUE_FIELD), "true");
Assert.assertEquals(map.get(NAME_FIELD), "This is My Name");
Assert.assertEquals(map.get(DOUBLE_VALUE_FIELD), "123.45");
Assert.assertEquals(map.get(LONG_VALUE_FIELD), "566780");
Assert.assertEquals(map.get(INTEGER_VALUE_FIELD), "3");
Assert.assertEquals(map.get(SHORT_VALUE_FIELD), "12345");
}
@Test
public void testFormResource() throws Exception
{
InputStream in = null;
try
{
ClientRequest request = new ClientRequest(TEST_URI);
request.header("custom-header", "42");
request.formParameter(BOOLEAN_VALUE_FIELD, "true");
request.formParameter(NAME_FIELD, "This is My Name");
request.formParameter(DOUBLE_VALUE_FIELD, "123.45");
request.formParameter(LONG_VALUE_FIELD, "566780");
request.formParameter(INTEGER_VALUE_FIELD, "3");
request.formParameter(SHORT_VALUE_FIELD, "12345");
ClientResponse<InputStream> response = request.post(InputStream.class);
Assert.assertEquals(HttpServletResponse.SC_OK, response.getStatus());
String contentType = response.getResponseHeaders().getFirst("content-type");
Assert.assertEquals("application/x-www-form-urlencoded", contentType);
InputStream responseStream = response.getEntity();
in = new BufferedInputStream(responseStream);
String formData = readString(in);
String[] keys = formData.split("&");
Map<String, String> values = new HashMap<String, String>();
for (String pair : keys)
{
int index = pair.indexOf('=');
if (index < 0)
{
values.put(URLDecoder.decode(pair, "UTF-8"), null);
}
else if (index > 0)
{
values.put(URLDecoder.decode(pair.substring(0, index), "UTF-8"), URLDecoder.decode(pair
.substring(index + 1), "UTF-8"));
}
}
Assert.assertEquals(values.get(BOOLEAN_VALUE_FIELD), "true");
Assert.assertEquals(values.get(NAME_FIELD), "This is My Name");
Assert.assertEquals(values.get(DOUBLE_VALUE_FIELD), "123.45");
Assert.assertEquals(values.get(LONG_VALUE_FIELD), "566780");
Assert.assertEquals(values.get(INTEGER_VALUE_FIELD), "3");
}
finally
{
if (in != null)
{
in.close();
}
}
}
static class StringPair
{
public String key;
public String value;
public StringPair(String key, String value)
{
this.key = key;
this.value = value;
}
}
}