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, SAXException, IOException
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 testIdReturnsFirstElementWithMatchingId() 
179       throws JaxenException, SAXException, IOException {
180         
181         BaseXPath xpath = new DOMXPath("id('p1')");
182         String text = "<!DOCTYPE root [<!ATTLIST a id ID #REQUIRED>" +
183                 "<!ATTLIST b id ID #REQUIRED>]><root><a id='p1'/><b id='p1'/></root>";
184         StringReader reader = new StringReader(text);
185         InputSource in = new InputSource(reader);
186         Document doc = builder.parse(in);
187         List result = xpath.selectNodes(doc);
188         assertEquals(1, result.size());
189         Element a = (Element) result.get(0);
190         assertEquals("a", a.getNodeName());
191         
192     }    
193 
194 }