/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.tuscany.databinding.jaxb.axiom;
import java.io.StringWriter;
import javax.xml.bind.JAXBElement;
import javax.xml.stream.XMLStreamReader;
import org.apache.axiom.om.OMElement;
import org.apache.axiom.om.impl.builder.StAXOMBuilder;
import org.apache.tuscany.sca.core.DefaultExtensionPointRegistry;
import org.apache.tuscany.sca.core.ExtensionPointRegistry;
import org.apache.tuscany.sca.databinding.TransformationContext;
import org.apache.tuscany.sca.databinding.impl.TransformationContextImpl;
import org.apache.tuscany.sca.databinding.jaxb.JAXB2Node;
import org.apache.tuscany.sca.databinding.jaxb.JAXBContextHelper;
import org.apache.tuscany.sca.databinding.jaxb.axiom.JAXB2OMElement;
import org.apache.tuscany.sca.databinding.xml.Node2XMLStreamReader;
import org.apache.tuscany.sca.interfacedef.DataType;
import org.apache.tuscany.sca.interfacedef.impl.DataTypeImpl;
import org.apache.tuscany.sca.interfacedef.util.XMLType;
import org.junit.Test;
import org.w3c.dom.Node;
import com.example.ipo.jaxb.ObjectFactory;
import com.example.ipo.jaxb.PurchaseOrderType;
import com.example.ipo.jaxb.USAddress;
import com.example.ipo.jaxb.USState;
/**
* @version $Rev: 831163 $ $Date: 2009-10-30 00:04:47 +0000 (Fri, 30 Oct 2009) $
*/
public class JAXB2OMTestCase {
@Test
public void testTransformElement() throws Exception {
JAXBElement<PurchaseOrderType> po = createPO();
DataType<?> sourceDataType = new DataTypeImpl<XMLType>(PurchaseOrderType.class, XMLType.UNKNOWN);
DataType<?> targetDataType = new DataTypeImpl<XMLType>(PurchaseOrderType.class, new XMLType(po.getName(), null));
TransformationContext tContext = new TransformationContextImpl();
tContext.setSourceDataType(sourceDataType);
tContext.setTargetDataType(targetDataType);
ExtensionPointRegistry registry = new DefaultExtensionPointRegistry();
JAXBContextHelper contextHelper = JAXBContextHelper.getInstance(registry);
// Force the JAXBContext to be cached
contextHelper.createJAXBContext(tContext, true);
long start = System.currentTimeMillis();
JAXB2OMElement t1 = new JAXB2OMElement(registry);
OMElement om = t1.transform(po, tContext);
long duration1 = System.currentTimeMillis() - start;
StringWriter sw = new StringWriter();
// serializeAndConsume() will trigger the JAXBDataSource.serialize(Writer, OMOutputFormat)
om.serializeAndConsume(sw);
System.out.println(sw.toString());
start = System.currentTimeMillis();
Node node = new JAXB2Node(new DefaultExtensionPointRegistry()).transform(po, tContext);
XMLStreamReader reader = new Node2XMLStreamReader().transform(node, null);
om = new StAXOMBuilder(reader).getDocumentElement();
sw = new StringWriter();
om.serializeAndConsume(sw);
long duration2 = System.currentTimeMillis() - start;
System.out.println(sw.toString());
System.out.println(duration1 + " vs. " + duration2);
}
@Test
public void testTransformType() throws Exception {
JAXBElement<PurchaseOrderType> po = createPO();
DataType<?> sourceDataType = new DataTypeImpl<XMLType>(PurchaseOrderType.class, XMLType.UNKNOWN);
DataType<?> targetDataType = new DataTypeImpl<XMLType>(PurchaseOrderType.class, new XMLType(po.getName(), null));
TransformationContext tContext = new TransformationContextImpl();
tContext.setSourceDataType(sourceDataType);
tContext.setTargetDataType(targetDataType);
OMElement om = new JAXB2OMElement(new DefaultExtensionPointRegistry()).transform(po.getValue(), tContext);
StringWriter sw = new StringWriter();
om.serializeAndConsume(sw);
System.out.println(sw.toString());
}
private JAXBElement<PurchaseOrderType> createPO() {
ObjectFactory factory = new ObjectFactory();
PurchaseOrderType type = factory.createPurchaseOrderType();
JAXBElement<PurchaseOrderType> po = factory.createPurchaseOrder(type);
type.setItems(factory.createItems());
type.setComment("123");
USAddress address = factory.createUSAddress();
address.setCity("San Jose");
address.setStreet("ABC St.");
address.setState(USState.CA);
type.setShipTo(address);
return po;
}
}