/*
* Copyright (c) MuleSoft, Inc. All rights reserved. http://www.mulesoft.com
* The software in this package is published under the terms of the CPAL v1.0
* license, a copy of which has been included with this distribution in the
* LICENSE.txt file.
*/
package org.mule.transport.http.transformers;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.assertArrayEquals;
import java.io.Serializable;
import org.mule.DefaultMessageCollection;
import org.mule.RequestContext;
import org.mule.api.MuleEvent;
import org.mule.api.MuleMessage;
import org.mule.api.MuleMessageCollection;
import org.mule.api.config.MuleProperties;
import org.mule.api.endpoint.InboundEndpoint;
import org.mule.tck.junit4.AbstractMuleContextTestCase;
import org.mule.transport.NullPayload;
import org.mule.transport.http.HttpConnector;
import org.mule.transport.http.HttpConstants;
import org.mule.transport.http.HttpRequest;
import org.mule.transport.http.RequestLine;
import org.apache.commons.httpclient.HttpMethod;
import org.apache.commons.httpclient.HttpVersion;
import org.apache.commons.httpclient.methods.ByteArrayRequestEntity;
import org.apache.commons.httpclient.methods.GetMethod;
import org.apache.commons.httpclient.methods.PostMethod;
import org.apache.commons.httpclient.methods.PutMethod;
import org.apache.commons.lang.SerializationUtils;
import org.junit.Test;
public class ObjectToHttpClientMethodRequestTestCase extends AbstractMuleContextTestCase
{
private InboundEndpoint endpoint;
private MuleMessage setupRequestContext(final String url, final String method) throws Exception
{
HttpRequest request = new HttpRequest(new RequestLine(method, url, HttpVersion.HTTP_1_1), null, "UTF-8");
endpoint = muleContext.getEndpointFactory().getInboundEndpoint(url);
MuleEvent event = getTestEvent(request, endpoint);
MuleMessage message = event.getMessage();
message.setOutboundProperty(HttpConnector.HTTP_METHOD_PROPERTY, method);
message.setOutboundProperty(MuleProperties.MULE_ENDPOINT_PROPERTY, url);
RequestContext.setEvent(event);
return message;
}
private MuleMessageCollection setupRequestContextForCollection(final String url, final String method) throws Exception
{
HttpRequest request = new HttpRequest(new RequestLine(method, url, HttpVersion.HTTP_1_1), null, "UTF-8");
endpoint = muleContext.getEndpointFactory().getInboundEndpoint(url);
MuleEvent event = getTestEvent(request, endpoint);
MuleMessageCollection message = new DefaultMessageCollection(muleContext);
message.setOutboundProperty(HttpConnector.HTTP_METHOD_PROPERTY, method);
message.setOutboundProperty(MuleProperties.MULE_ENDPOINT_PROPERTY, url);
RequestContext.setEvent(event);
return message;
}
private ObjectToHttpClientMethodRequest createTransformer() throws Exception
{
ObjectToHttpClientMethodRequest transformer = new ObjectToHttpClientMethodRequest();
transformer.setMuleContext(muleContext);
transformer.setEndpoint(endpoint);
transformer.initialise();
return transformer;
}
@Override
protected void doTearDown() throws Exception
{
RequestContext.setEvent(null);
}
@Test
public void testUrlWithoutQuery() throws Exception
{
MuleMessage message = setupRequestContext("http://localhost:8080/services", HttpConstants.METHOD_GET);
// transforming NullPayload will make sure that no body=xxx query is added
message.setPayload(NullPayload.getInstance());
ObjectToHttpClientMethodRequest transformer = createTransformer();
Object response = transformer.transform(message);
assertTrue(response instanceof HttpMethod);
HttpMethod httpMethod = (HttpMethod) response;
assertEquals(null, httpMethod.getQueryString());
}
@Test
public void testUrlWithQuery() throws Exception
{
MuleMessage message = setupRequestContext("http://localhost:8080/services?method=echo", HttpConstants.METHOD_GET);
// transforming NullPayload will make sure that no body=xxx query is added
message.setPayload(NullPayload.getInstance());
ObjectToHttpClientMethodRequest transformer = createTransformer();
Object response = transformer.transform(message);
assertTrue(response instanceof HttpMethod);
HttpMethod httpMethod = (HttpMethod) response;
assertEquals("method=echo", httpMethod.getQueryString());
}
@Test
public void testUrlWithUnescapedQuery() throws Exception
{
MuleMessage message = setupRequestContext("http://mycompany.com/test?fruits=apple%20orange", HttpConstants.METHOD_GET);
// transforming NullPayload will make sure that no body=xxx query is added
message.setPayload(NullPayload.getInstance());
ObjectToHttpClientMethodRequest transformer = createTransformer();
Object response = transformer.transform(message);
assertTrue(response instanceof HttpMethod);
HttpMethod httpMethod = (HttpMethod) response;
assertEquals("fruits=apple%20orange", httpMethod.getQueryString());
}
@Test
public void testAppendedUrl() throws Exception
{
MuleMessage message = setupRequestContext("http://mycompany.com/test?fruits=apple%20orange", HttpConstants.METHOD_GET);
// transforming a payload here will add it as body=xxx query parameter
message.setPayload("test");
message.setOutboundProperty(HttpConnector.HTTP_GET_BODY_PARAM_PROPERTY, "body");
ObjectToHttpClientMethodRequest transformer = createTransformer();
Object response = transformer.transform(message);
assertTrue(response instanceof HttpMethod);
HttpMethod httpMethod = (HttpMethod) response;
assertEquals("fruits=apple%20orange&body=test", httpMethod.getQueryString());
}
@Test
public void testEncodingOfParamValueTriggeredByMessageProperty() throws Exception
{
// the payload is already encoded, switch off encoding it in the transformer
String encodedPayload = "encoded%20payload";
MuleMessage message = setupRequestContext("http://mycompany.com/", "GET");
message.setOutboundProperty(HttpConnector.HTTP_ENCODE_PARAMVALUE, false);
message.setOutboundProperty(HttpConnector.HTTP_GET_BODY_PARAM_PROPERTY, "body");
message.setPayload(encodedPayload);
ObjectToHttpClientMethodRequest transformer = createTransformer();
Object result = transformer.transform(message);
assertTrue(result instanceof GetMethod);
String expected = "body=" + encodedPayload;
assertEquals(expected, ((GetMethod) result).getQueryString());
}
public void testPostMethod() throws Exception
{
final MuleMessage message = setupRequestContext("http://localhost:8080/services", HttpConstants.METHOD_POST);
final String contentType = "text/plain";
message.setPayload("I'm a payload");
message.setInvocationProperty(HttpConstants.HEADER_CONTENT_TYPE, contentType);
final ObjectToHttpClientMethodRequest transformer = createTransformer();
final Object response = transformer.transform(message);
assertTrue(response instanceof PostMethod);
final HttpMethod httpMethod = (HttpMethod) response;
assertEquals(null, httpMethod.getQueryString());
assertEquals(contentType, httpMethod.getRequestHeader(HttpConstants.HEADER_CONTENT_TYPE).getValue());
}
public void testPutMethod() throws Exception
{
final MuleMessage message = setupRequestContext("http://localhost:8080/services", HttpConstants.METHOD_PUT);
final String contentType = "text/plain";
message.setPayload("I'm a payload");
message.setInvocationProperty(HttpConstants.HEADER_CONTENT_TYPE, contentType);
final ObjectToHttpClientMethodRequest transformer = createTransformer();
final Object response = transformer.transform(message);
assertTrue(response instanceof PutMethod);
final HttpMethod httpMethod = (HttpMethod) response;
assertEquals(null, httpMethod.getQueryString());
assertEquals(contentType, httpMethod.getRequestHeader(HttpConstants.HEADER_CONTENT_TYPE).getValue());
}
@Test
public void testPostMethodWithHttp10ForMuleMessage() throws Exception
{
final MuleMessage message = setupRequestContext("http://localhost:8080/services", HttpConstants.METHOD_POST);
final String contentType = "text/plain";
String payload = "I'm a payload";
message.setPayload(payload);
message.setInvocationProperty(HttpConstants.HEADER_CONTENT_TYPE, contentType);
message.setOutboundProperty(HttpConnector.HTTP_VERSION_PROPERTY, HttpConstants.HTTP10);
final ObjectToHttpClientMethodRequest transformer = createTransformer();
final Object response = transformer.transform(message);
assertTrue(response instanceof PostMethod);
final HttpMethod httpMethod = (HttpMethod) response;
assertEquals(null, httpMethod.getQueryString());
final byte[] byteArrayContent = ((ByteArrayRequestEntity)((PostMethod)httpMethod).getRequestEntity()).getContent();
assertArrayEquals(payload.getBytes(), byteArrayContent);
}
@Test
public void testPostMethodWithHttp10ForMuleMessageCollection() throws Exception
{
final MuleMessageCollection message = setupRequestContextForCollection("http://localhost:8080/services", HttpConstants.METHOD_POST);
MuleMessage messageOne = setupRequestContext("http://localhost:8080/services", HttpConstants.METHOD_POST);
message.addMessage(messageOne);
final String contentType = "text/plain";
String payload = "I'm a payload";
messageOne.setPayload(payload);
message.setInvocationProperty(HttpConstants.HEADER_CONTENT_TYPE, contentType);
message.setOutboundProperty(HttpConnector.HTTP_VERSION_PROPERTY, HttpConstants.HTTP10);
final ObjectToHttpClientMethodRequest transformer = createTransformer();
final Object response = transformer.transform(message);
assertTrue(response instanceof PostMethod);
final HttpMethod httpMethod = (HttpMethod) response;
assertEquals(null, httpMethod.getQueryString());
final byte[] byteArrayContent = ((ByteArrayRequestEntity)((PostMethod)httpMethod).getRequestEntity()).getContent();
final byte[] expectedByteArrayContent = SerializationUtils.serialize((Serializable) message.getPayload());
assertArrayEquals(expectedByteArrayContent, byteArrayContent);
}
}