1
2
3
4
5
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 }