v2/test/src/drt/drtcases/MarshalTests.java

package drtcases;

import junit.framework.Assert;
import junit.framework.Test;
import junit.framework.TestCase;
import junit.framework.TestSuite;
import org.apache.xmlbeans.XmlException;
import org.apache.xmlbeans.impl.binding.bts.BindingFile;
import org.apache.xmlbeans.impl.marshal.BindingContext;
import org.apache.xmlbeans.impl.marshal.BindingContextFactory;
import org.apache.xmlbeans.impl.marshal.Unmarshaller;
import org.apache.xmlbeans.x2003.x09.bindingConfig.BindingConfigDocument;

import javax.xml.stream.XMLInputFactory;
import javax.xml.stream.XMLStreamReader;
import java.io.CharArrayReader;
import java.io.CharArrayWriter;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.io.Reader;
import java.io.StringReader;
import java.io.Writer;

public class MarshalTests extends TestCase
{
    public MarshalTests(String name)
    {
        super(name);
    }

    public static Test suite()
    {
        return new TestSuite(MarshalTests.class);
    }

    public void testManySimpleTypesUnmarshall()
        throws Exception
    {
        testSimpleTypeUnmarshall(new Long(554345354445555555L), "long");
        testSimpleTypeUnmarshall(new Float(54.5423f), "float");
        testSimpleTypeUnmarshall("random string", "string");
    }


    //only works for values where .toString() is equivalent to marshalling
    public void testSimpleTypeUnmarshall(Object expected, String xsd_type)
        throws Exception
    {
        Unmarshaller unmarshaller = getSimpleUnmarshaller();

        String xmldoc = "<a xmlns:xsi='http://www.w3.org/2001/XMLSchema-instance'" +
            " xmlns:xs='http://www.w3.org/2001/XMLSchema' xsi:type='xs:" +
            xsd_type + "' >" + expected + "</a>";

        StringReader stringReader = new StringReader(xmldoc);
        XMLStreamReader xrdr =
            XMLInputFactory.newInstance().createXMLStreamReader(stringReader);

        Object obj = unmarshaller.unmarshal(xrdr);

        Assert.assertEquals(expected, obj);

        System.out.println("OK for " + expected);
    }

    public void testByNameBeanUnmarshall()
        throws Exception
    {
        BindingConfigDocument bcdoc = getBindingConfigDocument();

        BindingContext bindingContext =
            BindingContextFactory.createBindingContext(bcdoc);

        Unmarshaller unmarshaller = bindingContext.createUnmarshaller();

        Assert.assertNotNull(unmarshaller);

        //TODO: remove hard coded path
        File doc = new File("test/cases/marshal/doc.xml");


        final XMLInputFactory xmlInputFactory = XMLInputFactory.newInstance();
        XMLStreamReader xrdr =
            xmlInputFactory.createXMLStreamReader(new FileReader(doc));

        Object obj = unmarshaller.unmarshal(xrdr);

        System.out.println("obj = " + obj);
    }

    public void testPerfByNameBeanUnmarshall()
        throws Exception
    {
        BindingConfigDocument bcdoc = getBindingConfigDocument();

        BindingContext bindingContext =
            BindingContextFactory.createBindingContext(bcdoc);

        Unmarshaller unmarshaller = bindingContext.createUnmarshaller();

        Assert.assertNotNull(unmarshaller);

        //TODO: remove hard coded path
        File doc = new File("test/cases/marshal/doc.xml");
        final FileReader fileReader = new FileReader(doc);
        CharArrayWriter cw = new CharArrayWriter();

        bufferedStreamCopy(fileReader, cw);
        final char[] chars = cw.toCharArray();
        final CharArrayReader cr = new CharArrayReader(chars);

        final int trials = 5000;

        final XMLInputFactory xmlInputFactory = XMLInputFactory.newInstance();

        final long before_millis = System.currentTimeMillis();
        for (int i = 0; i < trials; i++) {
            cr.reset();
            XMLStreamReader xrdr =
                xmlInputFactory.createXMLStreamReader(cr);

            Object obj = unmarshaller.unmarshal(xrdr);

            if ((i % 1000) == 0)
                System.out.println("i=" + i + "\tobj = " + obj);
        }
        final long after_millis = System.currentTimeMillis();
        final long diff = (after_millis - before_millis);
        System.out.println("milliseconds: " + diff + " trials: " + trials);
        System.out.println("milliseconds PER trial: " + (diff / (double)trials));
    }

    protected static void bufferedStreamCopy(Reader in, Writer out)
        throws IOException
    {
        int charsRead;
        char[] buf = new char[1024];

        while ((charsRead = in.read(buf)) != -1) {
            out.write(buf, 0, charsRead);
        }
    }


    private BindingConfigDocument getBindingConfigDocument()
        throws IOException, XmlException
    {
        //TODO: remove hard coded path
        File loc = new File("test/cases/marshal/example_config.xml");
        BindingConfigDocument bcdoc = BindingConfigDocument.Factory.parse(loc);
        return bcdoc;
    }

    private static Unmarshaller getSimpleUnmarshaller() throws IOException
    {
        BindingFile bf = new BindingFile();
        BindingConfigDocument bindingConfigDocument = bf.write();

        BindingContext bindingContext =
            BindingContextFactory.createBindingContext(bindingConfigDocument);

        Unmarshaller unmarshaller = bindingContext.createUnmarshaller();
        return unmarshaller;
    }
    
    public static void main(String args[])
    {
        junit.textui.TestRunner.run(suite());
    }

}