// Copyright 2018 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "garnet/lib/mime_sniffer/mime_sniffer.h"
#include "gtest/gtest.h"

namespace mime_sniffer {
namespace {

// Turn |str|, a constant string with one or more embedded NULs, along with
// a NUL terminator, into an std::string() containing just that data.
// Turn |str|, a string with one or more embedded NULs, into an std::string()
template <size_t N>
std::string MakeConstantString(const char (&str)[N]) {
  return std::string(str, N - 1);
}

static bool IsHtmlType(const std::string& content) {
  std::string mime_type;
  bool have_enough_content;
  return SniffForHTML(content.data(), content.size(), &have_enough_content,
                      &mime_type);
}

TEST(MimeSnifferTest, HtmlSniffingTest) {
  std::string result;
  EXPECT_TRUE(IsHtmlType(MakeConstantString("<!DOCTYPE html PUBLIC")));
  EXPECT_TRUE(IsHtmlType(MakeConstantString(" \n <hTmL>\n <hea")));
  EXPECT_TRUE(IsHtmlType(MakeConstantString("<html><body>text</body></html>")));
  EXPECT_TRUE(IsHtmlType(MakeConstantString("<body>text</body>")));
  EXPECT_TRUE(IsHtmlType(MakeConstantString("<p>text</p>")));
  EXPECT_TRUE(IsHtmlType(MakeConstantString("<div>text</div>")));
  EXPECT_TRUE(IsHtmlType(MakeConstantString("<head>text</head>")));
  EXPECT_TRUE(IsHtmlType(MakeConstantString("<iframe>text</iframe>")));
  EXPECT_TRUE(IsHtmlType(MakeConstantString("<h1>text</h1>")));
  EXPECT_TRUE(IsHtmlType(MakeConstantString("<br/>")));
  EXPECT_TRUE(IsHtmlType(MakeConstantString("<a>text</a>")));
  EXPECT_TRUE(IsHtmlType(MakeConstantString("<table>text</table>")));
  EXPECT_TRUE(IsHtmlType(MakeConstantString("<script>text</script>")));
  EXPECT_TRUE(IsHtmlType(MakeConstantString("<font>text</font>")));
  EXPECT_TRUE(IsHtmlType(MakeConstantString("<b>text</b>")));
  EXPECT_TRUE(IsHtmlType(MakeConstantString("<title>text</title>")));
  EXPECT_TRUE(IsHtmlType(MakeConstantString("<style>text</style>")));

  EXPECT_FALSE(
      IsHtmlType(MakeConstantString("GIF87a\n<html>\n<body>"
                                    "<script>alert('haxorzed');\n</script>"
                                    "</body></html>\n")));
  EXPECT_FALSE(
      IsHtmlType(MakeConstantString("a\n<html>\n<body>"
                                    "<script>alert('haxorzed');\n</script>"
                                    "</body></html>\n")));
}

}  // namespace
}  // namespace mime_sniffer