1
2
3
4
5
6
7
8
9
10 package org.mule.transformers.jaxb;
11
12 import org.mule.api.transformer.Transformer;
13 import org.mule.jaxb.model.EmailAddress;
14 import org.mule.jaxb.model.Person;
15 import org.mule.module.xml.transformer.jaxb.JAXBMarshallerTransformer;
16 import org.mule.module.xml.transformer.jaxb.JAXBUnmarshallerTransformer;
17 import org.mule.transformer.AbstractTransformerTestCase;
18 import org.mule.transformer.types.DataTypeFactory;
19
20 import java.util.Arrays;
21
22 import javax.xml.bind.JAXBContext;
23
24 import org.custommonkey.xmlunit.XMLUnit;
25
26 import static org.junit.Assert.fail;
27
28 public class JaxbRoundtripTestCase extends AbstractTransformerTestCase
29 {
30 protected JAXBContext ctx;
31
32 @Override
33 protected void doSetUp() throws Exception
34 {
35 ctx = JAXBContext.newInstance(Person.class);
36 super.doSetUp();
37 }
38
39 @Override
40 public Transformer getTransformer() throws Exception
41 {
42 JAXBUnmarshallerTransformer t = new JAXBUnmarshallerTransformer(ctx, DataTypeFactory.create(Person.class));
43 initialiseObject(t);
44 return t;
45 }
46
47 @Override
48 public Transformer getRoundTripTransformer() throws Exception
49 {
50 JAXBMarshallerTransformer t = new JAXBMarshallerTransformer(ctx, DataTypeFactory.STRING);
51 initialiseObject(t);
52 return t;
53 }
54
55 @Override
56 public Object getTestData()
57 {
58 return "<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>";
59 }
60
61 @Override
62 public Object getResultData()
63 {
64 Person p = new Person();
65 p.setName("John Doe");
66 p.setDob("01/01/1970");
67
68 EmailAddress ea = new EmailAddress("john.doe@gmail.com", "home");
69 EmailAddress ea2 = new EmailAddress("jdoe@bigco.com", "work");
70 p.setEmailAddresses(Arrays.asList(ea, ea2));
71
72 return p;
73 }
74
75 @Override
76 public boolean compareRoundtripResults(Object expected, Object result)
77 {
78 try
79 {
80 return XMLUnit.compareXML(expected.toString(), result.toString()).similar();
81 }
82 catch (Exception e)
83 {
84 fail("Failed to compare roudtrip XML strings: " + e.getMessage());
85 return false;
86 }
87 }
88 }