package parse;

import java.io.IOException;
import java.io.StringReader;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.List;
import static org.testng.Assert.assertEquals;
import org.testng.annotations.Test;
import org.xml.sax.Attributes;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;
import org.xml.sax.XMLReader;
import org.xml.sax.ext.DefaultHandler2;
import org.xml.sax.helpers.XMLReaderFactory;

/**
 * JDK-6770436: Entity callback order differs between Java1.5 and Java1.6
 * https://bugs.openjdk.java.net/browse/JDK-6770436
 *
 */

public class EntityCharacterEventOrder {

    protected final static String xmlEncoding = "ISO-8859-15";
    protected static Charset xmlEncodingCharset = null;

    String _xml;
    static {
        xmlEncodingCharset = Charset.forName(xmlEncoding);
    }
    /**
    public static void main(String[] args) {
        TestRunner.run(JDK6770436Test.class);
    }
*/
    @Test
    public void entityCallbackOrderJava() throws SAXException, IOException {
        final String input = "<element> &amp; some more text</element>";

        final MockContentHandler handler = new MockContentHandler();
        final XMLReader xmlReader = XMLReaderFactory.createXMLReader();

        xmlReader.setContentHandler(handler);
        xmlReader.setProperty("http://xml.org/sax/properties/lexical-handler", handler);

        xmlReader.parse(new InputSource(new StringReader(input)));

        final List<String> events = handler.getEvents();
        printEvents(events);
        assertCallbackOrder(events); //regression from JDK5
    }

    private void assertCallbackOrder(final List<String> events) {
        assertEquals("startDocument", events.get(0));
        assertEquals("startElement 'element'", events.get(1));
        assertEquals("characters ' '", events.get(2));
        assertEquals("startEntity 'amp'", events.get(3));
        assertEquals("characters '&'", events.get(4));
        assertEquals("endEntity 'amp'", events.get(5));
        assertEquals("characters ' some more text'", events.get(6));
        assertEquals("endElement 'element'", events.get(7));
        assertEquals("endDocument", events.get(8));
    }

    private void printEvents(final List<String> events) {
        events.stream().forEach((e) -> {
            System.out.println(e);
        });
    }

    private class MockContentHandler extends DefaultHandler2 {

        private List<String> events;

        public List<String> getEvents() {
            return events;
        }

        @Override
        public void startDocument() throws SAXException {
            events = new ArrayList<String>();
            events.add("startDocument");
        }

        @Override
        public void characters(char[] ch, int start, int length) throws SAXException {
            events.add("characters '" + new String(ch, start, length) + "'");
        }

        @Override
        public void startElement(String uri, String localName, String name, Attributes atts) throws SAXException {
            events.add("startElement '" + name + "'");
        }

        @Override
        public void endElement(String uri, String localName, String name) throws SAXException {
            events.add("endElement '" + name + "'");
        }

        @Override
        public void endDocument() throws SAXException {
            events.add("endDocument");
        }

        @Override
        public void startEntity(String name) throws SAXException {
            events.add("startEntity '" + name + "'");
        }

        @Override
        public void endEntity(String name) throws SAXException {
            events.add("endEntity '" + name + "'");
        }
    }
}