--- /dev/null 2014-08-20 22:07:27.000000000 -0700 +++ new/test/javax/xml/jaxp/functional/org/xml/sax/ptests/ResolverTest.java 2014-08-20 22:07:27.000000000 -0700 @@ -0,0 +1,133 @@ +/* + * Copyright (c) 2014, Oracle and/or its affiliates. All rights reserved. + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. + * + * This code is free software; you can redistribute it and/or modify it + * under the terms of the GNU General Public License version 2 only, as + * published by the Free Software Foundation. + * + * This code is distributed in the hope that it will be useful, but WITHOUT + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License + * version 2 for more details (a copy is included in the LICENSE file that + * accompanied this code). + * + * You should have received a copy of the GNU General Public License version + * 2 along with this work; if not, write to the Free Software Foundation, + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. + * + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA + * or visit www.oracle.com if you need additional information or have any + * questions. + */ +package org.xml.sax.ptests; + +import java.io.BufferedWriter; +import java.io.FileInputStream; +import java.io.FileWriter; +import java.io.IOException; +import java.nio.file.Files; +import java.nio.file.Path; +import java.nio.file.Paths; +import javax.xml.parsers.ParserConfigurationException; +import javax.xml.parsers.SAXParser; +import javax.xml.parsers.SAXParserFactory; +import static jaxp.library.JAXPTestUtilities.compareWithGold; +import static jaxp.library.JAXPTestUtilities.failCleanup; +import static jaxp.library.JAXPTestUtilities.failUnexpected; +import static org.testng.Assert.assertTrue; +import org.xml.sax.InputSource; +import org.xml.sax.SAXException; +import org.xml.sax.XMLReader; +import org.xml.sax.helpers.XMLFilterImpl; +import static org.xml.sax.ptests.SAXTestConst.CLASS_DIR; +import static org.xml.sax.ptests.SAXTestConst.GOLDEN_DIR; +import static org.xml.sax.ptests.SAXTestConst.XML_DIR; + +/** + * Entity resolver should be invoked in XML parse. This test verifies parsing + * process by checking the output with golden file. + */ +public class ResolverTest { + /** + * Unit test for entityResolver setter. + */ + public void testResolver() { + String outputFile = CLASS_DIR + "EntityResolver.out"; + String goldFile = GOLDEN_DIR + "EntityResolverGF.out"; + String xmlFile = XML_DIR + "publish.xml"; + + try(FileInputStream instream = new FileInputStream(xmlFile); + MyEntityResolver eResolver = new MyEntityResolver(outputFile)) { + SAXParser saxParser = SAXParserFactory.newInstance().newSAXParser(); + XMLReader xmlReader = saxParser.getXMLReader(); + xmlReader.setEntityResolver(eResolver); + InputSource is = new InputSource(instream); + xmlReader.parse(is); + } catch(IOException | SAXException | ParserConfigurationException ex ) { + failUnexpected(ex); + } + // Need close the output file before we compare it with golden file. + try { + assertTrue(compareWithGold(goldFile, outputFile)); + } catch (IOException ex) { + failUnexpected(ex); + } finally { + try { + Path outputPath = Paths.get(outputFile); + if(Files.exists(outputPath)) + Files.delete(outputPath); + } catch (IOException ex) { + failCleanup(ex, outputFile); + } + } + } +} + +/** + * Simple entity resolver to write every entity to an output file. + */ +class MyEntityResolver extends XMLFilterImpl implements AutoCloseable { + /** + * FileWriter to write string to output file. + */ + private final BufferedWriter bWriter; + + /** + * Initiate FileWriter when construct a MyContentHandler. + * @param outputFileName output file name. + * @throws SAXException creation of FileWriter failed. + */ + MyEntityResolver(String outputFileName) throws SAXException { + super(); + try { + bWriter = new BufferedWriter(new FileWriter(outputFileName)); + } catch (IOException ex) { + throw new SAXException(ex); + } + } + + /** + * Write In resolveEntity tag along with publicid and systemId when meet + * resolveEntity event. + * @throws IOException error happen when writing file. + */ + @Override + public InputSource resolveEntity(String publicid, String systemid) + throws SAXException, IOException { + String str = "In resolveEntity.." + " " + publicid + " " + systemid; + bWriter.write( str, 0,str.length()); + bWriter.newLine(); + return super.resolveEntity(publicid, systemid); + } + + /** + * Flush the content and close the file. + * @throws IOException error happen when writing file or closing file. + */ + @Override + public void close() throws IOException { + bWriter.flush(); + bWriter.close(); + } +}