Escape element attribute values

1 files changed, 14 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 71a5b67..9e90e8b 100644
--- a/src/test/java/org/zwobble/mammoth/tests/html/HtmlWriteTests.java
+++ b/src/test/java/org/zwobble/mammoth/tests/html/HtmlWriteTests.java
@@ -24,6 +24,13 @@ public class HtmlWriteTests {
     }
 
     @Test
+    public void doubleQuotesInTextNodesArentHtmlEscaped() {
+        assertEquals(
+            "><&\"",
+            write(Html.text("><&\"")));
+    }
+
+    @Test
     public void canWriteSelfClosingElement() {
         assertEquals(
             "<br />",
@@ -53,6 +60,13 @@ public class HtmlWriteTests {
             write(Html.element("a", map("class", "external", "href", "http://example.com"))));
     }
 
+    @Test
+    public void attributeValuesAreEscaped() {
+        assertEquals(
+            "<a href=\"&gt;&lt;&amp;&quot;\"></a>",
+            write(Html.element("a", map("href", "><&\""))));
+    }
+
     private String write(HtmlNode node) {
         return Html.write(list(node));
     }