    1   /*
    2    * $Header$
    3    * $Revision$
    4    * $Date$
    5    *
    6    * ====================================================================
    7    *
    8    * Copyright (C) 2005 Elliotte Rusty Harold
    9    * All rights reserved.
    10   *
    11   * Redistribution and use in source and binary forms, with or without
    12   * modification, are permitted provided that the following conditions
    13   * are met:
    14   *
    15   * 1. Redistributions of source code must retain the above copyright
    16   *    notice, this list of conditions, and the following disclaimer.
    17   *
    18   * 2. Redistributions in binary form must reproduce the above copyright
    19   *    notice, this list of conditions, and the disclaimer that follows
    20   *    these conditions in the documentation and/or other materials
    21   *    provided with the distribution.
    22   *
    23   * 3. The name "Jaxen" must not be used to endorse or promote products
    24   *    derived from this software without prior written permission.  For
    25   *    written permission, please contact license@jaxen.org.
    26   *
    27   * 4. Products derived from this software may not be called "Jaxen", nor
    28   *    may "Jaxen" appear in their name, without prior written permission
    29   *    from the Jaxen Project Management (pm@jaxen.org).
    30   *
    31   * In addition, we request (but do not require) that you include in the
    32   * end-user documentation provided with the redistribution and/or in the
    33   * software itself an acknowledgement equivalent to the following:
    34   *     "This product includes software developed by the
    35   *      Jaxen Project (http://www.jaxen.org/)."
    36   * Alternatively, the acknowledgment may be graphical using the logos
    37   * available at http://www.jaxen.org/
    38   *
    39   * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
    40   * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
    41   * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
    42   * DISCLAIMED.  IN NO EVENT SHALL THE Jaxen AUTHORS OR THE PROJECT
    43   * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
    44   * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
    45   * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
    46   * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
    47   * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
    48   * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
    49   * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
    50   * SUCH DAMAGE.
    51   *
    52   * ====================================================================
    53   * This software consists of voluntary contributions made by many
    54   * individuals on behalf of the Jaxen Project and was originally
    55   * created by bob mcwhirter <bob@werken.com> and
    56   * James Strachan <jstrachan@apache.org>.  For more information on the
    57   * Jaxen Project, please see <http://www.jaxen.org/>.
    58   *
    59   * $Id$
    60   */
    61
    62  package org.jaxen.function;
    63
    64  import java.io.IOException;
    65  import java.io.StringReader;
    66  import java.util.List;
    67
    68  import javax.xml.parsers.DocumentBuilder;
    69  import javax.xml.parsers.DocumentBuilderFactory;
    70  import javax.xml.parsers.ParserConfigurationException;
    71
    72  import junit.framework.TestCase;
    73
    74  import org.jaxen.BaseXPath;
    75  import org.jaxen.FunctionCallException;
    76  import org.jaxen.JaxenException;
    77  import org.jaxen.dom.DOMXPath;
    78  import org.w3c.dom.Attr;
    79  import org.w3c.dom.Document;
    80  import org.w3c.dom.Element;
    81  import org.xml.sax.InputSource;
    82  import org.xml.sax.SAXException;
    83
    84  /***
    85   * @author Elliotte Rusty Harold
    86   *
    87   */
    88  public class IdTest extends TestCase {
    89
    90      private Document doc;
    91      private DocumentBuilder builder;
    92
    93      public void setUp() throws ParserConfigurationException
    94      {
    95          DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
    96          factory.setNamespaceAware(true);
    97          builder = factory.newDocumentBuilder();
    98          doc = builder.newDocument();
    99      }
    100
    101     public IdTest(String name) {
    102         super(name);
    103     }
    104
    105     public void testIDFunctionSelectsNothingInDocumentWithNoIds()
    106       throws JaxenException {
    107
    108         BaseXPath xpath = new DOMXPath("id('p1')");
    109         org.w3c.dom.Element a = doc.createElementNS("", "a");
    110         org.w3c.dom.Element b = doc.createElementNS("", "b");
    111         doc.appendChild(a);
    112         a.appendChild(b);
    113         org.w3c.dom.Element x2 = doc.createElementNS("", "x");
    114         org.w3c.dom.Element x3 = doc.createElementNS("", "x");
    115         org.w3c.dom.Element x4 = doc.createElementNS("", "x");
    116         a.appendChild(x4);
    117         b.appendChild(x2);
    118         b.appendChild(x3);
    119         x2.appendChild(doc.createTextNode("2"));
    120         x3.appendChild(doc.createTextNode("3"));
    121         x4.appendChild(doc.createTextNode("4"));
    122         Attr id = doc.createAttribute("id");
    123         id.setNodeValue("p1");
    124         x2.setAttributeNode(id);
    125
    126         List result = xpath.selectNodes(doc);
    127         assertEquals(0, result.size());
    128
    129     }
    130
    131     public void testIDFunctionRequiresAtLeastOneArgument()
    132       throws JaxenException {
    133
    134         try {
    135             BaseXPath xpath = new DOMXPath("id()");
    136             org.w3c.dom.Element a = doc.createElementNS("", "a");
    137             doc.appendChild(a);
    138             xpath.selectNodes(doc);
    139             fail("Allowed empty id() function");
    140         }
    141         catch (FunctionCallException success) {
    142             assertNotNull(success.getMessage());
    143         }
    144
    145     }
    146
    147     public void testIDFunctionRequiresAtMostOneArgument()
    148       throws JaxenException {
    149
    150         try {
    151             BaseXPath xpath = new DOMXPath("id('p', 'q')");
    152             org.w3c.dom.Element a = doc.createElementNS("", "a");
    153             doc.appendChild(a);
    154             xpath.selectNodes(doc);
    155             fail("Allowed two-argument id() function");
    156         }
    157         catch (FunctionCallException success) {
    158             assertNotNull(success.getMessage());
    159         }
    160
    161     }
    162
    163     public void testFindElementById()
    164       throws JaxenException, SAXException, IOException {
    165
    166         BaseXPath xpath = new DOMXPath("id('p1')");
    167         String text = "<!DOCTYPE root [<!ATTLIST a id ID #REQUIRED>]><root><a id='p1'/></root>";
    168         StringReader reader = new StringReader(text);
    169         InputSource in = new InputSource(reader);
    170         Document doc = builder.parse(in);
    171         List result = xpath.selectNodes(doc);
    172         assertEquals(1, result.size());
    173         Element a = (Element) result.get(0);
    174         assertEquals("a", a.getNodeName());
    175
    176     }
    177
    178     /* public void testFindElementByXMLId()
    179       throws JaxenException, SAXException, IOException {
    180
    181         BaseXPath xpath = new DOMXPath("id('p1')");
    182         String text = "<root><a xml:id='p1'/></root>";
    183         StringReader reader = new StringReader(text);
    184         InputSource in = new InputSource(reader);
    185         Document doc = builder.parse(in);
    186         List result = xpath.selectNodes(doc);
    187         assertEquals(1, result.size());
    188         Element a = (Element) result.get(0);
    189         assertEquals("a", a.getNodeName());
    190
    191     }    */
    192
    193     public void testFindMultipleElementsByMultipleIDs()
    194       throws JaxenException, SAXException, IOException {
    195
    196         BaseXPath xpath = new DOMXPath("id(//id)");
    197         String text = "<!DOCTYPE root [<!ATTLIST a id ID #REQUIRED>]><root><id>p1</id><id>p2</id><id>p3</id><a id='p1'/><a id='p2'/></root>";
    198         StringReader reader = new StringReader(text);
    199         InputSource in = new InputSource(reader);
    200         Document doc = builder.parse(in);
    201         List result = xpath.selectNodes(doc);
    202         assertEquals(2, result.size());
    203         Element a1 = (Element) result.get(0);
    204         Element a2 = (Element) result.get(1);
    205         assertEquals("a", a1.getNodeName());
    206         assertEquals("a", a2.getNodeName());
    207
    208     }
    209
    210     public void testIdReturnsFirstElementWithMatchingId()
    211       throws JaxenException, SAXException, IOException {
    212
    213         BaseXPath xpath = new DOMXPath("id('p1')");
    214         String text = "<!DOCTYPE root [<!ATTLIST a id ID #REQUIRED>" +
    215                 "<!ATTLIST b id ID #REQUIRED>]><root><a id='p1'/><b id='p1'/></root>";
    216         StringReader reader = new StringReader(text);
    217         InputSource in = new InputSource(reader);
    218         Document doc = builder.parse(in);
    219         List result = xpath.selectNodes(doc);
    220         assertEquals(1, result.size());
    221         Element a = (Element) result.get(0);
    222         assertEquals("a", a.getNodeName());
    223
    224     }
    225
    226 }