Add tests for writing elements

2 files changed, 22 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/mammoth/html/Html.java b/src/main/java/org/zwobble/mammoth/html/Html.java
index 9b6ccbb..b0c864f 100644
--- a/src/main/java/org/zwobble/mammoth/html/Html.java
+++ b/src/main/java/org/zwobble/mammoth/html/Html.java
@@ -2,6 +2,8 @@ package org.zwobble.mammoth.html;
 
 import java.util.List;
 
+import static org.zwobble.mammoth.util.MammothLists.list;
+
 public class Html {
     public static String write(List<HtmlNode> nodes) {
         StringBuilder builder = new StringBuilder();
@@ -13,6 +15,10 @@ public class Html {
         return new HtmlTextNode(value);
     }
 
+    public static HtmlNode element(String tagName) {
+        return element(tagName, list());
+    }
+
     public static HtmlNode element(String tagName, List<HtmlNode> children) {
         return new HtmlElementNode(tagName, children);
     }
diff --git a/src/test/java/org/zwobble/mammoth/tests/html/HtmlWriteTests.java b/src/test/java/org/zwobble/mammoth/tests/html/HtmlWriteTests.java
index 9ce869d..d363626 100644
--- a/src/test/java/org/zwobble/mammoth/tests/html/HtmlWriteTests.java
+++ b/src/test/java/org/zwobble/mammoth/tests/html/HtmlWriteTests.java
@@ -22,6 +22,22 @@ public class HtmlWriteTests {
             write(Html.text("><&")));
     }
 
+    @Test
+    public void canWriteElementWithNoChildren() {
+        assertEquals(
+            "<p></p>",
+            write(Html.element("p")));
+    }
+
+    @Test
+    public void canWriteElementWithChildren() {
+        assertEquals(
+            "<div><p></p><ul></ul></div>",
+            write(Html.element("div", list(
+                Html.element("p"),
+                Html.element("ul")))));
+    }
+
     private String write(HtmlNode node) {
         return Html.write(list(node));
     }