JUnit Test cases to test URI resolution using custom URIResolvers. They are not yet integrated into the test suite as the second test fails and the first test needs some changes in the code (see TODO).

git-svn-id: https://svn.apache.org/repos/asf/xmlgraphics/fop/trunk@232788 13f79535-47bb-0310-9956-ffa450edef68

1 files changed, 209 insertions, 0 deletions

diff --git a/test/java/org/apache/fop/URIResolutionTestCase.java b/test/java/org/apache/fop/URIResolutionTestCase.java
new file mode 100644
index 000000000..148195c88
--- /dev/null
+++ b/test/java/org/apache/fop/URIResolutionTestCase.java
@@ -0,0 +1,209 @@
+/*
+ * Copyright 2005 The Apache Software Foundation.
+ * 
+ * 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.
+ */
+
+/* $Id$ */
+ 
+package org.apache.fop;
+
+import java.io.File;
+import java.io.FileNotFoundException;
+import java.io.OutputStream;
+
+import javax.xml.transform.Result;
+import javax.xml.transform.Source;
+import javax.xml.transform.Transformer;
+import javax.xml.transform.TransformerConfigurationException;
+import javax.xml.transform.TransformerException;
+import javax.xml.transform.URIResolver;
+import javax.xml.transform.dom.DOMResult;
+import javax.xml.transform.dom.DOMSource;
+import javax.xml.transform.sax.SAXResult;
+import javax.xml.transform.sax.SAXTransformerFactory;
+import javax.xml.transform.sax.TransformerHandler;
+import javax.xml.transform.stream.StreamResult;
+import javax.xml.transform.stream.StreamSource;
+
+import org.apache.commons.io.IOUtils;
+import org.apache.commons.io.output.ByteArrayOutputStream;
+import org.apache.fop.apps.FOPException;
+import org.apache.fop.apps.FOUserAgent;
+import org.apache.fop.apps.Fop;
+import org.apache.fop.render.xml.XMLRenderer;
+import org.apache.xpath.XPathAPI;
+import org.apache.xpath.objects.XObject;
+import org.w3c.dom.Document;
+
+/**
+ * Tests URI resolution facilities.
+ */
+public class URIResolutionTestCase extends AbstractFOPTestCase {
+
+    private SAXTransformerFactory tfactory 
+            = (SAXTransformerFactory)SAXTransformerFactory.newInstance();
+
+    private File backupDir = new File(getBaseDir(), "build/test-results");
+    
+    /** @see junit.framework.TestCase#TestCase(String) */
+    public URIResolutionTestCase(String name) {
+        super(name);
+    }
+
+    /**
+     * Test custom URI resolution with a hand-written URIResolver.
+     * @throws Exception if anything fails
+     */
+    public void testFO1() throws Exception {
+        File foFile = new File(getBaseDir(), "test/xml/uri-resolution1.fo");
+        
+        FOUserAgent ua = new FOUserAgent();
+        MyURIResolver resolver = new MyURIResolver(); 
+        ua.setURIResolver(resolver);
+        ua.setBaseURL(foFile.getParentFile().toURL().toString());
+
+        Document doc = createAreaTree(foFile, ua);
+        
+        //Check how many times the resolver was consulted
+        assertEquals(1, resolver.successCount);
+        assertEquals(0, resolver.failureCount);
+        //Additional XPath checking on the area tree
+        assertEquals("viewport for external-graphic is missing", 
+                "true", evalXPath(doc, "boolean(//flow/block[1]/lineArea/viewport)"));
+        assertEquals("46080", evalXPath(doc, "//flow/block[1]/lineArea/viewport/@ipd"));
+        assertEquals("46080", evalXPath(doc, "//flow/block[1]/lineArea/viewport/@bpd"));
+    }
+
+    /**
+     * Test custom URI resolution with a hand-written URIResolver.
+     * @throws Exception if anything fails
+     */
+    public void testFO2() throws Exception {
+        File foFile = new File(getBaseDir(), "test/xml/uri-resolution2.fo");
+        
+        FOUserAgent ua = new FOUserAgent();
+        MyURIResolver resolver = new MyURIResolver(); 
+        ua.setURIResolver(resolver);
+        ua.setBaseURL(foFile.getParentFile().toURL().toString());
+
+        Fop fop = new Fop(Fop.RENDER_PDF, ua);
+
+        ByteArrayOutputStream baout = new ByteArrayOutputStream();
+        fop.setOutputStream(baout);
+
+        Transformer transformer = tfactory.newTransformer(); //Identity transf.
+        Source src = new StreamSource(foFile);
+        Result res = new SAXResult(fop.getDefaultHandler());
+        transformer.transform(src, res);
+        
+        OutputStream out = new java.io.FileOutputStream(
+                new File(backupDir, foFile.getName() + ".pdf"));
+        try {
+            baout.writeTo(out);
+        } finally {
+            IOUtils.closeQuietly(out);
+        }
+        
+        //Check how many times the resolver was consulted
+        assertEquals(1, resolver.successCount);
+        assertEquals(0, resolver.failureCount);
+        //Test using PDF as the area tree doesn't invoke Batik so we could check
+        //if the resolver is actually passed to Batik by FOP
+        assertTrue("Generated PDF has zero length", baout.size() > 0);
+    }
+
+    private Document createAreaTree(File fo, FOUserAgent ua) 
+                throws TransformerException, FOPException {
+        DOMResult domres = new DOMResult();
+        //Setup Transformer to convert the area tree to a DOM
+        TransformerHandler athandler = tfactory.newTransformerHandler();
+        athandler.setResult(domres);
+
+        XMLRenderer atrenderer = new XMLRenderer();
+        atrenderer.setUserAgent(ua);
+        atrenderer.setTransformerHandler(athandler);
+        ua.setRendererOverride(atrenderer);
+        
+        Fop fop = new Fop(Fop.RENDER_XML, ua);
+
+        Transformer transformer = tfactory.newTransformer(); //Identity transf.
+        Source src = new StreamSource(fo);
+        Result res = new SAXResult(fop.getDefaultHandler());
+        transformer.transform(src, res);
+        
+        Document doc = (Document)domres.getNode();
+        saveAreaTreeXML(doc, new File(backupDir, fo.getName() + ".at.xml"));
+        return doc;
+    }
+    
+    private String evalXPath(Document doc, String xpath) {
+        XObject res;
+        try {
+            res = XPathAPI.eval(doc, xpath);
+        } catch (TransformerException e) {
+            throw new RuntimeException("XPath evaluation failed: " + e.getMessage());
+        }
+        return res.str();
+    }
+
+    /**
+     * Save the area tree XML for later inspection.
+     * @param doc area tree as a DOM document
+     * @param target target file
+     * @throws TransformerException if a problem occurs during serialization
+     */
+    protected void saveAreaTreeXML(Document doc, File target) throws TransformerException {
+        Transformer transformer = tfactory.newTransformer();
+        Source src = new DOMSource(doc);
+        Result res = new StreamResult(target);
+        transformer.transform(src, res);
+    }
+    
+    private class MyURIResolver implements URIResolver {
+
+        private static final String PREFIX = "funky:";
+        
+        private int successCount = 0;
+        private int failureCount = 0;
+        
+        /**
+         * @see javax.xml.transform.URIResolver#resolve(java.lang.String, java.lang.String)
+         */
+        public Source resolve(String href, String base) throws TransformerException {
+            if (href.startsWith(PREFIX)) {
+                String name = href.substring(PREFIX.length());
+                if ("myimage123".equals(name)) {
+                    File image = new File(getBaseDir(), "test/resources/images/bgimg300dpi.jpg");
+                    StreamSource src = new StreamSource(image);
+                    try {
+                        //TODO The next line should not be necessary!!!
+                        src.setInputStream(new java.io.FileInputStream(image));
+                    } catch (FileNotFoundException e) {
+                        throw new TransformerException(e.getMessage(), e);
+                    }
+                    successCount++;
+                    return src;
+                } else {
+                    failureCount++;
+                    throw new TransformerException("funky image not found");
+                }
+            } else {
+                failureCount++;
+                return null;
+            }
+        }
+        
+    }
+    
+}