View Javadoc

1   /*
2    * $Id: AbstractSplitterTestCase.java 20320 2010-11-24 15:03:31Z dfeist $
3    * --------------------------------------------------------------------------------------
4    * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
5    *
6    * The software in this package is published under the terms of the CPAL v1.0
7    * license, a copy of which has been included with this distribution in the
8    * LICENSE.txt file.
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  }