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

#include "base/strings/string_util.h"
#include "base/test/scoped_feature_list.h"
#include "build/buildflag.h"
#include "components/omnibox/browser/autocomplete_scheme_classifier.h"
#include "components/omnibox/browser/omnibox_field_trial.h"
#include "components/omnibox/browser/suggestion_group_util.h"
#include "components/omnibox/common/omnibox_features.h"
#include "components/strings/grit/components_strings.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/omnibox_proto/groups.pb.h"
#include "ui/base/l10n/l10n_util.h"

class TestingSchemeClassifier : public AutocompleteSchemeClassifier {
 public:
  TestingSchemeClassifier() = default;
  TestingSchemeClassifier(const TestingSchemeClassifier&) = delete;
  TestingSchemeClassifier& operator=(const TestingSchemeClassifier&) = delete;

  metrics::OmniboxInputType GetInputTypeForScheme(
      const std::string& scheme) const override {
    DCHECK_EQ(scheme, base::ToLowerASCII(scheme));
    return (scheme == url::kHttpScheme || scheme == url::kHttpsScheme)
               ? metrics::OmniboxInputType::URL
               : metrics::OmniboxInputType::EMPTY;
  }
};

// Ensures that accessing unset fields is safe and verifies the default values.
// https://developers.google.com/protocol-buffers/docs/reference/cpp-generated
TEST(SuggestionGroupTest, DefaultValuesForUnsetFields) {
  omnibox::GroupConfig group_config;

  ASSERT_FALSE(group_config.has_header_text());
  ASSERT_EQ("", group_config.header_text());

  ASSERT_FALSE(group_config.has_side_type());
  ASSERT_EQ(omnibox::GroupConfig_SideType_DEFAULT_PRIMARY,
            group_config.side_type());

  ASSERT_FALSE(group_config.has_render_type());
  ASSERT_EQ(omnibox::GroupConfig_RenderType_DEFAULT_VERTICAL,
            group_config.render_type());

  ASSERT_FALSE(group_config.has_visibility());
  ASSERT_EQ(omnibox::GroupConfig_Visibility_DEFAULT_VISIBLE,
            group_config.visibility());

  ASSERT_FALSE(group_config.has_why_this_result_reason());
  ASSERT_EQ(0U, group_config.why_this_result_reason());

  ASSERT_FALSE(group_config.has_section());
  ASSERT_EQ(omnibox::SECTION_DEFAULT, group_config.section());
}

// Ensures that omnibox::GroupIdForNumber() returns the omnibox::GroupId enum
// object corresponding to the given value; and returns omnibox::GROUP_INVALID
// when there is no corresponding enum object.
TEST(SuggestionGroupTest, GroupIdForNumber) {
  ASSERT_EQ(omnibox::GROUP_PERSONALIZED_ZERO_SUGGEST,
            omnibox::GroupIdForNumber(40000));
  ASSERT_EQ(omnibox::GROUP_INVALID, omnibox::GroupIdForNumber(-1));
  ASSERT_EQ(omnibox::GROUP_INVALID, omnibox::GroupIdForNumber(0));
  ASSERT_EQ(omnibox::GROUP_INVALID, omnibox::GroupIdForNumber(123456789));
}

TEST(SuggestionGroupTest, SectionMobileMostVisited_HorizontalRenderType) {
  omnibox::ResetDefaultGroupsForTest();

  base::test::ScopedFeatureList features;
  features.InitWithFeatures({omnibox::kMostVisitedTilesHorizontalRenderGroup},
                            {});

  AutocompleteInput input;
  auto default_groups =
      omnibox::BuildDefaultGroupsForInput(input, /*is_incognito=*/false);
  auto most_visited_group_config =
      default_groups.find(omnibox::GROUP_MOBILE_MOST_VISITED);

  ASSERT_NE(most_visited_group_config, default_groups.end());
  ASSERT_EQ(omnibox::GroupConfig_RenderType_HORIZONTAL,
            most_visited_group_config->second.render_type());
}

TEST(SuggestionGroupTest, SectionMobileMostVisited_VerticalRenderType) {
  omnibox::ResetDefaultGroupsForTest();

  base::test::ScopedFeatureList features;
  features.InitWithFeatures({},
                            {omnibox::kMostVisitedTilesHorizontalRenderGroup});

  AutocompleteInput input;
  auto default_groups =
      omnibox::BuildDefaultGroupsForInput(input, /*is_incognito=*/false);
  auto most_visited_group_config =
      default_groups.find(omnibox::GROUP_MOBILE_MOST_VISITED);

  ASSERT_NE(most_visited_group_config, default_groups.end());
  ASSERT_EQ(omnibox::GroupConfig_RenderType_DEFAULT_VERTICAL,
            most_visited_group_config->second.render_type());
}

TEST(SuggestionGroupTest, AndroidHubZPS) {
  omnibox::ResetDefaultGroupsForTest();

  base::test::ScopedFeatureList features;
  features.InitWithFeatures({},
                            {omnibox::kMostVisitedTilesHorizontalRenderGroup});

  using OEP = ::metrics::OmniboxEventProto;
  AutocompleteInput input(u"", OEP::ANDROID_HUB, TestingSchemeClassifier());
  auto default_groups =
      omnibox::BuildDefaultGroupsForInput(input, /*is_incognito=*/false);
  auto open_tabs_group_config =
      default_groups.find(omnibox::GROUP_MOBILE_OPEN_TABS);

  ASSERT_NE(open_tabs_group_config, default_groups.end());
  ASSERT_EQ(omnibox::GroupConfig_RenderType_DEFAULT_VERTICAL,
            open_tabs_group_config->second.render_type());
  ASSERT_EQ("Last open tabs", open_tabs_group_config->second.header_text());
}

TEST(SuggestionGroupTest, AndroidHubTyped) {
  omnibox::ResetDefaultGroupsForTest();

  base::test::ScopedFeatureList features;
  features.InitWithFeatures({},
                            {omnibox::kMostVisitedTilesHorizontalRenderGroup});

  using OEP = ::metrics::OmniboxEventProto;
  AutocompleteInput input(u"test", OEP::ANDROID_HUB, TestingSchemeClassifier());
  auto default_groups =
      omnibox::BuildDefaultGroupsForInput(input, /*is_incognito=*/false);

  auto open_tabs_group_config =
      default_groups.find(omnibox::GROUP_MOBILE_OPEN_TABS);
  ASSERT_NE(open_tabs_group_config, default_groups.end());
  ASSERT_EQ(omnibox::GroupConfig_RenderType_DEFAULT_VERTICAL,
            open_tabs_group_config->second.render_type());
#if BUILDFLAG(IS_ANDROID)
  ASSERT_EQ("Tabs and tab groups",
            open_tabs_group_config->second.header_text());
#else   // !BUILDFLAG(IS_ANDROID)
  ASSERT_EQ("", open_tabs_group_config->second.header_text());
#endif  // BUILDFLAG(IS_ANDROID)

  auto search_group_config = default_groups.find(omnibox::GROUP_SEARCH);
  ASSERT_NE(search_group_config, default_groups.end());
  ASSERT_EQ(omnibox::GroupConfig_RenderType_DEFAULT_VERTICAL,
            search_group_config->second.render_type());
  ASSERT_EQ("Search the web", search_group_config->second.header_text());
}

#if BUILDFLAG(IS_ANDROID)
TEST(SuggestionGroupTest, AndroidHubTypedTabGroups) {
  omnibox::ResetDefaultGroupsForTest();

  base::test::ScopedFeatureList features;
  features.InitWithFeatures({omnibox::kMostVisitedTilesHorizontalRenderGroup},
                            {});

  using OEP = ::metrics::OmniboxEventProto;
  AutocompleteInput input(u"test", OEP::ANDROID_HUB, TestingSchemeClassifier());
  auto default_groups =
      omnibox::BuildDefaultGroupsForInput(input, /*is_incognito=*/false);

  auto open_tabs_group_config =
      default_groups.find(omnibox::GROUP_MOBILE_OPEN_TABS);
  ASSERT_NE(open_tabs_group_config, default_groups.end());
  ASSERT_EQ(omnibox::GroupConfig_RenderType_DEFAULT_VERTICAL,
            open_tabs_group_config->second.render_type());
  ASSERT_EQ("Tabs and tab groups",
            open_tabs_group_config->second.header_text());

  auto search_group_config = default_groups.find(omnibox::GROUP_SEARCH);
  ASSERT_NE(search_group_config, default_groups.end());
  ASSERT_EQ(omnibox::GroupConfig_RenderType_DEFAULT_VERTICAL,
            search_group_config->second.render_type());
  ASSERT_EQ("Search the web", search_group_config->second.header_text());
}

TEST(SuggestionGroupTest, AndroidHubTyped_IncognitoAndRegularInterleaved) {
  omnibox::ResetDefaultGroupsForTest();

  using OEP = ::metrics::OmniboxEventProto;
  AutocompleteInput input(u"test", OEP::ANDROID_HUB, TestingSchemeClassifier());

  // First call in incognito mode.
  auto incognito_groups =
      omnibox::BuildDefaultGroupsForInput(input, /*is_incognito=*/true);
  auto incognito_open_tabs =
      incognito_groups.find(omnibox::GROUP_MOBILE_OPEN_TABS);
  ASSERT_NE(incognito_open_tabs, incognito_groups.end());
  ASSERT_EQ("", incognito_open_tabs->second.header_text());

  // Subsequent call in non-incognito mode should return the regular cached map
  // with header text and not be polluted by the prior incognito invocation.
  auto regular_groups =
      omnibox::BuildDefaultGroupsForInput(input, /*is_incognito=*/false);
  auto regular_open_tabs = regular_groups.find(omnibox::GROUP_MOBILE_OPEN_TABS);
  ASSERT_NE(regular_open_tabs, regular_groups.end());
  ASSERT_EQ("Tabs and tab groups", regular_open_tabs->second.header_text());

  // Calling incognito mode again still returns the incognito cached map.
  auto incognito_groups_2 =
      omnibox::BuildDefaultGroupsForInput(input, /*is_incognito=*/true);
  auto incognito_open_tabs_2 =
      incognito_groups_2.find(omnibox::GROUP_MOBILE_OPEN_TABS);
  ASSERT_NE(incognito_open_tabs_2, incognito_groups_2.end());
  ASSERT_EQ("", incognito_open_tabs_2->second.header_text());
}

TEST(SuggestionGroupTest, AndroidTabSearchOverlay) {
  omnibox::ResetDefaultGroupsForTest();

  using OEP = ::metrics::OmniboxEventProto;
  AutocompleteInput input(u"test", OEP::ANDROID_TAB_SEARCH_OVERLAY,
                          TestingSchemeClassifier());
  auto default_groups =
      omnibox::BuildDefaultGroupsForInput(input, /*is_incognito=*/false);

  auto open_tabs_group_config =
      default_groups.find(omnibox::GROUP_MOBILE_OPEN_TABS);
  ASSERT_NE(open_tabs_group_config, default_groups.end());
  ASSERT_EQ(omnibox::GroupConfig_RenderType_DEFAULT_VERTICAL,
            open_tabs_group_config->second.render_type());
  ASSERT_EQ("Tabs and tab groups",
            open_tabs_group_config->second.header_text());

  // Bookmark and Search groups should NOT be present.
  auto bookmark_group_config =
      default_groups.find(omnibox::GROUP_MOBILE_BOOKMARKS);
  ASSERT_EQ(bookmark_group_config, default_groups.end());

  auto search_group_config = default_groups.find(omnibox::GROUP_SEARCH);
  ASSERT_EQ(search_group_config, default_groups.end());

  omnibox::ResetDefaultGroupsForTest();
}

TEST(SuggestionGroupTest, AndroidTabSearchOverlayZPS) {
  omnibox::ResetDefaultGroupsForTest();

  using OEP = ::metrics::OmniboxEventProto;
  AutocompleteInput input(u"", OEP::ANDROID_TAB_SEARCH_OVERLAY,
                          TestingSchemeClassifier());
  auto default_groups =
      omnibox::BuildDefaultGroupsForInput(input, /*is_incognito=*/false);
  auto open_tabs_group_config =
      default_groups.find(omnibox::GROUP_MOBILE_OPEN_TABS);

  ASSERT_NE(open_tabs_group_config, default_groups.end());
  ASSERT_EQ(omnibox::GroupConfig_RenderType_DEFAULT_VERTICAL,
            open_tabs_group_config->second.render_type());
  ASSERT_EQ("Last open tabs", open_tabs_group_config->second.header_text());

  // Bookmark, history and Search groups should NOT be present.
  auto bookmark_group_config =
      default_groups.find(omnibox::GROUP_MOBILE_BOOKMARKS);
  ASSERT_EQ(bookmark_group_config, default_groups.end());

  auto history_group_config =
      default_groups.find(omnibox::GROUP_MOBILE_HISTORY);
  ASSERT_EQ(history_group_config, default_groups.end());

  auto search_group_config = default_groups.find(omnibox::GROUP_SEARCH);
  ASSERT_EQ(search_group_config, default_groups.end());

  omnibox::ResetDefaultGroupsForTest();
}
#endif  // BUILDFLAG(IS_ANDROID)