// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <stddef.h>

#include <utility>

#include "base/values.h"
#include "chrome/browser/extensions/api/omnibox/omnibox_api.h"
#include "chrome/common/extensions/api/omnibox.h"
#include "extensions/common/value_builder.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "testing/platform_test.h"

namespace extensions {

namespace omnibox = api::omnibox;
namespace SendSuggestions = omnibox::SendSuggestions;
namespace SetDefaultSuggestion = omnibox::SetDefaultSuggestion;

namespace {

const int kNone = ACMatchClassification::NONE;
const int kUrl = ACMatchClassification::URL;
const int kMatch = ACMatchClassification::MATCH;
const int kDim = ACMatchClassification::DIM;

void CompareClassification(const ACMatchClassifications& expected,
                           const ACMatchClassifications& actual) {
  EXPECT_EQ(expected.size(), actual.size());
  for (size_t i = 0; i < expected.size() && i < actual.size(); ++i) {
    EXPECT_EQ(expected[i].offset, actual[i].offset) << "Index:" << i;
    EXPECT_EQ(expected[i].style, actual[i].style) << "Index:" << i;
  }
}

}  // namespace

// Test output key: n = character with no styling, d = dim, m = match, u = url
// u = 1, m = 2, d = 4. u+d = 5, etc.

//   0123456789
//    mmmm
// +       ddd
// = nmmmmndddn
TEST(ExtensionOmniboxTest, DescriptionStylesSimple) {
  std::unique_ptr<base::ListValue> list =
      ListBuilder()
          .Append(42)
          .Append(ListBuilder()
                      .Append(DictionaryBuilder()
                                  .Set("content", "content")
                                  .Set("description", "description")
                                  .Set("descriptionStyles",
                                       ListBuilder()
                                           .Append(DictionaryBuilder()
                                                       .Set("type", "match")
                                                       .Set("offset", 1)
                                                       .Set("length", 4)
                                                       .Build())
                                           .Append(DictionaryBuilder()
                                                       .Set("type", "dim")
                                                       .Set("offset", 6)
                                                       .Set("length", 3)
                                                       .Build())
                                           .Build())
                                  .Build())
                      .Build())
          .Build();

  ACMatchClassifications styles_expected;
  styles_expected.push_back(ACMatchClassification(0, kNone));
  styles_expected.push_back(ACMatchClassification(1, kMatch));
  styles_expected.push_back(ACMatchClassification(5, kNone));
  styles_expected.push_back(ACMatchClassification(6, kDim));
  styles_expected.push_back(ACMatchClassification(9, kNone));

  std::unique_ptr<SendSuggestions::Params> params(
      SendSuggestions::Params::Create(list->GetList()));
  EXPECT_TRUE(params);
  ASSERT_FALSE(params->suggest_results.empty());
  CompareClassification(styles_expected, StyleTypesToACMatchClassifications(
                                             params->suggest_results[0]));

  // Same input, but swap the order. Ensure it still works.
  std::unique_ptr<base::ListValue> swap_list =
      ListBuilder()
          .Append(42)
          .Append(ListBuilder()
                      .Append(DictionaryBuilder()
                                  .Set("content", "content")
                                  .Set("description", "description")
                                  .Set("descriptionStyles",
                                       ListBuilder()
                                           .Append(DictionaryBuilder()
                                                       .Set("type", "dim")
                                                       .Set("offset", 6)
                                                       .Set("length", 3)
                                                       .Build())
                                           .Append(DictionaryBuilder()
                                                       .Set("type", "match")
                                                       .Set("offset", 1)
                                                       .Set("length", 4)
                                                       .Build())
                                           .Build())
                                  .Build())
                      .Build())
          .Build();

  std::unique_ptr<SendSuggestions::Params> swapped_params(
      SendSuggestions::Params::Create(swap_list->GetList()));
  EXPECT_TRUE(swapped_params);
  ASSERT_FALSE(swapped_params->suggest_results.empty());
  CompareClassification(
      styles_expected,
      StyleTypesToACMatchClassifications(swapped_params->suggest_results[0]));
}

//   0123456789
//   uuuuu
// +          dd
// +          mm
// + mmmm
// +  dd
// = 3773unnnn66
TEST(ExtensionOmniboxTest, DescriptionStylesCombine) {
  std::unique_ptr<base::ListValue> list =
      ListBuilder()
          .Append(42)
          .Append(ListBuilder()
                      .Append(DictionaryBuilder()
                                  .Set("content", "content")
                                  .Set("description", "description")
                                  .Set("descriptionStyles",
                                       ListBuilder()
                                           .Append(DictionaryBuilder()
                                                       .Set("type", "url")
                                                       .Set("offset", 0)
                                                       .Set("length", 5)
                                                       .Build())
                                           .Append(DictionaryBuilder()
                                                       .Set("type", "dim")
                                                       .Set("offset", 9)
                                                       .Set("length", 2)
                                                       .Build())
                                           .Append(DictionaryBuilder()
                                                       .Set("type", "match")
                                                       .Set("offset", 9)
                                                       .Set("length", 2)
                                                       .Build())
                                           .Append(DictionaryBuilder()
                                                       .Set("type", "match")
                                                       .Set("offset", 0)
                                                       .Set("length", 4)
                                                       .Build())
                                           .Append(DictionaryBuilder()
                                                       .Set("type", "dim")
                                                       .Set("offset", 1)
                                                       .Set("length", 2)
                                                       .Build())
                                           .Build())
                                  .Build())
                      .Build())
          .Build();

  ACMatchClassifications styles_expected;
  styles_expected.push_back(ACMatchClassification(0, kUrl | kMatch));
  styles_expected.push_back(ACMatchClassification(1, kUrl | kMatch | kDim));
  styles_expected.push_back(ACMatchClassification(3, kUrl | kMatch));
  styles_expected.push_back(ACMatchClassification(4, kUrl));
  styles_expected.push_back(ACMatchClassification(5, kNone));
  styles_expected.push_back(ACMatchClassification(9, kMatch | kDim));

  std::unique_ptr<SendSuggestions::Params> params(
      SendSuggestions::Params::Create(list->GetList()));
  EXPECT_TRUE(params);
  ASSERT_FALSE(params->suggest_results.empty());
  CompareClassification(styles_expected, StyleTypesToACMatchClassifications(
                                             params->suggest_results[0]));

  // Try moving the "dim/match" style pair at offset 9. Output should be the
  // same.
  std::unique_ptr<base::ListValue> moved_list =
      ListBuilder()
          .Append(42)
          .Append(ListBuilder()
                      .Append(DictionaryBuilder()
                                  .Set("content", "content")
                                  .Set("description", "description")
                                  .Set("descriptionStyles",
                                       ListBuilder()
                                           .Append(DictionaryBuilder()
                                                       .Set("type", "url")
                                                       .Set("offset", 0)
                                                       .Set("length", 5)
                                                       .Build())
                                           .Append(DictionaryBuilder()
                                                       .Set("type", "match")
                                                       .Set("offset", 0)
                                                       .Set("length", 4)
                                                       .Build())
                                           .Append(DictionaryBuilder()
                                                       .Set("type", "dim")
                                                       .Set("offset", 9)
                                                       .Set("length", 2)
                                                       .Build())
                                           .Append(DictionaryBuilder()
                                                       .Set("type", "match")
                                                       .Set("offset", 9)
                                                       .Set("length", 2)
                                                       .Build())
                                           .Append(DictionaryBuilder()
                                                       .Set("type", "dim")
                                                       .Set("offset", 1)
                                                       .Set("length", 2)
                                                       .Build())
                                           .Build())
                                  .Build())
                      .Build())
          .Build();

  std::unique_ptr<SendSuggestions::Params> moved_params(
      SendSuggestions::Params::Create(moved_list->GetList()));
  EXPECT_TRUE(moved_params);
  ASSERT_FALSE(moved_params->suggest_results.empty());
  CompareClassification(styles_expected, StyleTypesToACMatchClassifications(
                                             moved_params->suggest_results[0]));
}

//   0123456789
//   uuuuu
// + mmmmm
// + mmm
// +   ddd
// + ddd
// = 77777nnnnn
TEST(ExtensionOmniboxTest, DescriptionStylesCombine2) {
  std::unique_ptr<base::ListValue> list =
      ListBuilder()
          .Append(42)
          .Append(ListBuilder()
                      .Append(DictionaryBuilder()
                                  .Set("content", "content")
                                  .Set("description", "description")
                                  .Set("descriptionStyles",
                                       ListBuilder()
                                           .Append(DictionaryBuilder()
                                                       .Set("type", "url")
                                                       .Set("offset", 0)
                                                       .Set("length", 5)
                                                       .Build())
                                           .Append(DictionaryBuilder()
                                                       .Set("type", "match")
                                                       .Set("offset", 0)
                                                       .Set("length", 5)
                                                       .Build())
                                           .Append(DictionaryBuilder()
                                                       .Set("type", "match")
                                                       .Set("offset", 0)
                                                       .Set("length", 3)
                                                       .Build())
                                           .Append(DictionaryBuilder()
                                                       .Set("type", "dim")
                                                       .Set("offset", 2)
                                                       .Set("length", 3)
                                                       .Build())
                                           .Append(DictionaryBuilder()
                                                       .Set("type", "dim")
                                                       .Set("offset", 0)
                                                       .Set("length", 3)
                                                       .Build())
                                           .Build())
                                  .Build())
                      .Build())
          .Build();

  ACMatchClassifications styles_expected;
  styles_expected.push_back(ACMatchClassification(0, kUrl | kMatch | kDim));
  styles_expected.push_back(ACMatchClassification(5, kNone));

  std::unique_ptr<SendSuggestions::Params> params(
      SendSuggestions::Params::Create(list->GetList()));
  EXPECT_TRUE(params);
  ASSERT_FALSE(params->suggest_results.empty());
  CompareClassification(styles_expected, StyleTypesToACMatchClassifications(
                                             params->suggest_results[0]));
}

//   0123456789
//   uuuuu
// + mmmmm
// + mmm
// +   ddd
// + ddd
// = 77777nnnnn
TEST(ExtensionOmniboxTest, DefaultSuggestResult) {
  // Default suggestions should not have a content parameter.
  std::unique_ptr<base::ListValue> list =
      ListBuilder()
          .Append(DictionaryBuilder()
                      .Set("description", "description")
                      .Set("descriptionStyles",
                           ListBuilder()
                               .Append(DictionaryBuilder()
                                           .Set("type", "url")
                                           .Set("offset", 0)
                                           .Set("length", 5)
                                           .Build())
                               .Append(DictionaryBuilder()
                                           .Set("type", "match")
                                           .Set("offset", 0)
                                           .Set("length", 5)
                                           .Build())
                               .Append(DictionaryBuilder()
                                           .Set("type", "match")
                                           .Set("offset", 0)
                                           .Set("length", 3)
                                           .Build())
                               .Append(DictionaryBuilder()
                                           .Set("type", "dim")
                                           .Set("offset", 2)
                                           .Set("length", 3)
                                           .Build())
                               .Append(DictionaryBuilder()
                                           .Set("type", "dim")
                                           .Set("offset", 0)
                                           .Set("length", 3)
                                           .Build())
                               .Build())
                      .Build())
          .Build();

  std::unique_ptr<SetDefaultSuggestion::Params> params(
      SetDefaultSuggestion::Params::Create(list->GetList()));
  EXPECT_TRUE(params);
}

}  // namespace extensions