/*
* Copyright 2004 Sun Microsystems, Inc.
*
* Licensed 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 com.rometools.rome.unittest;
import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStreamWriter;
import java.io.Reader;
import java.io.StringReader;
import java.io.Writer;
import junit.framework.TestCase;
import org.jdom2.input.SAXBuilder;
import com.rometools.rome.io.XmlReader;
import com.rometools.rome.io.impl.XmlFixerReader;
/**
* @author pat, tucu
*
*/
public class TestXmlFixerReader extends TestCase {
private static final String XML_PROLOG = "<?xml version=\"1.0\" ?>";
public void testTrim() throws Exception {
testValidTrim("", "<hello></hello>");
testValidTrim("", XML_PROLOG + "<hello></hello>");
testValidTrim(" ", "<hello></hello>");
testValidTrim(" ", XML_PROLOG + "<hello></hello>");
testValidTrim(" \n", "<hello></hello>");
testValidTrim(" \n", XML_PROLOG + "<hello></hello>");
testValidTrim("<!-- - -- -->", "<hello></hello>");
testValidTrim("<!-- - -- -->", XML_PROLOG + "<hello></hello>");
testValidTrim(" <!-- - -- -->", "<hello></hello>");
testValidTrim(" <!-- - -- -->", XML_PROLOG + "<hello></hello>");
testValidTrim(" <!-- - -- --> ", "<hello></hello>");
testValidTrim(" <!-- - -- --> ", XML_PROLOG + "<hello></hello>");
testValidTrim(" <!-- - -- --> <!-- - -- --> ", "<hello></hello>");
testValidTrim(" <!-- - -- --> <!-- - -- --> ", XML_PROLOG + "<hello></hello>");
testValidTrim(" <!-- - -- --> \n <!-- - -- --> ", "<hello></hello>");
testValidTrim(" <!-- - -- --> \n <!-- - -- --> ", XML_PROLOG + "<hello></hello>");
testInvalidTrim("x", "<hello></hello>");
testInvalidTrim("x", XML_PROLOG + "<hello></hello>");
testInvalidTrim(" x", "<hello></hello>");
testInvalidTrim(" x", XML_PROLOG + "<hello></hello>");
testInvalidTrim(" x\n", "<hello></hello>");
testInvalidTrim(" x\n", XML_PROLOG + "<hello></hello>");
testInvalidTrim("<!-- - -- - ->", "<hello></hello>");
testInvalidTrim("<!-- - -- - ->", XML_PROLOG + "<hello></hello>");
testInvalidTrim(" <!-- - -- -- >", "<hello></hello>");
testInvalidTrim(" <!-- - -- -- >", XML_PROLOG + "<hello></hello>");
testInvalidTrim(" <!-- - -- -->x ", "<hello></hello>");
testInvalidTrim(" <!-- - -- -->x ", XML_PROLOG + "<hello></hello>");
testInvalidTrim(" <!-- - -- --> x <!-- - -- --> ", "<hello></hello>");
testInvalidTrim(" <!-- - -- --> x <!-- - -- --> ", XML_PROLOG + "<hello></hello>");
testInvalidTrim(" <!-- - -- --> x\n <!-- - -- --> ", "<hello></hello>");
testInvalidTrim(" <!-- - -- --> x\n <!-- - -- --> ", XML_PROLOG + "<hello></hello>");
}
public void testAmpHandling() throws Exception {
final String input = "& &aa &";
final BufferedReader reader = new BufferedReader(new XmlFixerReader(new StringReader(input)));
final String output = reader.readLine();
reader.close();
assertEquals("& &aa &", output);
}
public void testHtmlEntities() throws Exception {
testValidEntities("<hello></hello>");
testValidEntities(XML_PROLOG + "<hello></hello>");
testValidEntities(" <!-- just in case -->\n" + XML_PROLOG + "<hello></hello>");
testValidEntities("<hello>'¥ú¥</hello>");
testValidEntities(XML_PROLOG + "<hello>'¥ú¥</hello>");
testValidEntities(" <!-- just in case -->\n" + XML_PROLOG + "<hello>'¥ú¥</hello>");
testValidEntities("<hello>ΠΡ#913;Ρ</hello>");
testValidEntities(XML_PROLOG + "<hello>ΠΡΑΡ</hello>");
testValidEntities(" <!-- just in case -->\n" + XML_PROLOG + "<hello>ΠΡΑΡ</hello>");
testValidEntities("<hello>Œ—–—</hello>");
testValidEntities(XML_PROLOG + "<hello>Œ—–—</hello>");
testValidEntities(" <!-- just in case -->\n" + XML_PROLOG + "<hello>Œ—–—</hello>");
testInvalidEntities("<hello>'&yexn;ú¥</hello>");
testInvalidEntities(XML_PROLOG + "<hello>'&yexn;ú¥</hello>");
testInvalidEntities(" <!-- just in case -->\n" + XML_PROLOG + "<hello>'&yexn;ú¥</hello>");
testInvalidEntities("<hello>Π&Rhox;#913;Ρ</hello>");
testInvalidEntities(XML_PROLOG + "<hello>Π&Rhox;ΑΡ</hello>");
testInvalidEntities(" <!-- just in case -->\n" + XML_PROLOG + "<hello>Π&Rhox;ΑΡ</hello>");
testInvalidEntities("<hello>'¥x50;¥</hello>");
testInvalidEntities(XML_PROLOG + "<hello>'¥x50;¥</hello>");
testInvalidEntities(" <!-- just in case -->\n" + XML_PROLOG + "<hello>'¥x50;¥</hello>");
testInvalidEntities("<hello>ΠΡ	x13;Ρ</hello>");
testInvalidEntities(XML_PROLOG + "<hello>ΠΡ	x13;Ρ</hello>");
testInvalidEntities(" <!-- just in case -->\n" + XML_PROLOG + "<hello>ΠΡ	x13;Ρ</hello>");
}
protected void testXmlParse(final String garbish, final String xmlDoc) throws Exception {
final InputStream is = getStream(garbish, xmlDoc);
Reader reader = new XmlReader(is);
reader = new XmlFixerReader(reader);
final SAXBuilder saxBuilder = new SAXBuilder();
saxBuilder.build(reader);
}
protected void testValidTrim(final String garbish, final String xmlDoc) throws Exception {
testXmlParse(garbish, xmlDoc);
}
protected void testInvalidTrim(final String garbish, final String xmlDoc) throws Exception {
try {
testXmlParse(garbish, xmlDoc);
assertTrue(false);
} catch (final Exception ex) {
}
}
protected void testValidEntities(final String xmlDoc) throws Exception {
testXmlParse("", xmlDoc);
}
protected void testInvalidEntities(final String xmlDoc) throws Exception {
try {
testXmlParse("", xmlDoc);
assertTrue(false);
} catch (final Exception ex) {
}
}
// XML Stream generator
protected InputStream getStream(final String garbish, final String xmlDoc) throws IOException {
final ByteArrayOutputStream baos = new ByteArrayOutputStream(1024);
final Writer writer = new OutputStreamWriter(baos);
writer.write(garbish);
writer.write(xmlDoc);
writer.close();
return new ByteArrayInputStream(baos.toByteArray());
}
}