diff options
| -rw-r--r-- | src/main/java/org/zwobble/mammoth/html/Html.java | 6 | ||||
| -rw-r--r-- | src/test/java/org/zwobble/mammoth/tests/html/HtmlWriteTests.java | 16 |
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)); } |
