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 }