Now using the dom level 3 API for parsing. Also extended the test case
[utils] / support / general / src / test / java / org / wamblee / xml / XmlUtils.java
index fc894251eb0aedf0429a6bb0ec609759edec3182..fd49a8a239f9ca858d953bcc58775d69e1132c8d 100644 (file)
@@ -1,5 +1,5 @@
 /*
- * Copyright 2005 the original author or authors.
+ * Copyright 2005-2010 the original author or authors.
  * 
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-
 package org.wamblee.xml;
 
+import java.io.Serializable;
+import java.util.ArrayList;
 import java.util.Collections;
 import java.util.Comparator;
 import java.util.List;
@@ -24,17 +25,20 @@ import java.util.TreeMap;
 
 import junit.framework.TestCase;
 
-import org.dom4j.Attribute;
-import org.dom4j.Document;
-import org.dom4j.Element;
+import org.w3c.dom.Attr;
+import org.w3c.dom.Document;
+import org.w3c.dom.Element;
+import org.w3c.dom.NamedNodeMap;
+import org.w3c.dom.Node;
+import org.w3c.dom.NodeList;
+import org.w3c.dom.Text;
 
 /**
  * XML test support utilities.
- *
+ * 
  * @author Erik Brakkee
  */
 public final class XmlUtils {
-
     /**
      * Disabled constructor.
      * 
@@ -46,116 +50,152 @@ public final class XmlUtils {
     /**
      * Checks equality of two XML documents excluding comment and processing
      * nodes and trimming the text of the elements. In case of problems, it
-     * provides an xpath-like expression describing where the problem is.  
+     * provides an xpath-like expression describing where the problem is.
      * 
      * @param aMsg
      * @param aExpected
      * @param aActual
      */
-    public static void assertEquals(String aMsg,
-            org.w3c.dom.Document aExpected, org.w3c.dom.Document aActual) {
-        assertEquals(aMsg, DomUtils.convert(aExpected), DomUtils
-                .convert(aActual));
+    public static void assertEquals(String aMsg, Document aExpected,
+        Document aActual) {
+        assertEquals(
+            aMsg + "/" + aExpected.getDocumentElement().getLocalName(),
+            aExpected.getDocumentElement(), aActual.getDocumentElement());
     }
 
-    /**
-     * Checks equality of two XML documents excluding comment and processing
-     * nodes and trimming the text of the elements.  In case of problems, it
-     * provides an xpath-like expression describing where the problem is.  
-     * 
-     * @param aMsg
-     * @param aExpected
-     * @param aActual
-     */
-    public static void assertEquals(String aMsg, Document aExpected,
-            Document aActual) {
-        assertEquals(aMsg + "/" + aExpected.getRootElement().getName(), aExpected.getRootElement(), aActual.getRootElement());
+    private static <T> List<T> convertMap(NamedNodeMap aMap, Class<T> aType) {
+        List<T> result = new ArrayList<T>();
+        for (int i = 0; i < aMap.getLength(); i++) {
+            result.add((T) aMap.item(i));
+        }
+        return result;
+    }
+
+    private static List<Element> getChildElements(Element aElement) {
+        List<Element> result = new ArrayList<Element>();
+        NodeList children = aElement.getChildNodes();
+        for (int i = 0; i < children.getLength(); i++) {
+            Node node = children.item(i);
+            if (node instanceof Element) {
+                result.add((Element) node);
+            }
+        }
+        return result;
+    }
+    
+    private static String getDirectText(Element aElement) { 
+        String res = "";
+        NodeList children = aElement.getChildNodes();
+        for (int i = 0; i < children.getLength(); i++) {
+            Node node = children.item(i);
+            if (node instanceof Text) {
+                res += node.getTextContent() + " ";
+            }
+        }
+        return trim(res); 
+    }
+    
+    private static String trim(String aText) { 
+        return aText.trim().replaceAll("\\n", " ").replaceAll("\\s+", " ");
     }
 
     /**
      * Checks equality of two XML elements excluding comment and processing
-     * nodes and trimming the text of the elements. In case of problems, it
-     * provides an xpath-like expression describing where the problem is.  
+     * nodes and trimming the text of the elements (including whitespace in the middle). 
+     * In case of problems, it
+     * provides an xpath-like expression describing where the problem is.
      * 
      * @param aMsg
      * @param aExpected
      * @param aActual
      */
-    public static void assertEquals(String aMsg, Element aExpected,
-            Element aActual) {
-
+    private static void assertEquals(String aMsg, Element aExpected,
+        Element aActual) {
         // Name.
-        TestCase.assertEquals(aMsg + "/name()", aExpected.getName(), aActual
-                .getName());
+        TestCase.assertEquals(aMsg + "/name()", aExpected.getLocalName(),
+            aActual.getLocalName());
+        TestCase.assertEquals(aMsg + "/namespace()",
+            aExpected.getNamespaceURI(), aActual.getNamespaceURI());
 
         // Text
-        TestCase.assertEquals(aMsg + "/text()", aExpected.getTextTrim(),
-                aActual.getTextTrim());
+        TestCase.assertEquals(aMsg + "/text()", getDirectText(aExpected)
+            , getDirectText(aActual));
 
         // Attributes
-        List<Attribute> expectedAttrs = aExpected.attributes();
+        List<Attr> expectedAttrs = convertMap(aExpected.getAttributes(),
+            Attr.class);
+
         Collections.sort(expectedAttrs, new AttributeComparator());
-        List<Attribute> actualAttrs = aActual.attributes();
+
+        List<Attr> actualAttrs = convertMap(aActual.getAttributes(), Attr.class);
         Collections.sort(actualAttrs, new AttributeComparator());
 
         TestCase.assertEquals("count(" + aMsg + "/@*)", expectedAttrs.size(),
-                actualAttrs.size());
+            actualAttrs.size());
+
         for (int i = 0; i < expectedAttrs.size(); i++) {
             String msg = aMsg + "/@" + expectedAttrs.get(i).getName();
             assertEquals(msg, expectedAttrs.get(i), actualAttrs.get(i));
         }
 
         // Nested elements.
-        List<Element> expectedElems = aExpected.elements();
-        List<Element> actualElems = aActual.elements();
+        List<Element> expectedElems = getChildElements(aExpected);
+        List<Element> actualElems = getChildElements(aActual);
         TestCase.assertEquals("count(" + aMsg + "/*)", expectedElems.size(),
-                actualElems.size());
+            actualElems.size());
+
         // determine the how-manyth element of the given name we are at.
         // Maps element name to the last used index (or null if not yet used)
         Map<String, Integer> elementIndex = new TreeMap<String, Integer>();
+
         for (int i = 0; i < expectedElems.size(); i++) {
-            String elemName = expectedElems.get(i).getName();
+            String elemName = expectedElems.get(i).getLocalName();
             Integer index = elementIndex.get(elemName);
+
             if (index == null) {
                 index = 1;
             } else {
                 index++;
             }
+
             elementIndex.put(elemName, index);
-            String msg = aMsg + "/" + expectedElems.get(i).getName() + "["
-                    + index + "]";
+
+            String msg = aMsg + "/" + expectedElems.get(i).getLocalName() + "[" +
+                index + "]";
 
             assertEquals(msg, expectedElems.get(i), actualElems.get(i));
         }
     }
 
     /**
-     * Checks equality of two attributes. In case of problems, it
-     * provides an xpath-like expression describing where the problem is.  
+     * Checks equality of two attributes. In case of problems, it provides an
+     * xpath-like expression describing where the problem is.
      * 
      * @param aMsg
      * @param aExpected
      * @param aActual
      */
-    public static void assertEquals(String aMsg, Attribute aExpected,
-            Attribute aActual) {
-        TestCase.assertEquals(aMsg + ":name", aExpected.getName(), aActual
-                .getName());
-        TestCase.assertEquals(aMsg + ":value", aExpected.getValue(), aActual
-                .getValue());
+    private static void assertEquals(String aMsg, Attr aExpected, Attr aActual) {
+        TestCase.assertEquals(aMsg + ":name", aExpected.getName(),
+            aActual.getName());
+        TestCase.assertEquals(aMsg + ":value", aExpected.getValue(),
+            aActual.getValue());
     }
 
     /**
      * Comparator which compares attributes by name.
      */
-    private static final class AttributeComparator implements
-            Comparator<Attribute> {
+    private static final class AttributeComparator implements Comparator<Attr>,
+        Serializable {
+
+        private static final long serialVersionUID = 7897287273519886301L;
+
         /*
          * (non-Javadoc)
          * 
          * @see java.util.Comparator#compare(T, T)
          */
-        public int compare(Attribute aAttribute1, Attribute aAttribute2) {
+        public int compare(Attr aAttribute1, Attr aAttribute2) {
             return aAttribute1.getName().compareTo(aAttribute2.getName());
         }
     }