fr.efl.saxon.basex.BaseXQueryTest.java Source code

Introduction

Here is the source code for fr.efl.saxon.basex.BaseXQueryTest.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package fr.efl.saxon.basex;

import net.sf.saxon.Configuration;
import net.sf.saxon.om.StructuredQName;
import net.sf.saxon.s9api.DocumentBuilder;
import net.sf.saxon.s9api.Processor;
import net.sf.saxon.s9api.QName;
import net.sf.saxon.s9api.XPathCompiler;
import net.sf.saxon.s9api.XPathSelector;
import net.sf.saxon.s9api.XdmNode;
import net.sf.saxon.value.SequenceType;
import org.basex.BaseXServer;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;
import javax.xml.transform.stream.StreamSource;
import net.sf.saxon.om.Item;
import net.sf.saxon.om.SequenceIterator;
import net.sf.saxon.s9api.Axis;
import net.sf.saxon.s9api.SaxonApiException;
import net.sf.saxon.s9api.XdmValue;
import net.sf.saxon.s9api.XsltCompiler;
import net.sf.saxon.s9api.XsltTransformer;
import net.sf.saxon.trans.XPathException;
import org.apache.commons.io.output.NullOutputStream;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;
import static org.junit.Assert.*;

/**
 *
 * @author ext-cmarchand
 */
public class BaseXQueryTest {
    private static final String CONNECT_STRING = "<?xml version='1.0' encoding='UTF-8'?>" + "<basex>"
            + "<server>localhost</server>" + "<port>1984</port>" + "<user>admin</user>"
            + "<password>admin</password>" + "</basex>";
    private static BaseXServer server;

    @BeforeClass
    public static void initServer() throws Exception {
        server = new BaseXServer();
    }

    @AfterClass
    public static void stopServer() throws Exception {
        if (server != null) {
            try {
                server.stop();
                server = null;
            } catch (Exception ex) {
                // ignore
            }
        }
    }

    /**
     * Test of getFunctionQName method, of class BaseXQuery.
     */
    @Test
    public void testGetFunctionQName() {
        //        System.out.println("getFunctionQName");
        BaseXQuery instance = new BaseXQuery();
        StructuredQName expResult = new StructuredQName("efl-ext", "fr:efl:saxon:extensions", "basex-query");
        StructuredQName result = instance.getFunctionQName();
        assertEquals(expResult, result);
    }

    /**
     * Test of makeCallExpression method, of class BaseXQuery.
     */
    @Test
    public void testMakeCallExpression() {
        Configuration config = new Configuration();
        config.registerExtensionFunction(new BaseXQuery());
        Processor proc = new Processor(config);
        XPathCompiler xpc = proc.newXPathCompiler();
        try {
            xpc.declareNamespace(BaseXQuery.EXT_NS_COMMON_PREFIX, BaseXQuery.EXT_NAMESPACE_URI);
            QName var = new QName("connect");
            xpc.declareVariable(var);
            XPathSelector xp = xpc.compile(BaseXQuery.EXT_NS_COMMON_PREFIX + ":" + BaseXQuery.FUNCTION_NAME
                    + "('for $i in 1 to 10 return <test>{$i}</test>', $connect)").load();
            DocumentBuilder builder = proc.newDocumentBuilder();
            XdmNode docConnect = builder
                    .build(new StreamSource(new ByteArrayInputStream(CONNECT_STRING.getBytes("UTF-8"))));
            XdmNode connect = (XdmNode) docConnect.axisIterator(Axis.DESCENDANT_OR_SELF, new QName("basex")).next();
            //            System.err.println("connect is "+connect.getClass().getName());
            //            System.err.println("connect is "+connect.getNodeName());
            //            System.err.println("connect is "+connect.getNodeKind());

            xp.setVariable(var, connect);
            xp.setContextItem(docConnect);
            XdmValue result = xp.evaluate();
            SequenceIterator it = result.getUnderlyingValue().iterate();
            Item item = it.next();
            int count = 1;
            while (item != null) {
                assertEquals(Integer.toString(count++), item.getStringValue());
                item = it.next();
            }
            it.close();
        } catch (SaxonApiException | UnsupportedEncodingException | XPathException ex) {
            ex.printStackTrace(System.err);
            fail(ex.getMessage());
        }
    }

    /**
     * Test of getArgumentTypes method, of class BaseXQuery.
     */
    @Test
    public void testGetArgumentTypes() {
        BaseXQuery instance = new BaseXQuery();
        SequenceType[] expResult = new SequenceType[] { SequenceType.SINGLE_STRING,
                SequenceType.SINGLE_ELEMENT_NODE };
        SequenceType[] result = instance.getArgumentTypes();
        assertEquals(expResult.length, result.length);
        for (int i = 0; i < expResult.length; i++)
            assertEquals("entry " + i + " differs from expected", expResult[i], result[i]);
    }

    /**
     * Test of getResultType method, of class BaseXQuery.
     */
    @Test
    public void testGetResultType() {
        SequenceType[] sts = null;
        BaseXQuery instance = new BaseXQuery();
        SequenceType expResult = SequenceType.ANY_SEQUENCE;
        SequenceType result = instance.getResultType(sts);
        assertEquals(expResult, result);
    }

    @Test
    public void testXsl() {
        Configuration config = new Configuration();
        config.registerExtensionFunction(new BaseXQuery());
        Processor proc = new Processor(config);
        XsltCompiler compiler = proc.newXsltCompiler();
        try {
            InputStream is = new FileInputStream(
                    new File(new File(System.getProperty("user.dir")), "src/test/resources/test1.xsl"));
            if (is == null) {
                fail("Impossible de charger la XSL");
            }
            XsltTransformer t = compiler.compile(new StreamSource(is)).load();
            t.setDestination(proc.newSerializer(new NullOutputStream()));
            t.setInitialContextNode(proc.newDocumentBuilder().build(new StreamSource(new FileInputStream(
                    new File(new File(System.getProperty("user.dir")), "src/test/resources/test1.xsl")))));
            t.transform();
        } catch (SaxonApiException | FileNotFoundException ex) {
            ex.printStackTrace(System.err);
            fail(ex.getMessage());
        }
    }
}