1 /* 2 * Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. 8 * 9 * This code is distributed in the hope that it will be useful, but WITHOUT 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12 * version 2 for more details (a copy is included in the LICENSE file that 13 * accompanied this code). 14 * 15 * You should have received a copy of the GNU General Public License version 16 * 2 along with this work; if not, write to the Free Software Foundation, 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18 * 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20 * or visit www.oracle.com if you need additional information or have any 21 * questions. 22 */ 23 24 package transform; 25 26 import java.io.StringReader; 27 import java.io.StringWriter; 28 29 import javax.xml.transform.Source; 30 import javax.xml.transform.Transformer; 31 import javax.xml.transform.TransformerException; 32 import javax.xml.transform.TransformerFactory; 33 import javax.xml.transform.URIResolver; 34 import javax.xml.transform.stream.StreamResult; 35 import javax.xml.transform.stream.StreamSource; 36 37 import org.testng.annotations.DataProvider; 38 import org.testng.annotations.Listeners; 39 import org.testng.annotations.Test; 40 41 import static org.testng.Assert.assertEquals; 42 43 /* 44 * @summary This class contains tests for XSLT functions. 45 */ 46 47 @Listeners({jaxp.library.BasePolicy.class}) 48 public class XSLTFunctionsTest { 49 50 /** 51 * @bug 8062518 52 * Verifies that a reference to the DTM created by XSLT document function is 53 * actually read from the DTM by an extension function. 54 * @param xml Content of xml file to process 55 * @param xsl stylesheet content that loads external document {@code externalDoc} 56 * with XSLT 'document' function and then reads it with 57 * DocumentExtFunc.test() function 58 * @param externalDoc Content of the external xml document 59 * @param expectedResult Expected transformation result 60 **/ 61 @Test(dataProvider = "document") 62 public void testDocument(final String xml, final String xsl, 63 final String externalDoc, final String expectedResult) throws Exception { 64 // Prepare sources for transormation 65 Source src = new StreamSource(new StringReader(xml)); 66 Source xslsrc = new StreamSource(new StringReader(xsl)); 67 68 // Create factory and transformer 69 TransformerFactory tf = TransformerFactory.newInstance(); 70 tf.setFeature("http://www.oracle.com/xml/jaxp/properties/enableExtensionFunctions", true); 71 Transformer t = tf.newTransformer( xslsrc ); 72 t.setErrorListener(tf.getErrorListener()); 73 74 // Set URI Resolver to return the newly constructed xml 75 // stream source object from xml test string 76 t.setURIResolver(new URIResolver() { 77 @Override 78 public Source resolve(String href, String base) 79 throws TransformerException { 80 if (href.contains("externalDoc")) { 81 return new StreamSource(new StringReader(externalDoc)); 82 } else { 83 return new StreamSource(new StringReader(xml)); 84 } 85 } 86 }); 87 88 // Prepare output stream 89 StringWriter xmlResultString = new StringWriter(); 90 StreamResult xmlResultStream = new StreamResult(xmlResultString); 91 92 //Transform the xml 93 t.transform(src, xmlResultStream); 94 95 // If the document can't be accessed and the bug is in place then 96 // reported exception will be thrown during transformation 97 System.out.println("Transformation result:"+xmlResultString.toString().trim()); 98 99 // Check the result - it should contain two (node name, node values) entries - 100 // one for original document, another for a document created with 101 // call to 'document' function 102 assertEquals(xmlResultString.toString().trim(), expectedResult); 103 } 104 105 @DataProvider(name = "document") 106 public static Object[][] documentTestData() { 107 return new Object[][] { 108 {documentTestXml, documentTestXsl, documentTestExternalDoc, documentTesteExpectedResult}, 109 }; 110 } 111 112 static final String documentTestXml = "<?xml version=\"1.0\" encoding=\"UTF-8\"?><Test>Doc</Test>"; 113 114 static final String documentTestExternalDoc = "<?xml version=\"1.0\" encoding=\"UTF-8\"?><Test>External Doc</Test>"; 115 116 static final String documentTestXsl = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>" 117 + "<xsl:transform version=\"1.0\"" 118 + " xmlns:xsl=\"http://www.w3.org/1999/XSL/Transform\" " 119 + " xmlns:cfunc=\"http://xml.apache.org/xalan/java/\">" 120 + "<xsl:template match=\"/\">" 121 + "<xsl:element name=\"root\">" 122 + "<xsl:variable name=\"other_doc\" select=\"document('externalDoc')\"/>" 123 + "<!-- Source -->" 124 + "<xsl:value-of select=\"cfunc:transform.DocumentExtFunc.test(/Test)\"/>" 125 + "<!-- document() -->" 126 + "<xsl:value-of select=\"cfunc:transform.DocumentExtFunc.test($other_doc/Test)\"/>" 127 + "</xsl:element></xsl:template></xsl:transform>"; 128 129 static final String documentTesteExpectedResult = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>" 130 + "<root>[Test:Doc][Test:External Doc]</root>"; 131 }