1
2
3
4
5
6
7
8
9
10 package org.mule.transformers.jaxb;
11
12 import org.mule.DefaultMuleMessage;
13 import org.mule.api.MuleMessage;
14 import org.mule.jaxb.model.EmailAddress;
15 import org.mule.jaxb.model.Person;
16 import org.mule.tck.AbstractMuleTestCase;
17 import org.mule.transformer.types.DataTypeFactory;
18 import org.mule.transformer.types.ListDataType;
19
20 import java.io.ByteArrayInputStream;
21 import java.util.List;
22
23 public class JaxbTransformerTestCase extends AbstractMuleTestCase
24 {
25 public static final String PERSON_XML = "<person><name>John Doe</name><dob>01/01/1970</dob><emailAddresses><emailAddress><type>home</type><address>john.doe@gmail.com</address></emailAddress><emailAddress><type>work</type><address>jdoe@bigco.com</address></emailAddress></emailAddresses></person>";
26
27 @Override
28 public void doSetUp() throws Exception
29 {
30 muleContext.getRegistry().registerObject("trans", new JAXBTestTransformers());
31 }
32
33 public void testCustomTransform() throws Exception
34 {
35 MuleMessage message = new DefaultMuleMessage(PERSON_XML, muleContext);
36 Person person = (Person) message.getPayload(DataTypeFactory.create(Person.class));
37 assertNotNull(person);
38 assertEquals("John Doe", person.getName());
39 assertEquals("01/01/1970", person.getDob());
40 assertEquals(2, person.getEmailAddresses().size());
41 assertEquals("home", person.getEmailAddresses().get(0).getType());
42 assertEquals("john.doe@gmail.com", person.getEmailAddresses().get(0).getAddress());
43 assertEquals("work", person.getEmailAddresses().get(1).getType());
44 assertEquals("jdoe@bigco.com", person.getEmailAddresses().get(1).getAddress());
45 }
46
47 public void testCustomTransformWithMuleMessage() throws Exception
48 {
49 ByteArrayInputStream in = new ByteArrayInputStream(PERSON_XML.getBytes());
50 DefaultMuleMessage msg = new DefaultMuleMessage(in, muleContext);
51 msg.setInboundProperty("foo", "fooValue");
52
53 List<EmailAddress> emailAddresses = msg.getPayload(new ListDataType<List<EmailAddress>>(EmailAddress.class));
54 assertNotNull(emailAddresses);
55 assertEquals(2, emailAddresses.size());
56 assertEquals("home", emailAddresses.get(0).getType());
57 assertEquals("john.doe@gmail.com", emailAddresses.get(0).getAddress());
58 assertEquals("work", emailAddresses.get(1).getType());
59 assertEquals("jdoe@bigco.com", emailAddresses.get(1).getAddress());
60 }
61
62
63 }