ADD unit test for XmlMarshaller

2026-08-01 12:20:57 +00:00 · 2013-05-29 12:03:49 -07:00
parent c53b821c85
commit 3a54600a79
1 changed files with 112 additions and 0 deletions
@@ -0,0 +1,112 @@
+/**
+ * 
+ */
+package org.archive.crawler.restlet;
+
+import java.io.StringWriter;
+import java.util.LinkedHashMap;
+import java.util.Map;
+
+import javax.xml.bind.annotation.XmlRootElement;
+import javax.xml.bind.annotation.XmlType;
+
+import junit.framework.TestCase;
+
+/**
+ * @author kenji
+ *
+ */
+public class XmlMarshallerTest extends TestCase {
+
+    /* (non-Javadoc)
+     * @see junit.framework.TestCase#setUp()
+     */
+    protected void setUp() throws Exception {
+        super.setUp();
+    }
+    
+    /**
+     * Map with nested Map
+     * @throws Exception
+     */
+    public void testMarshalMap() throws Exception {
+        Map<String, Object> map = new LinkedHashMap<String, Object>();
+        map.put("c", 1);
+        map.put("a", "a-value");
+        Map<String, Object> nestedMap = new LinkedHashMap<String, Object>();
+        nestedMap.put("x", 10);
+        nestedMap.put("y", "y-value");
+        map.put("m", nestedMap);
+        
+        StringWriter w = new StringWriter();
+        XmlMarshaller.marshalDocument(w, "doc", map);
+        
+        String expected = "(?s:)" +
+                "^<\\?xml version=\"1\\.0\" standalone='yes'\\?>\\s*" +
+                "<doc>\\s*<c>1</c>\\s*<a>a-value</a>\\s*" +
+                "<m>\\s*<x>10</x>\\s*<y>y-value</y>\\s*</m>\\s*" +
+                "</doc>\\s*$";
+        String xml = w.toString();
+        System.out.println(xml);
+        assertTrue("xml matches expected RE", xml.matches(expected));
+    }
+    
+    @XmlRootElement
+    @XmlType(propOrder={"c", "a", "m"})
+    public static class Model {
+        public int getC() { return 1; }
+        public String getA() { return "a-value"; }
+        public Object getM() {
+            return new NestedModel();
+        }
+    }
+    @XmlRootElement
+    @XmlType(propOrder={"x", "y"})
+    public static class NestedModel {
+        public int getX() { return 10; }
+        public String getY() { return "y-value"; }
+    }
+    
+    /**
+     * Bean with nested Bean
+     * @throws Exception
+     */
+    public void testMashalBean() throws Exception {
+        Object bean = new Model();
+        StringWriter w = new StringWriter();
+        XmlMarshaller.marshalDocument(w, "doc", bean);
+        
+        String expected = "(?s:)" +
+                "^<\\?xml version=\"1\\.0\" standalone='yes'\\?>\\s*" +
+                "<doc>\\s*<c>1</c>\\s*<a>a-value</a>\\s*" +
+                "<m>\\s*<x>10</x>\\s*<y>y-value</y>\\s*</m>\\s*" +
+                "</doc>\\s*$";
+        String xml = w.toString();
+        System.out.println(xml);
+        assertTrue("xml matches expected RE", xml.matches(expected));
+    }
+
+    /**
+     * Map with nested Bean
+     * @throws Exception
+     */
+    public void testMarshalBeanInMap() throws Exception {
+        Map<String, Object> map = new LinkedHashMap<String, Object>();
+        map.put("c", 1);
+        map.put("a", "a-value");
+        Object m = new NestedModel();
+        map.put("m", m);
+        
+        StringWriter w = new StringWriter();
+        XmlMarshaller.marshalDocument(w, "doc", map);
+        
+        String expected = "(?s:)" +
+                "^<\\?xml version=\"1\\.0\" standalone='yes'\\?>\\s*" +
+                "<doc>\\s*<c>1</c>\\s*<a>a-value</a>\\s*" +
+                "<m>\\s*<x>10</x>\\s*<y>y-value</y>\\s*</m>\\s*" +
+                "</doc>\\s*$";
+        String xml = w.toString();
+        System.out.println(xml);
+        assertTrue("xml matches expected RE", xml.matches(expected));
+    }
+}