/*
* $Id: TransformDiscoveryTestCase.java 20321 2010-11-24 15:21:24Z 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.transformer;
import org.mule.api.transformer.Transformer;
import org.mule.api.transformer.TransformerException;
import org.mule.tck.AbstractMuleTestCase;
import org.mule.tck.testmodels.fruit.Apple;
import org.mule.tck.testmodels.fruit.Banana;
import org.mule.tck.testmodels.fruit.Orange;
import org.mule.tck.testmodels.fruit.RedApple;
import org.mule.transformer.types.DataTypeFactory;
public class TransformDiscoveryTestCase extends AbstractMuleTestCase
{
@Override
protected void doSetUp() throws Exception
{
muleContext.getRegistry().registerTransformer(new StringToApple());
muleContext.getRegistry().registerTransformer(new StringToOrange());
}
public void testSimpleDiscovery() throws Exception
{
Transformer t = muleContext.getRegistry().lookupTransformer(DataTypeFactory.STRING, DataTypeFactory.create(Apple.class));
assertNotNull(t);
assertEquals(StringToApple.class, t.getClass());
t = muleContext.getRegistry().lookupTransformer(DataTypeFactory.STRING, DataTypeFactory.create(Orange.class));
assertNotNull(t);
assertEquals(StringToOrange.class, t.getClass());
try
{
muleContext.getRegistry().lookupTransformer(DataTypeFactory.STRING, DataTypeFactory.create(Banana.class));
fail("There is no transformer to go from String to Banana");
}
catch (TransformerException e)
{
//expected
}
muleContext.getRegistry().registerTransformer(new StringToRedApple());
t = muleContext.getRegistry().lookupTransformer(DataTypeFactory.STRING, DataTypeFactory.create(RedApple.class));
assertNotNull(t);
assertEquals(StringToRedApple.class, t.getClass());
}
protected class StringToApple extends org.mule.transformer.AbstractDiscoverableTransformer
{
public StringToApple()
{
setReturnDataType(DataTypeFactory.create(Apple.class));
}
protected Object doTransform(Object src, String encoding) throws TransformerException
{
return new Apple();
}
}
protected class StringToRedApple extends org.mule.transformer.AbstractDiscoverableTransformer
{
public StringToRedApple()
{
setReturnDataType(DataTypeFactory.create(RedApple.class));
setPriorityWeighting(MAX_PRIORITY_WEIGHTING);
}
protected Object doTransform(Object src, String encoding) throws TransformerException
{
return new RedApple();
}
}
protected class StringToOrange extends org.mule.transformer.AbstractDiscoverableTransformer
{
public StringToOrange()
{
setReturnDataType(DataTypeFactory.create(Orange.class));
}
protected Object doTransform(Object src, String encoding) throws TransformerException
{
return new Orange();
}
}
}