/*
* $Id: AbstractMessageProcessorTestCase.java 22156 2011-06-08 21:36:30Z dfeist $
* --------------------------------------------------------------------------------------
* 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.endpoint;
import org.mule.DefaultMuleEvent;
import org.mule.DefaultMuleMessage;
import org.mule.MessageExchangePattern;
import org.mule.api.MuleContext;
import org.mule.api.MuleEvent;
import org.mule.api.MuleException;
import org.mule.api.MuleMessage;
import org.mule.api.context.MuleContextAware;
import org.mule.api.context.MuleContextBuilder;
import org.mule.api.context.notification.EndpointMessageNotificationListener;
import org.mule.api.context.notification.SecurityNotificationListener;
import org.mule.api.endpoint.EndpointBuilder;
import org.mule.api.endpoint.EndpointException;
import org.mule.api.endpoint.ImmutableEndpoint;
import org.mule.api.endpoint.InboundEndpoint;
import org.mule.api.endpoint.OutboundEndpoint;
import org.mule.api.exception.MessagingExceptionHandler;
import org.mule.api.lifecycle.InitialisationException;
import org.mule.api.processor.MessageProcessor;
import org.mule.api.routing.filter.Filter;
import org.mule.api.security.EndpointSecurityFilter;
import org.mule.api.security.SecurityFilter;
import org.mule.api.service.Service;
import org.mule.api.transaction.RollbackMethod;
import org.mule.api.transaction.TransactionConfig;
import org.mule.api.transformer.Transformer;
import org.mule.context.notification.EndpointMessageNotification;
import org.mule.context.notification.SecurityNotification;
import org.mule.context.notification.ServerNotificationManager;
import org.mule.message.DefaultExceptionPayload;
import org.mule.processor.SecurityFilterMessageProcessor;
import org.mule.routing.MessageFilter;
import org.mule.tck.AbstractMuleTestCase;
import org.mule.transport.NullPayload;
import org.mule.util.concurrent.Latch;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
public abstract class AbstractMessageProcessorTestCase extends AbstractMuleTestCase
{
protected static final String TEST_URI = "test://myTestUri";
protected static String RESPONSE_MESSAGE = "response-message";
protected static MuleMessage responseMessage;
@Override
protected void doSetUp() throws Exception
{
super.doSetUp();
responseMessage = createTestResponseMuleMessage();
muleContext.start();
}
@Override
protected void configureMuleContext(MuleContextBuilder builder)
{
super.configureMuleContext(builder);
// Configure EndpointMessageNotificationListener for notifications test
ServerNotificationManager notificationManager = new ServerNotificationManager();
notificationManager.addInterfaceToType(EndpointMessageNotificationListener.class,
EndpointMessageNotification.class);
notificationManager.addInterfaceToType(SecurityNotificationListener.class, SecurityNotification.class);
builder.setNotificationManager(notificationManager);
}
protected InboundEndpoint createTestInboundEndpoint(Transformer transformer,
Transformer responseTransformer)
throws EndpointException, InitialisationException
{
return createTestInboundEndpoint(null, null, transformer, responseTransformer,
MessageExchangePattern.REQUEST_RESPONSE, null);
}
protected InboundEndpoint createTestInboundEndpoint(Filter filter,
SecurityFilter securityFilter,
MessageExchangePattern exchangePattern,
TransactionConfig txConfig)
throws InitialisationException, EndpointException
{
return createTestInboundEndpoint(filter, securityFilter, null, null, exchangePattern, txConfig);
}
protected InboundEndpoint createTestInboundEndpoint(Filter filter,
SecurityFilter securityFilter,
Transformer transformer,
Transformer responseTransformer,
MessageExchangePattern exchangePattern,
TransactionConfig txConfig)
throws EndpointException, InitialisationException
{
EndpointURIEndpointBuilder endpointBuilder = new EndpointURIEndpointBuilder(TEST_URI, muleContext);
if (filter != null)
{
endpointBuilder.addMessageProcessor(new MessageFilter(filter));
}
if (securityFilter != null)
{
endpointBuilder.addMessageProcessor(new SecurityFilterMessageProcessor(securityFilter));
}
if (transformer != null)
{
endpointBuilder.setMessageProcessors(Collections.<MessageProcessor>singletonList(transformer));
}
if (responseTransformer != null)
{
endpointBuilder.setResponseMessageProcessors(Collections.<MessageProcessor>singletonList(responseTransformer));
}
endpointBuilder.setExchangePattern(exchangePattern);
endpointBuilder.setTransactionConfig(txConfig);
InboundEndpoint endpoint = endpointBuilder.buildInboundEndpoint();
return endpoint;
}
protected MuleEvent createTestInboundEvent(InboundEndpoint endpoint) throws Exception
{
Map<String, Object> props = new HashMap<String, Object>();
props.put("prop1", "value1");
return new DefaultMuleEvent(new DefaultMuleMessage(TEST_MESSAGE, props, muleContext), endpoint,
getTestSession(getTestService(), muleContext));
}
protected OutboundEndpoint createTestOutboundEndpoint(Transformer transformer,
Transformer responseTransformer)
throws EndpointException, InitialisationException
{
return createTestOutboundEndpoint(null, null, transformer, responseTransformer,
MessageExchangePattern.REQUEST_RESPONSE, null);
}
protected OutboundEndpoint createTestOutboundEndpoint(Filter filter,
EndpointSecurityFilter securityFilter,
MessageExchangePattern exchangePattern,
TransactionConfig txConfig)
throws InitialisationException, EndpointException
{
return createTestOutboundEndpoint(filter, securityFilter, null, null, exchangePattern,
txConfig);
}
protected OutboundEndpoint createTestOutboundEndpoint(Filter filter,
EndpointSecurityFilter securityFilter,
Transformer transformer,
Transformer responseTransformer,
MessageExchangePattern exchangePattern,
TransactionConfig txConfig)
throws EndpointException, InitialisationException
{
return createTestOutboundEndpoint("test://test", filter, securityFilter, transformer, responseTransformer, exchangePattern, txConfig);
}
protected OutboundEndpoint createTestOutboundEndpoint(String uri, Filter filter,
SecurityFilter securityFilter,
Transformer transformer,
Transformer responseTransformer,
MessageExchangePattern exchangePattern,
TransactionConfig txConfig)
throws EndpointException, InitialisationException
{
EndpointURIEndpointBuilder endpointBuilder = new EndpointURIEndpointBuilder(uri,
muleContext);
if (filter != null)
{
endpointBuilder.addMessageProcessor(new MessageFilter(filter));
}
if (securityFilter != null)
{
endpointBuilder.addMessageProcessor(new SecurityFilterMessageProcessor(securityFilter));
}
if (transformer != null)
{
endpointBuilder.setMessageProcessors(Collections.<MessageProcessor>singletonList(transformer));
}
if (responseTransformer != null)
{
endpointBuilder.setResponseMessageProcessors(Collections.<MessageProcessor>singletonList(responseTransformer));
}
endpointBuilder.setExchangePattern(exchangePattern);
endpointBuilder.setTransactionConfig(txConfig);
customizeEndpointBuilder(endpointBuilder);
return endpointBuilder.buildOutboundEndpoint();
}
protected void customizeEndpointBuilder(EndpointBuilder endpointBuilder)
{
// template method
}
protected MuleEvent createTestOutboundEvent() throws Exception
{
return createTestOutboundEvent(null);
}
protected MuleEvent createTestOutboundEvent(MessagingExceptionHandler exceptionListener) throws Exception
{
Map<String, Object> props = new HashMap<String, Object>();
props.put("prop1", "value1");
props.put("port", 12345);
Service svc = getTestService();
if (exceptionListener != null)
{
svc.setExceptionListener(exceptionListener);
}
return new DefaultMuleEvent(new DefaultMuleMessage(TEST_MESSAGE, props, muleContext), getTestInboundEndpoint(MessageExchangePattern.REQUEST_RESPONSE), getTestSession(svc, muleContext));
}
protected MuleMessage createTestResponseMuleMessage()
{
return new DefaultMuleMessage(RESPONSE_MESSAGE, muleContext);
}
public static class TestFilter implements Filter
{
public boolean accept;
public TestFilter(boolean accept)
{
this.accept = accept;
}
public boolean accept(MuleMessage message)
{
return accept;
}
}
public static class TestSecurityNotificationListener implements SecurityNotificationListener<SecurityNotification>
{
public SecurityNotification securityNotification;
public Latch latch = new Latch();
public void onNotification(SecurityNotification notification)
{
securityNotification = notification;
latch.countDown();
}
}
public static class TestListener implements MessageProcessor
{
public MuleEvent sensedEvent;
public MuleEvent process(MuleEvent event) throws MuleException
{
sensedEvent = event;
return event;
}
}
public static class TestEndpointMessageNotificationListener implements EndpointMessageNotificationListener<EndpointMessageNotification>
{
public EndpointMessageNotification messageNotification;
public Latch latch = new Latch();
public void onNotification(EndpointMessageNotification notification)
{
messageNotification = notification;
latch.countDown();
}
}
public static class ExceptionThrowingMessageProcessr implements MessageProcessor
{
public MuleEvent process(MuleEvent event) throws MuleException
{
throw new IllegalStateException();
}
}
public static class TestExceptionListener implements MessagingExceptionHandler
{
public Exception sensedException;
public MuleEvent handleException(Exception exception, MuleEvent event, RollbackMethod rollbackMethod)
{
sensedException = exception;
event.getMessage().setPayload(NullPayload.getInstance());
event.getMessage().setExceptionPayload(new DefaultExceptionPayload(exception));
return event;
}
public MuleEvent handleException(Exception exception, MuleEvent event)
{
return handleException(exception, event, null);
}
}
public class ObjectAwareProcessor implements MessageProcessor, EndpointAware, MuleContextAware
{
public MuleContext context;
public ImmutableEndpoint endpoint;
public MuleEvent process(MuleEvent event) throws MuleException
{
return null;
}
public void setEndpoint(ImmutableEndpoint endpoint)
{
this.endpoint = endpoint;
}
public void setMuleContext(MuleContext context)
{
this.context = context;
}
}
}