/*
* 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.routing.outbound;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import static org.mockito.Matchers.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import org.mule.DefaultMuleMessage;
import org.mule.api.MuleEvent;
import org.mule.api.MuleMessage;
import org.mule.api.MuleMessageCollection;
import org.mule.api.MuleSession;
import org.mule.api.config.MuleProperties;
import org.mule.api.endpoint.OutboundEndpoint;
import org.mule.api.processor.MessageProcessor;
import org.mule.api.transport.PropertyScope;
import org.mule.routing.CorrelationMode;
import org.mule.routing.filters.RegExFilter;
import org.mule.tck.MuleEventCheckAnswer;
import org.mule.tck.junit4.AbstractMuleContextTestCase;
import java.util.ArrayList;
import java.util.List;
import org.junit.Test;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;
public class MulticastingRouterTestCase extends AbstractMuleContextTestCase
{
public MulticastingRouterTestCase()
{
setStartContext(true);
}
@Test
public void testMulticastingRouterAsync() throws Exception
{
RegExFilter filter = new RegExFilter("(.*) Message");
OutboundEndpoint endpoint1 = getTestOutboundEndpoint("Test1Provider", "test://test1", null, filter, null);
assertNotNull(endpoint1);
OutboundEndpoint mockendpoint1 = RouterTestUtils.createMockEndpoint(endpoint1);
OutboundEndpoint endpoint2 = getTestOutboundEndpoint("Test2Provider", "test://test2", null, filter, null);
assertNotNull(endpoint2);
OutboundEndpoint mockendpoint2 = RouterTestUtils.createMockEndpoint(endpoint2);
MulticastingRouter router = createObject(MulticastingRouter.class);
List<MessageProcessor> endpoints = new ArrayList<MessageProcessor>();
endpoints.add(mockendpoint1);
endpoints.add(mockendpoint2);
router.setRoutes(endpoints);
MuleMessage message = new DefaultMuleMessage(TEST_MESSAGE, muleContext);
assertTrue(router.isMatch(message));
when(mockendpoint1.process(any(MuleEvent.class))).thenAnswer(new MuleEventCheckAnswer());
when(mockendpoint2.process(any(MuleEvent.class))).thenAnswer(new MuleEventCheckAnswer());
MuleSession session = mock(MuleSession.class);
router.route(new OutboundRoutingTestEvent(message, session, muleContext));
}
@Test
public void testMulticastingRouterSync() throws Exception
{
OutboundEndpoint endpoint1 = getTestOutboundEndpoint("Test1Provider",
"test://Test1Provider?exchangePattern=request-response");
assertNotNull(endpoint1);
OutboundEndpoint mockendpoint1 = RouterTestUtils.createMockEndpoint(endpoint1);
OutboundEndpoint endpoint2 = getTestOutboundEndpoint("Test2Provider",
"test://Test2Provider?exchangePattern=request-response");
assertNotNull(endpoint2);
OutboundEndpoint mockendpoint2 = RouterTestUtils.createMockEndpoint(endpoint2);
MulticastingRouter router = createObject(MulticastingRouter.class);
RegExFilter filter = new RegExFilter("(.*) Message");
router.setFilter(filter);
List<MessageProcessor> endpoints = new ArrayList<MessageProcessor>();
endpoints.add(mockendpoint1);
endpoints.add(mockendpoint2);
router.setRoutes(endpoints);
assertEquals(filter, router.getFilter());
MuleMessage message = new DefaultMuleMessage(TEST_MESSAGE, muleContext);
assertTrue(router.isMatch(message));
MuleEvent event = new OutboundRoutingTestEvent(message, null, muleContext);
when(mockendpoint1.process(any(MuleEvent.class))).thenAnswer(new MuleEventCheckAnswer(event));
when(mockendpoint2.process(any(MuleEvent.class))).thenAnswer(new MuleEventCheckAnswer(event));
MuleSession session = mock(MuleSession.class);
MuleEvent result = router.route(new OutboundRoutingTestEvent(message, session, muleContext));
assertNotNull(result);
MuleMessage resultMessage = result.getMessage();
assertNotNull(resultMessage);
assertTrue(resultMessage instanceof MuleMessageCollection);
assertEquals(2, ((MuleMessageCollection)resultMessage).size());
}
@Test
public void testMulticastingRouterMixedSyncAsync() throws Exception
{
OutboundEndpoint endpoint1 = getTestOutboundEndpoint("Test1Provider",
"test://Test1Provider?exchangePattern=request-response");
assertNotNull(endpoint1);
OutboundEndpoint mockendpoint1 = RouterTestUtils.createMockEndpoint(endpoint1);
OutboundEndpoint endpoint2 = getTestOutboundEndpoint("Test2Provider",
"test://Test2Provider?exchangePattern=request-response");
assertNotNull(endpoint2);
OutboundEndpoint mockendpoint2 = RouterTestUtils.createMockEndpoint(endpoint2);
MulticastingRouter router = createObject(MulticastingRouter.class);
List<MessageProcessor> endpoints = new ArrayList<MessageProcessor>();
endpoints.add(mockendpoint1);
endpoints.add(mockendpoint2);
router.setRoutes(endpoints);
MuleMessage message = new DefaultMuleMessage(TEST_MESSAGE, muleContext);
assertTrue(router.isMatch(message));
MuleEvent event = new OutboundRoutingTestEvent(message, null, muleContext);
when(mockendpoint1.process(any(MuleEvent.class))).thenAnswer(new MuleEventCheckAnswer(event));
when(mockendpoint2.process(any(MuleEvent.class))).thenAnswer(new MuleEventCheckAnswer());
MuleSession session = mock(MuleSession.class);
MuleEvent result = router.route(new OutboundRoutingTestEvent(message, session, muleContext));
assertNotNull(result);
assertEquals(getPayload(message), getPayload(result.getMessage()));
}
@Test
public void testMulticastingRouterCorrelationIdPropagation() throws Exception
{
RegExFilter filter = new RegExFilter("(.*) Message");
OutboundEndpoint endpoint1 = getTestOutboundEndpoint("Test1Provider", "test://test1", null, filter, null);
assertNotNull(endpoint1);
OutboundEndpoint mockendpoint1 = RouterTestUtils.createMockEndpoint(endpoint1);
OutboundEndpoint endpoint2 = getTestOutboundEndpoint("Test2Provider", "test://test2", null, filter, null);
assertNotNull(endpoint2);
OutboundEndpoint mockendpoint2 = RouterTestUtils.createMockEndpoint(endpoint2);
MulticastingRouter router = createObject(MulticastingRouter.class);
List<MessageProcessor> endpoints = new ArrayList<MessageProcessor>();
endpoints.add(mockendpoint1);
endpoints.add(mockendpoint2);
router.setRoutes(endpoints);
router.setEnableCorrelation(CorrelationMode.NEVER);
MuleMessage message = new DefaultMuleMessage(TEST_MESSAGE, muleContext);
message.setProperty(MuleProperties.MULE_CORRELATION_ID_PROPERTY, "MyCustomCorrelationId", PropertyScope.INBOUND);
assertTrue(router.isMatch(message));
Answer<MuleEvent> answer = new Answer<MuleEvent>()
{
@Override
public MuleEvent answer(InvocationOnMock invocation) throws Throwable
{
Object[] arguments = invocation.getArguments();
assertEquals(1, arguments.length);
assertTrue(arguments[0] instanceof MuleEvent);
MuleEvent event = (MuleEvent) arguments[0];
String correlationId = event.getMessage().getOutboundProperty(MuleProperties.MULE_CORRELATION_ID_PROPERTY);
assertNotNull(correlationId);
assertEquals("MyCustomCorrelationId", correlationId);
return event;
}
};
when(mockendpoint1.process(any(MuleEvent.class))).thenAnswer(answer);
when(mockendpoint2.process(any(MuleEvent.class))).thenAnswer(answer);
MuleSession session = mock(MuleSession.class);
router.route(new OutboundRoutingTestEvent(message, session, muleContext));
}
private String getPayload(MuleMessage message) throws Exception
{
Object payload = message.getPayload();
if (payload instanceof List)
{
payload = ((List<?>) payload).get(0);
}
return payload.toString();
}
}