1   /*
2    * Copyright 2005 the original author or authors.
3    *
4    * Licensed under the Apache License, Version 2.0 (the "License");
5    * you may not use this file except in compliance with the License.
6    * You may obtain a copy of the License at
7    *
8    *      http://www.apache.org/licenses/LICENSE-2.0
9    *
10   * Unless required by applicable law or agreed to in writing, software
11   * distributed under the License is distributed on an "AS IS" BASIS,
12   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13   * See the License for the specific language governing permissions and
14   * limitations under the License.
15   */
16  package org.springframework.xml.namespace;
17  
18  import java.util.Collections;
19  import java.util.Iterator;
20  
21  import javax.xml.XMLConstants;
22  
23  import junit.framework.TestCase;
24  
25  public class SimpleNamespaceContextTest extends TestCase {
26  
27      private SimpleNamespaceContext context;
28  
29      protected void setUp() throws Exception {
30          context = new SimpleNamespaceContext();
31          context.bindNamespaceUri("prefix", "namespaceURI");
32      }
33  
34      public void testGetNamespaceURI() {
35          assertEquals("Invalid namespaceURI for default namespace", "", context
36                  .getNamespaceURI(XMLConstants.DEFAULT_NS_PREFIX));
37          String defaultNamespaceUri = "defaultNamespace";
38          context.bindNamespaceUri(XMLConstants.DEFAULT_NS_PREFIX, defaultNamespaceUri);
39          assertEquals("Invalid namespaceURI for default namespace", defaultNamespaceUri, context
40                  .getNamespaceURI(XMLConstants.DEFAULT_NS_PREFIX));
41          assertEquals("Invalid namespaceURI for bound prefix", "namespaceURI", context.getNamespaceURI("prefix"));
42          assertEquals("Invalid namespaceURI for unbound prefix", "", context.getNamespaceURI("unbound"));
43          assertEquals("Invalid namespaceURI for namespace prefix", XMLConstants.XML_NS_URI, context
44                  .getNamespaceURI(XMLConstants.XML_NS_PREFIX));
45          assertEquals("Invalid namespaceURI for attribute prefix", XMLConstants.XMLNS_ATTRIBUTE_NS_URI, context
46                  .getNamespaceURI(XMLConstants.XMLNS_ATTRIBUTE));
47  
48      }
49  
50      public void testGetPrefix() {
51          assertEquals("Invalid prefix for default namespace", XMLConstants.DEFAULT_NS_PREFIX, context.getPrefix(""));
52          assertEquals("Invalid prefix for bound namespace", "prefix", context.getPrefix("namespaceURI"));
53          assertNull("Invalid prefix for unbound namespace", context.getPrefix("unbound"));
54          assertEquals("Invalid prefix for namespace", XMLConstants.XML_NS_PREFIX, context
55                  .getPrefix(XMLConstants.XML_NS_URI));
56          assertEquals("Invalid prefix for attribute namespace", XMLConstants.XMLNS_ATTRIBUTE, context
57                  .getPrefix(XMLConstants.XMLNS_ATTRIBUTE_NS_URI));
58      }
59  
60      public void testGetPrefixes() {
61          assertPrefixes("", XMLConstants.DEFAULT_NS_PREFIX);
62          assertPrefixes("namespaceURI", "prefix");
63          assertFalse("Invalid prefix for unbound namespace", context.getPrefixes("unbound").hasNext());
64          assertPrefixes(XMLConstants.XML_NS_URI, XMLConstants.XML_NS_PREFIX);
65          assertPrefixes(XMLConstants.XMLNS_ATTRIBUTE_NS_URI, XMLConstants.XMLNS_ATTRIBUTE);
66      }
67  
68      public void testMultiplePrefixes() {
69          context.bindNamespaceUri("prefix1", "namespace");
70          context.bindNamespaceUri("prefix2", "namespace");
71          Iterator iterator = context.getPrefixes("namespace");
72          assertNotNull("getPrefixes returns null", iterator);
73          assertTrue("iterator is empty", iterator.hasNext());
74          String result = (String) iterator.next();
75          assertTrue("Invalid prefix", result.equals("prefix1") || result.equals("prefix2"));
76          assertTrue("iterator is empty", iterator.hasNext());
77          result = (String) iterator.next();
78          assertTrue("Invalid prefix", result.equals("prefix1") || result.equals("prefix2"));
79          assertFalse("iterator contains more than two values", iterator.hasNext());
80      }
81  
82      private void assertPrefixes(String namespaceUri, String prefix) {
83          Iterator iterator = context.getPrefixes(namespaceUri);
84          assertNotNull("getPrefixes returns null", iterator);
85          assertTrue("iterator is empty", iterator.hasNext());
86          String result = (String) iterator.next();
87          assertEquals("Invalid prefix", prefix, result);
88          assertFalse("iterator contains multiple values", iterator.hasNext());
89      }
90  
91      public void testGetBoundPrefixes() throws Exception {
92          Iterator iterator = context.getBoundPrefixes();
93          assertNotNull("getPrefixes returns null", iterator);
94          assertTrue("iterator is empty", iterator.hasNext());
95          String result = (String) iterator.next();
96          assertEquals("Invalid prefix", "prefix", result);
97          assertFalse("iterator contains multiple values", iterator.hasNext());
98      }
99  
100     public void testSetBindings() throws Exception {
101         context.setBindings(Collections.singletonMap("prefix", "namespace"));
102         assertEquals("Invalid namespace uri", "namespace", context.getNamespaceURI("prefix"));
103     }
104 }