View Javadoc
1   /*
2    * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
3    * The software in this package is published under the terms of the CPAL v1.0
4    * license, a copy of which has been included with this distribution in the
5    * LICENSE.txt file.
6    */
7   package org.mule.routing.outbound;
8   
9   import org.mule.DefaultMuleMessage;
10  import org.mule.api.MuleEvent;
11  import org.mule.api.MuleMessage;
12  import org.mule.api.MuleMessageCollection;
13  import org.mule.api.MuleSession;
14  import org.mule.api.endpoint.OutboundEndpoint;
15  import org.mule.api.service.Service;
16  import org.mule.routing.filters.PayloadTypeFilter;
17  import org.mule.tck.MuleTestUtils;
18  import org.mule.tck.junit4.AbstractMuleContextTestCase;
19  import org.mule.tck.testmodels.fruit.Apple;
20  import org.mule.tck.testmodels.fruit.Orange;
21  
22  import com.mockobjects.dynamic.Mock;
23  
24  import java.util.ArrayList;
25  import java.util.List;
26  
27  import org.junit.Test;
28  
29  import static org.junit.Assert.assertEquals;
30  import static org.junit.Assert.assertNotNull;
31  import static org.junit.Assert.assertTrue;
32  
33  public class ListMessageSplitterTestCase extends AbstractMuleContextTestCase
34  {
35      public ListMessageSplitterTestCase()
36      {
37          setStartContext(true);
38      }
39  
40      @Test
41      public void testCorrelationGroupSizePropertySet() throws Exception
42      {
43          Service testService = getTestService("test", Apple.class);
44          MuleSession session = getTestSession(testService, muleContext);
45  
46          OutboundEndpoint endpoint = getTestOutboundEndpoint("Test1Endpoint", 
47              "test://endpoint?exchangePattern=request-response");
48          ListMessageSplitter router = new ListMessageSplitter();
49          router.setFilter(null);
50          router.addRoute(endpoint);
51          router.setMuleContext(muleContext);
52  
53          List<String> payload = new ArrayList<String>();
54          payload.add("one");
55          payload.add("two");
56          payload.add("three");
57          payload.add("four");
58  
59          MuleMessage message = new DefaultMuleMessage(payload, muleContext);
60  
61          MuleEvent result = router.route(new OutboundRoutingTestEvent(message, session, endpoint));
62          assertNotNull(result);
63          MuleMessage resultMessage = result.getMessage();
64          assertNotNull(resultMessage);
65          assertTrue(resultMessage instanceof MuleMessageCollection);
66          assertEquals("There should be 4 results for 4 split messages.", 4, ((MuleMessageCollection) resultMessage).size());
67      }
68  
69      @Test
70      public void testMessageSplitterRouter() throws Exception
71      {
72          Mock session = MuleTestUtils.getMockSession();
73          session.matchAndReturn("getFlowConstruct", getTestService());
74          session.matchAndReturn("setFlowConstruct", RouterTestUtils.getArgListCheckerFlowConstruct(), null);
75  
76          OutboundEndpoint endpoint1 = getTestOutboundEndpoint("Test1endpoint", "test://endpointUri.1", null, new PayloadTypeFilter(Apple.class), null);
77          OutboundEndpoint endpoint2 = getTestOutboundEndpoint("Test2Endpoint", "test://endpointUri.2", null, new PayloadTypeFilter(Orange.class), null);
78          OutboundEndpoint endpoint3 = getTestOutboundEndpoint("Test3Endpoint", "test://endpointUri.3");
79          Mock mockendpoint1 = RouterTestUtils.getMockEndpoint(endpoint1);
80          Mock mockendpoint2 = RouterTestUtils.getMockEndpoint(endpoint2);
81          Mock mockendpoint3 = RouterTestUtils.getMockEndpoint(endpoint3);
82  
83          OutboundEndpoint endpoint4 = getTestOutboundEndpoint("Test4endpoint", 
84              "test://endpointUri.4?exchangePattern=request-response", null, 
85              new PayloadTypeFilter(Apple.class), null);
86          OutboundEndpoint endpoint5 = getTestOutboundEndpoint("Test5Endpoint", 
87              "test://endpointUri.5?exchangePattern=request-response", null, 
88              new PayloadTypeFilter(Orange.class), null);
89          OutboundEndpoint endpoint6 = getTestOutboundEndpoint("Test6Endpoint", 
90              "test://endpointUri.6?exchangePattern=request-response");
91          Mock mockendpoint4 = RouterTestUtils.getMockEndpoint(endpoint4);
92          Mock mockendpoint5 = RouterTestUtils.getMockEndpoint(endpoint5);
93          Mock mockendpoint6 = RouterTestUtils.getMockEndpoint(endpoint6);
94  
95          ListMessageSplitter asyncSplitter = new ListMessageSplitter();
96          asyncSplitter.setMuleContext(muleContext);
97          asyncSplitter.setDisableRoundRobin(true);
98          asyncSplitter.setFilter(new PayloadTypeFilter(List.class));
99          asyncSplitter.addRoute((OutboundEndpoint) mockendpoint1.proxy());
100         asyncSplitter.addRoute((OutboundEndpoint) mockendpoint2.proxy());
101         asyncSplitter.addRoute((OutboundEndpoint) mockendpoint3.proxy());
102 
103         ListMessageSplitter syncSplitter = new ListMessageSplitter();
104         syncSplitter.setMuleContext(muleContext);
105         syncSplitter.setDisableRoundRobin(true);
106         syncSplitter.setFilter(new PayloadTypeFilter(List.class));
107         syncSplitter.addRoute((OutboundEndpoint) mockendpoint4.proxy());
108         syncSplitter.addRoute((OutboundEndpoint) mockendpoint5.proxy());
109         syncSplitter.addRoute((OutboundEndpoint) mockendpoint6.proxy());
110         List<Object> payload = new ArrayList<Object>();
111         payload.add(new Apple());
112         payload.add(new Apple());
113         payload.add(new Orange());
114         payload.add("");
115         MuleMessage message = new DefaultMuleMessage(payload, muleContext);
116 
117         assertTrue(asyncSplitter.isMatch(message));
118         mockendpoint1.expect("process", RouterTestUtils.getArgListCheckerMuleEvent());
119         mockendpoint1.expect("process", RouterTestUtils.getArgListCheckerMuleEvent());
120         mockendpoint2.expect("process", RouterTestUtils.getArgListCheckerMuleEvent());
121         mockendpoint3.expect("process", RouterTestUtils.getArgListCheckerMuleEvent());
122         asyncSplitter.route(new OutboundRoutingTestEvent(message, (MuleSession) session.proxy()));
123         session.verify();
124 
125         message = new DefaultMuleMessage(payload, muleContext);
126         MuleEvent event = new OutboundRoutingTestEvent(message, null);
127 
128         mockendpoint4.expectAndReturn("process", RouterTestUtils.getArgListCheckerMuleEvent(), event);
129         mockendpoint4.expectAndReturn("process", RouterTestUtils.getArgListCheckerMuleEvent(), event);
130         mockendpoint5.expectAndReturn("process", RouterTestUtils.getArgListCheckerMuleEvent(), event);
131         mockendpoint6.expectAndReturn("process", RouterTestUtils.getArgListCheckerMuleEvent(), event);
132         MuleEvent result = syncSplitter.route(new OutboundRoutingTestEvent(message, (MuleSession) session.proxy()));
133         assertNotNull(result);
134         MuleMessage resultMessage = result.getMessage();
135         assertNotNull(resultMessage);
136         assertTrue(resultMessage instanceof MuleMessageCollection);
137         assertEquals(4, ((MuleMessageCollection) resultMessage).size());
138         session.verify();
139     }
140 }