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