/*
* $Id: HttpPersistentQueueTestCase.java 21939 2011-05-18 13:32:09Z aperepel $
* --------------------------------------------------------------------------------------
* 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.functional;
import org.mule.api.MuleEventContext;
import org.mule.api.MuleMessage;
import org.mule.tck.DynamicPortTestCase;
import org.mule.tck.functional.EventCallback;
import org.mule.tck.functional.FunctionalTestComponent;
import org.mule.transport.http.HttpConstants;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
import org.apache.commons.httpclient.HttpClient;
import org.apache.commons.httpclient.HttpMethod;
import org.apache.commons.httpclient.HttpStatus;
import org.apache.commons.httpclient.NameValuePair;
import org.apache.commons.httpclient.methods.GetMethod;
import org.apache.commons.httpclient.methods.PostMethod;
public class HttpPersistentQueueTestCase extends DynamicPortTestCase
{
private CountDownLatch messageDidArrive = new CountDownLatch(1);
private int port = -1;
@Override
protected String getConfigResources()
{
return "http-persistent-queue.xml";
}
@Override
protected void doSetUp() throws Exception
{
super.doSetUp();
FunctionalTestComponent testComponent = (FunctionalTestComponent) getComponent("PersistentQueueAsync");
assertNotNull(testComponent);
testComponent.setEventCallback(new Callback(messageDidArrive));
port = getPorts().get(0);
}
public void testPersistentMessageDeliveryWithGet() throws Exception
{
GetMethod method = new GetMethod("http://localhost:" + port + "/services/Echo?foo=bar");
method.addRequestHeader(HttpConstants.HEADER_CONNECTION, "close");
doTestPersistentMessageDelivery(method);
}
public void testPersistentMessageDeliveryWithPost() throws Exception
{
PostMethod method = new PostMethod("http://localhost:" + port + "/services/Echo");
method.addRequestHeader(HttpConstants.HEADER_CONNECTION, "close");
method.addParameter(new NameValuePair("foo", "bar"));
doTestPersistentMessageDelivery(method);
}
private void doTestPersistentMessageDelivery(HttpMethod httpMethod) throws Exception
{
HttpClient client = new HttpClient();
int rc = client.executeMethod(httpMethod);
assertEquals(HttpStatus.SC_OK, rc);
assertTrue(messageDidArrive.await(RECEIVE_TIMEOUT, TimeUnit.MILLISECONDS));
}
private static class Callback implements EventCallback
{
private CountDownLatch messageDidArrive;
public Callback(CountDownLatch latch)
{
super();
messageDidArrive = latch;
}
public void eventReceived(MuleEventContext context, Object component) throws Exception
{
MuleMessage message = context.getMessage();
Object httpMethod = message.getInboundProperty("http.method");
if (HttpConstants.METHOD_GET.equals(httpMethod))
{
assertEquals("/services/Echo?foo=bar", message.getPayloadAsString());
}
else if (HttpConstants.METHOD_POST.equals(httpMethod))
{
assertEquals("foo=bar", message.getPayloadAsString());
}
else
{
fail("invalid HTTP method : " + httpMethod);
}
assertEquals("true", message.getInboundProperty(HttpConstants.HEADER_CONNECTION));
assertEquals("true", message.getInboundProperty(HttpConstants.HEADER_KEEP_ALIVE));
messageDidArrive.countDown();
}
}
@Override
protected int getNumPortsToFind()
{
return 1;
}
}