RELEASE_2_1_13/src/test/org/apache/cocoon/components/sax/XMLByteStreamCompilerInterpreterTestCase.java

/*
 * 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.cocoon.components.sax;

import java.io.ByteArrayInputStream;

import javax.xml.parsers.SAXParser;
import javax.xml.parsers.SAXParserFactory;

import org.apache.avalon.excalibur.pool.Recyclable;
import org.apache.cocoon.xml.AbstractXMLTestCase;
import org.apache.cocoon.xml.DefaultHandlerWrapper;
import org.apache.cocoon.xml.dom.DOMBuilder;
import org.xml.sax.ContentHandler;
import org.xml.sax.helpers.DefaultHandler;

/**
 * Testcase for XMLByteStreamCompiler and Interpreter
 *
 * @author <a href="mailto:tcurdt@apache.org">Torsten Curdt</a>
 * @version
 */

public final class XMLByteStreamCompilerInterpreterTestCase extends AbstractXMLTestCase {
    public XMLByteStreamCompilerInterpreterTestCase(String s) {
        super(s);
    }

    public void testCompareDOM() throws Exception {
        // reference
        DOMBuilder in = new DOMBuilder();
        generateLargeSAX(in);

        // capture events
        XMLSerializer xmlc = makeSerializer();
        generateLargeSAX(xmlc);

        // recall events and build a DOM from it
        XMLDeserializer xmli = makeDeserializer();
        DOMBuilder out = new DOMBuilder();
        xmli.setConsumer(out);
        final Object fragment = xmlc.getSAXFragment();
        System.out.println("Fragment is bytes: " + ((byte[])fragment).length);
		xmli.deserialize(fragment);

        // compare DOMs
        assertXMLEqual(in.getDocument(), out.getDocument());
    }

    public void testCompareByteArray() throws Exception {
        // capture events
        XMLSerializer sa = makeSerializer();
        generateLargeSAX(sa);

        // serialize events
        byte[] aa = (byte[]) sa.getSAXFragment();

        // deserialize and capture
        XMLSerializer sb = makeSerializer();
        XMLDeserializer xmli = makeDeserializer();
        xmli.setConsumer(sb);
        xmli.deserialize(aa);

        // serialize again
        byte[] ab = (byte[]) sb.getSAXFragment();

        assertTrue(aa.length == ab.length);

        for (int i=0;i<aa.length;i++) {
            assertEquals(aa[i],ab[i]);
        }
    }

    public void testStressLoop() throws Exception {
        XMLSerializer xmlc = makeSerializer();

        long loop = 10000;

        // simply consume documents
        long start = System.currentTimeMillis();
        for(int i=0;i<loop;i++) {
            generateSmallSAX(xmlc);
            xmlc.getSAXFragment();
            if (xmlc instanceof Recyclable) {
            	((Recyclable)xmlc).recycle();
            } else {
            	xmlc = makeSerializer();
            }
        }
        long stop = System.currentTimeMillis();

        double r = 1000*loop/(stop-start);
        System.out.println("consuming: "+ r + " documents per second");
    }

    public void testCompareToParsing() throws Exception {
        DOMBuilder in = new DOMBuilder();
        generateSmallSAX(in);

        SAXParserFactory pfactory = SAXParserFactory.newInstance();
        SAXParser p = pfactory.newSAXParser();

        XMLSerializer xmlc = makeSerializer();
        DefaultHandlerWrapper wrapper = new DefaultHandlerWrapper(xmlc);

        ByteArrayInputStream bis = new ByteArrayInputStream(generateByteArray());

        long loop = 10000;

        // parse documents
        long start = System.currentTimeMillis();
        for(int i=0;i<loop;i++) {
            if (xmlc instanceof Recyclable) {
            	((Recyclable)xmlc).recycle();
            } else {
            	xmlc = makeSerializer();
            }
            bis.reset();
            p.parse(bis,wrapper);
        }
        long stop = System.currentTimeMillis();

        double r = 1000*loop/(stop-start);
        System.out.println("parsed: " + r + " documents per second");


        XMLDeserializer xmli = makeDeserializer();
        ContentHandler ch = new DefaultHandler();

        // recall documents
        start = System.currentTimeMillis();
        for(int i=0;i<loop;i++) {
            ((AbstractXMLByteStreamInterpreter)xmli).setContentHandler(ch);
            xmli.deserialize(xmlc.getSAXFragment());
        }
        stop = System.currentTimeMillis();

        r = 1000*loop/(stop-start);
        System.out.println("recalling: " + r + " documents per second");
    }

	protected XMLDeserializer makeDeserializer() {
		return new XMLByteStreamInterpreter();
	}

	protected XMLSerializer makeSerializer() {
		return new XMLByteStreamCompiler();
	}
}