Add tests for writing elements

1 files changed, 16 insertions, 0 deletions

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));
     }