1
2
3
4
5
6
7
8
9
10
11 package org.mule.routing;
12
13 import org.mule.DefaultMessageCollection;
14 import org.mule.DefaultMuleEvent;
15 import org.mule.DefaultMuleMessage;
16 import org.mule.api.MuleEvent;
17 import org.mule.api.MuleException;
18 import org.mule.api.MuleMessage;
19 import org.mule.api.MuleMessageCollection;
20 import org.mule.api.processor.MessageProcessor;
21 import org.mule.tck.AbstractMuleTestCase;
22 import org.mule.tck.testmodels.fruit.Apple;
23 import org.mule.tck.testmodels.fruit.Banana;
24 import org.mule.tck.testmodels.fruit.Fruit;
25 import org.mule.tck.testmodels.fruit.FruitBowl;
26 import org.mule.tck.testmodels.fruit.Orange;
27
28 import java.util.ArrayList;
29 import java.util.List;
30
31 public class AbstractSplitterTestCase extends AbstractMuleTestCase
32 {
33
34 public void testSimpleSplitter() throws Exception
35 {
36 TestSplitter splitter = new TestSplitter();
37 MultipleEventSensingMessageProcessor listener = new MultipleEventSensingMessageProcessor();
38 splitter.setListener(listener);
39 splitter.setMuleContext(muleContext);
40
41 Apple apple = new Apple();
42 Banana banana = new Banana();
43 Orange orange = new Orange();
44 FruitBowl fruitBowl = new FruitBowl();
45 fruitBowl.addFruit(apple);
46 fruitBowl.addFruit(banana);
47 fruitBowl.addFruit(orange);
48
49 MuleEvent inEvent = new DefaultMuleEvent(new DefaultMuleMessage(fruitBowl, muleContext),
50 getTestEvent(""));
51
52 MuleEvent resultEvent = splitter.process(inEvent);
53
54 assertEquals(3, listener.events.size());
55 assertTrue(listener.events.get(0).getMessage().getPayload() instanceof Fruit);
56 assertTrue(listener.events.get(1).getMessage().getPayload() instanceof Fruit);
57 assertTrue(listener.events.get(2).getMessage().getPayload() instanceof Fruit);
58
59 assertEquals(DefaultMessageCollection.class, resultEvent.getMessage().getClass());
60 assertEquals(3, ((MuleMessageCollection) resultEvent.getMessage()).size());
61 assertTrue(((MuleMessageCollection) resultEvent.getMessage()).getMessage(0).getPayload() instanceof Fruit);
62 assertTrue(((MuleMessageCollection) resultEvent.getMessage()).getMessage(1).getPayload() instanceof Fruit);
63 assertTrue(((MuleMessageCollection) resultEvent.getMessage()).getMessage(2).getPayload() instanceof Fruit);
64
65 }
66
67 private static class MultipleEventSensingMessageProcessor implements MessageProcessor
68 {
69 List<MuleEvent> events = new ArrayList<MuleEvent>();
70
71 public MuleEvent process(MuleEvent event) throws MuleException
72 {
73 events.add(event);
74 return event;
75 }
76 }
77
78 private static class TestSplitter extends AbstractSplitter
79 {
80 @Override
81 protected List<MuleMessage> splitMessage(MuleEvent event)
82 {
83 FruitBowl bowl = (FruitBowl) event.getMessage().getPayload();
84 List<MuleMessage> parts = new ArrayList<MuleMessage>();
85 for (Fruit fruit : bowl.getFruit())
86 {
87 parts.add(new DefaultMuleMessage(fruit, muleContext));
88 }
89 return parts;
90 }
91 }
92
93 }