SAX_XPathSelectorsTest.java

/*
	Milyn - Copyright (C) 2006 - 2010

	This library is free software; you can redistribute it and/or
	modify it under the terms of the GNU Lesser General Public
	License (version 2.1) as published by the Free Software
	Foundation.

	This library 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 Lesser General Public License for more details:
	http://www.gnu.org/licenses/lgpl.txt
*/
package org.milyn.cdr.xpath;

import org.junit.Before;
import org.junit.Test;

import static org.junit.Assert.*;

import org.milyn.Smooks;
import org.milyn.FilterSettings;
import org.milyn.cdr.SmooksConfigurationException;
import org.xml.sax.SAXException;

import javax.xml.transform.stream.StreamSource;

import java.io.IOException;
import java.util.Properties;

/**
 * @author <a href="mailto:tom.fennelly@jboss.com">tom.fennelly@jboss.com</a>
 */
public class SAX_XPathSelectorsTest {

    private Properties namespaces;

    @Before
    public void setUp() throws Exception {
        XPathVisitor.saxVisitedBeforeElementStatic = null;
        XPathVisitor.saxVisitedAfterElementStatic = null;
        XPathVisitor.domVisitedBeforeElementStatic = null;
        XPathVisitor.domVisitedAfterElementStatic = null;
        XPathAfterVisitor.saxVisitedAfterElement = null;

        namespaces = new Properties();

        namespaces.put("a", "http://a");
        namespaces.put("b", "http://b");
        namespaces.put("c", "http://c");
        namespaces.put("d", "http://d");
    }

    @Test
    public void test_01() throws IOException, SAXException {
        Smooks smooks = new Smooks(getClass().getResourceAsStream("config-01.xml"));

        smooks.setFilterSettings(FilterSettings.DEFAULT_SAX);
        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order.xml")));
        assertEquals("8655", XPathVisitor.saxVisitedBeforeElementStatic.getAttribute("code"));
        assertEquals("8655", XPathVisitor.saxVisitedAfterElementStatic.getAttribute("code"));
    }

    @Test
    public void test_02() throws IOException, SAXException {
        Smooks smooks = new Smooks(getClass().getResourceAsStream("config-02.xml"));

        try {
            smooks.setFilterSettings(FilterSettings.DEFAULT_SAX);
            smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order.xml")));
            fail("Expected SmooksConfigurationException");
        } catch(SmooksConfigurationException e) {

            assertEquals("Unsupported selector 'item[@code = '8655']/units[text() = 1]' on resource 'Target Profile: [[org.milyn.profile.Profile#default_profile]], Selector: [item[@code = '8655']/units[text() = 1]], Selector Namespace URI: [null], Resource: [org.milyn.cdr.xpath.XPathVisitor], Num Params: [0]'.  The 'text()' XPath token is only supported on SAX Visitor implementations that implement the org.milyn.delivery.sax.SAXVisitAfter interface only.  Class 'org.milyn.cdr.xpath.XPathVisitor' implements other SAX Visitor interfaces.", e.getMessage());
        }
    }

    @Test
    public void test_03() throws IOException, SAXException {
        Smooks smooks = new Smooks(getClass().getResourceAsStream("config-03.xml"));

        smooks.setFilterSettings(FilterSettings.DEFAULT_SAX);
        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order.xml")));
        assertEquals("1", XPathAfterVisitor.saxVisitedAfterElement.getTextContent());
    }

    @Test
    public void test_04() throws IOException, SAXException {
        Smooks smooks = new Smooks();

        smooks.setFilterSettings(FilterSettings.DEFAULT_SAX);
        smooks.addVisitor(new XPathVisitor(), "item[@code = 8655]");
        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order.xml")));
        assertEquals("8655", XPathVisitor.saxVisitedBeforeElementStatic.getAttribute("code"));
        assertEquals("8655", XPathVisitor.saxVisitedAfterElementStatic.getAttribute("code"));
    }

    @Test
    public void test_05() throws IOException, SAXException {
        Smooks smooks = new Smooks();

        smooks.setFilterSettings(FilterSettings.DEFAULT_SAX);
        smooks.addVisitor(new XPathVisitor(), "item[@code = '8655']/units[text() = 1]");
        try {
            smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order.xml")));
            fail("Expected SmooksConfigurationException");
        } catch(SmooksConfigurationException e) {
            assertEquals("Unsupported selector 'item[@code = '8655']/units[text() = 1]' on resource 'Target Profile: [[org.milyn.profile.Profile#default_profile]], Selector: [item[@code = '8655']/units[text() = 1]], Selector Namespace URI: [null], Resource: [org.milyn.cdr.xpath.XPathVisitor], Num Params: [0]'.  The 'text()' XPath token is only supported on SAX Visitor implementations that implement the org.milyn.delivery.sax.SAXVisitAfter interface only.  Class 'org.milyn.cdr.xpath.XPathVisitor' implements other SAX Visitor interfaces.", e.getMessage());
        }
    }

    @Test
    public void test_06() throws IOException, SAXException {
        Smooks smooks = new Smooks();

        smooks.setFilterSettings(FilterSettings.DEFAULT_SAX);
        smooks.addVisitor(new XPathAfterVisitor(), "item[@code = '8655']/units[text() = 1]");
        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order.xml")));
        assertEquals("1", XPathAfterVisitor.saxVisitedAfterElement.getTextContent());
    }

    @Test
    public void test_07() throws IOException, SAXException {
        Smooks smooks = new Smooks(getClass().getResourceAsStream("config-04.xml"));

        smooks.setFilterSettings(FilterSettings.DEFAULT_SAX);
        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order.xml")));
        assertEquals("1", XPathAfterVisitor.saxVisitedAfterElement.getTextContent());
    }

    @Test
    public void test_08() throws IOException, SAXException {
        Smooks smooks = new Smooks(getClass().getResourceAsStream("config-05.xml"));

        smooks.setFilterSettings(FilterSettings.DEFAULT_SAX);
        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order.xml")));
        assertEquals(null, XPathAfterVisitor.saxVisitedAfterElement);
    }

    @Test
    public void test_09() throws IOException, SAXException {
        Smooks smooks = new Smooks(getClass().getResourceAsStream("config-06.xml"));

        smooks.setFilterSettings(FilterSettings.DEFAULT_SAX);
        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order.xml")));
        assertEquals("1", XPathAfterVisitor.saxVisitedAfterElement.getTextContent());
    }

    @Test
    public void test_10() throws IOException, SAXException {
        Smooks smooks = new Smooks();

        smooks.setFilterSettings(FilterSettings.DEFAULT_SAX);
        smooks.setNamespaces(namespaces);

        smooks.addVisitor(new XPathAfterVisitor(), "/a:ord[@num = 3122 and @state = 'finished']/a:items/c:item[@c:code = '8655']/d:units[text() = 1]");
        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order.xml")));
        assertEquals("1", XPathAfterVisitor.saxVisitedAfterElement.getTextContent());
    }

    @Test
    public void test_indexevaluator_sax_01() throws Exception {
        Smooks smooks = new Smooks(getClass().getResourceAsStream("config-07.xml"));

        smooks.setFilterSettings(FilterSettings.DEFAULT_SAX);
        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order.xml")));
        
        assertTrue(XPathVisitor.saxVisitedBeforeElementStatic != null);
        assertTrue(XPathVisitor.saxVisitedAfterElementStatic != null);
    }

    @Test
    public void test_indexevaluator_sax_02() {
        Smooks smooks = new Smooks();

        smooks.setFilterSettings(FilterSettings.DEFAULT_SAX);
        smooks.addVisitor(new XPathVisitor(), "items/item[2]/units");
        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order.xml")));

        assertTrue(XPathVisitor.saxVisitedBeforeElementStatic != null);
        assertTrue(XPathVisitor.saxVisitedAfterElementStatic != null);
    }

    @Test
    public void test_indexevaluator_sax_03() throws Exception {
        Smooks smooks = new Smooks(getClass().getResourceAsStream("config-08.xml"));

        smooks.setFilterSettings(FilterSettings.DEFAULT_SAX);
        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order.xml")));

        assertTrue(XPathVisitor.saxVisitedBeforeElementStatic == null);
        assertTrue(XPathVisitor.saxVisitedAfterElementStatic == null);
    }

    @Test
    public void test_indexevaluator_sax_04() {
        Smooks smooks = new Smooks();

        smooks.setFilterSettings(FilterSettings.DEFAULT_SAX);
        smooks.addVisitor(new XPathVisitor(), "items/item[3]/units");
        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order.xml")));

        assertTrue(XPathVisitor.saxVisitedBeforeElementStatic == null);
        assertTrue(XPathVisitor.saxVisitedAfterElementStatic == null);
    }

    @Test
    public void test_indexevaluator_sax_05() {
        Smooks smooks = new Smooks();
        XPathVisitor visitor1 = new XPathVisitor();
        XPathVisitor visitor2 = new XPathVisitor();

        smooks.setFilterSettings(FilterSettings.DEFAULT_SAX);

        smooks.addVisitor(visitor1, "items[1]/item[2]/units");
        smooks.addVisitor(visitor2, "items[2]/item[1]/units");

        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order_02.xml")));

        assertEquals("2", visitor1.getSaxVisitedAfterElement().getAttribute("index"));
        assertEquals("1", visitor2.getSaxVisitedAfterElement().getAttribute("index"));
    }

    @Test
    public void test_indexevaluator_sax_06() {
        Smooks smooks = new Smooks();
        XPathVisitor visitor1 = new XPathVisitor();
        XPathVisitor visitor2 = new XPathVisitor();

        smooks.setFilterSettings(FilterSettings.DEFAULT_SAX);
        smooks.setNamespaces(namespaces);

        smooks.addVisitor(visitor1, "items[1]/c:item[2]/units");
        smooks.addVisitor(visitor2, "items[2]/c:item[1]/units");

        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order_02.xml")));

        assertEquals("2", visitor1.getSaxVisitedAfterElement().getAttribute("index"));
        assertEquals("1", visitor2.getSaxVisitedAfterElement().getAttribute("index"));
    }


    @Test
    public void test_indexevaluator_sax_07() {
        Smooks smooks = new Smooks();
        XPathVisitor visitor1 = new XPathVisitor();
        XPathVisitor visitor2 = new XPathVisitor();

        smooks.setFilterSettings(FilterSettings.DEFAULT_SAX);
        smooks.setNamespaces(namespaces);

        smooks.addVisitor(visitor1, "items[1]/d:item[2]/units"); // wrong namespace prefix
        smooks.addVisitor(visitor2, "items[2]/d:item[1]/units"); // wrong namespace prefix

        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order_02.xml")));

        assertEquals(null, visitor1.getSaxVisitedAfterElement());
        assertEquals(null, visitor2.getSaxVisitedAfterElement());
    }

    @Test
    public void test_indexevaluator_dom_01() throws Exception {
        Smooks smooks = new Smooks(getClass().getResourceAsStream("config-07.xml"));

        smooks.setFilterSettings(FilterSettings.DEFAULT_DOM);
        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order.xml")));

        assertTrue(XPathVisitor.domVisitedBeforeElementStatic != null);
        assertTrue(XPathVisitor.domVisitedAfterElementStatic != null);
    }

    @Test
    public void test_indexevaluator_dom_02() {
        Smooks smooks = new Smooks();

        smooks.setFilterSettings(FilterSettings.DEFAULT_DOM);
        smooks.addVisitor(new XPathVisitor(), "items/item[2]/units");
        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order.xml")));

        assertTrue(XPathVisitor.domVisitedBeforeElementStatic != null);
        assertTrue(XPathVisitor.domVisitedAfterElementStatic != null);
    }

    @Test
    public void test_indexevaluator_dom_03() throws Exception {
        Smooks smooks = new Smooks(getClass().getResourceAsStream("config-08.xml"));

        smooks.setFilterSettings(FilterSettings.DEFAULT_DOM);
        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order.xml")));

        assertTrue(XPathVisitor.domVisitedBeforeElementStatic == null);
        assertTrue(XPathVisitor.domVisitedAfterElementStatic == null);
    }

    @Test
    public void test_indexevaluator_dom_04() {
        Smooks smooks = new Smooks();

        smooks.setFilterSettings(FilterSettings.DEFAULT_DOM);
        smooks.addVisitor(new XPathVisitor(), "items/item[3]/units");
        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order.xml")));

        assertTrue(XPathVisitor.domVisitedBeforeElementStatic == null);
        assertTrue(XPathVisitor.domVisitedAfterElementStatic == null);
    }

    @Test
    public void test_indexevaluator_dom_05() {
        Smooks smooks = new Smooks();
        XPathVisitor visitor1 = new XPathVisitor();
        XPathVisitor visitor2 = new XPathVisitor();

        smooks.setFilterSettings(FilterSettings.DEFAULT_DOM);

        smooks.addVisitor(visitor1, "items[1]/item[2]/units");
        smooks.addVisitor(visitor2, "items[2]/item[1]/units");

        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order_02.xml")));

        assertEquals("2", visitor1.getDomVisitedAfterElement().getAttribute("index"));
        assertEquals("1", visitor2.getDomVisitedAfterElement().getAttribute("index"));
    }

    @Test
    public void test_indexevaluator_dom_06() {
        Smooks smooks = new Smooks();
        XPathVisitor visitor1 = new XPathVisitor();
        XPathVisitor visitor2 = new XPathVisitor();

        smooks.setFilterSettings(FilterSettings.DEFAULT_DOM);
        smooks.setNamespaces(namespaces);

        smooks.addVisitor(visitor1, "items[1]/c:item[2]/units");
        smooks.addVisitor(visitor2, "items[2]/c:item[1]/units");

        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order_02.xml")));

        assertEquals("2", visitor1.getDomVisitedAfterElement().getAttribute("index"));
        assertEquals("1", visitor2.getDomVisitedAfterElement().getAttribute("index"));
    }


    @Test
    public void test_indexevaluator_dom_07() {
        Smooks smooks = new Smooks();
        XPathVisitor visitor1 = new XPathVisitor();
        XPathVisitor visitor2 = new XPathVisitor();

        smooks.setFilterSettings(FilterSettings.DEFAULT_DOM);
        smooks.setNamespaces(namespaces);

        smooks.addVisitor(visitor1, "items[1]/d:item[2]/units"); // wrong namespace prefix
        smooks.addVisitor(visitor2, "items[2]/d:item[1]/units"); // wrong namespace prefix

        smooks.filterSource(new StreamSource(getClass().getResourceAsStream("order_02.xml")));

        assertEquals(null, visitor1.getDomVisitedAfterElement());
        assertEquals(null, visitor2.getDomVisitedAfterElement());
    }
}