aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/main/java/org/zwobble/mammoth/html/Html.java6
-rw-r--r--src/test/java/org/zwobble/mammoth/tests/html/HtmlWriteTests.java16
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));
}