1 files changed, 26 insertions, 0 deletions
diff --git a/test/test_htmlutil.cc b/test/test_htmlutil.cc
new file mode 100644
index 0000000..629e646
--- /dev/null
+++ b/test/test_htmlutil.cc
@@ -0,0 +1,26 @@
+#include "common.hh"
+
+#include "htmlutil.hh"
+
+#include <gtest/gtest.h>
+
+TEST(htmlutil, escape_body) {
+  EXPECT_EQ("", html::escape(""));
+  EXPECT_EQ("foo", html::escape("foo"));
+  EXPECT_EQ("&lt;foo&gt;", html::escape("<foo>"));
+  EXPECT_EQ("foo &amp; bar", html::escape("foo & bar"));
+  EXPECT_EQ("\"&amp;lt;\" vs '&amp;gt;'",
+            html::escape("\"&lt;\" vs '&gt;'"));
+  EXPECT_EQ("&lt;&lt;&lt;", html::escape("<<<"));
+}
+
+TEST(htmlutil, escape_attribute) {
+  EXPECT_EQ("", html::escape("", html::EscapeTarget::ATTRIBUTE));
+  EXPECT_EQ("foo", html::escape("foo", html::EscapeTarget::ATTRIBUTE));
+  EXPECT_EQ("&lt;foo&gt;", html::escape("<foo>", html::EscapeTarget::ATTRIBUTE));
+  EXPECT_EQ("foo &amp; bar", html::escape("foo & bar",
+                                          html::EscapeTarget::ATTRIBUTE));
+  EXPECT_EQ("&quot;&amp;lt;&quot; vs &apos;&amp;gt;&apos;",
+            html::escape("\"&lt;\" vs '&gt;'", html::EscapeTarget::ATTRIBUTE));
+  EXPECT_EQ("&lt;&lt;&lt;", html::escape("<<<", html::EscapeTarget::ATTRIBUTE));
+}