/*
* $Id: MapSplitterTestCase.java 21312 2011-02-18 14:45:03Z dfeist $
* -------------------------------------------------------------------------------------
* 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;
import org.mule.api.MuleEvent;
import org.mule.api.MuleException;
import org.mule.api.processor.MessageProcessor;
import org.mule.api.transport.PropertyScope;
import org.mule.tck.AbstractMuleTestCase;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
public class MapSplitterTestCase extends AbstractMuleTestCase
{
private MapSplitter mapSplitter;
private List<String> splitPayloads = new ArrayList<String>();
private List<String> splitKeyProperties = new ArrayList<String>();
@Override
protected void doSetUp() throws Exception
{
super.doSetUp();
mapSplitter = new MapSplitter();
mapSplitter.setMuleContext(muleContext);
mapSplitter.setListener(new MessageProcessor()
{
public MuleEvent process(MuleEvent event) throws MuleException
{
splitPayloads.add(event.getMessageAsString());
splitKeyProperties.add((String) event.getMessage().getProperty(MapSplitter.MAP_ENTRY_KEY,
PropertyScope.INVOCATION));
return event;
}
});
}
public void testSplit() throws Exception
{
Map<String, Object> testMap = new HashMap<String, Object>();
testMap.put("1", "one");
testMap.put("2", "two");
testMap.put("3", "three");
mapSplitter.process(getTestEvent(testMap));
assertEquals(3, splitPayloads.size());
assertTrue(splitPayloads.contains("one"));
assertTrue(splitPayloads.contains("two"));
assertTrue(splitPayloads.contains("three"));
assertEquals(3, splitPayloads.size());
assertEquals("1", splitKeyProperties.get(splitPayloads.indexOf("one")));
assertEquals("2", splitKeyProperties.get(splitPayloads.indexOf("two")));
assertEquals("3", splitKeyProperties.get(splitPayloads.indexOf("three")));
}
}