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

#include "components/lens/lens_url_utils.h"

#include "base/strings/string_number_conversions.h"
#include "base/strings/stringprintf.h"
#include "base/test/metrics/histogram_tester.h"
#include "base/time/time.h"
#include "components/lens/lens_entrypoints.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "url/gurl.h"

using ::testing::MatchesRegex;

namespace lens {

TEST(LensUrlUtilsTest, GetRegionSearchParameterTest) {
  lens::EntryPoint lens_region_search_ep =
      lens::EntryPoint::CHROME_REGION_SEARCH_MENU_ITEM;
  std::string query_param =
      lens::GetQueryParametersForLensRequest(lens_region_search_ep);
  EXPECT_THAT(query_param, MatchesRegex("ep=crs&re=df&s=4&st=\\d+"));
}

TEST(LensUrlUtilsTest, GetImageSearcParameterTest) {
  lens::EntryPoint lens_image_search_ep =
      lens::EntryPoint::CHROME_SEARCH_WITH_GOOGLE_LENS_CONTEXT_MENU_ITEM;
  std::string query_param =
      lens::GetQueryParametersForLensRequest(lens_image_search_ep);
  EXPECT_THAT(query_param, MatchesRegex("ep=ccm&re=df&s=4&st=\\d+"));
}

TEST(LensUrlUtilsTest, GetVideoFrameSearchQueryParameterTest) {
  auto lens_ep = lens::EntryPoint::CHROME_VIDEO_FRAME_SEARCH_CONTEXT_MENU_ITEM;
  std::string query_param = lens::GetQueryParametersForLensRequest(lens_ep);
  EXPECT_THAT(query_param, MatchesRegex("ep=cvfs&re=df&s=4&st=\\d+"));
}

TEST(LensUrlUtilsTest, GetUnknownEntryPointTest) {
  std::string query_param =
      lens::GetQueryParametersForLensRequest(lens::EntryPoint::UNKNOWN);
  EXPECT_THAT(query_param, MatchesRegex("re=df&s=4&st=\\d+"));
}

TEST(LensUrlUtilsTest, Base64EncodeRequestIdTest) {
  lens::LensOverlayRequestId request_id;
  request_id.set_uuid(1);
  request_id.set_sequence_id(2);
  request_id.set_analytics_id("ABC");
  std::string encoded_id = lens::Base64EncodeRequestId(request_id);
  EXPECT_FALSE(encoded_id.empty());
  EXPECT_EQ(encoded_id, "CAEQAiIDQUJD");
}

struct InvocationSourceParamTestCase {
  lens::LensOverlayInvocationSource invocation_source;
  std::string expected_param_suffix;
};

class LensUrlUtilsTestWithParams
    : public testing::TestWithParam<InvocationSourceParamTestCase> {};

TEST_P(LensUrlUtilsTestWithParams, AppendInvocationSourceParamToUrl) {
  const auto& test_case = GetParam();
  constexpr char kResultsSearchBaseUrl[] = "https://www.google.com/search";
  const GURL base_url(kResultsSearchBaseUrl);

  std::string expected_url =
      base::StringPrintf("%s?source=chrome.cr.%s", kResultsSearchBaseUrl,
                         test_case.expected_param_suffix.c_str());
  EXPECT_EQ(lens::AppendInvocationSourceParamToURL(
                base_url, test_case.invocation_source,
                /*is_contextual_tasks=*/false),
            expected_url);

  std::string expected_contextual_tasks_url =
      base::StringPrintf("%s?source=chrome.crn.%s", kResultsSearchBaseUrl,
                         test_case.expected_param_suffix.c_str());
  EXPECT_EQ(lens::AppendInvocationSourceParamToURL(
                base_url, test_case.invocation_source,
                /*is_contextual_tasks=*/true),
            expected_contextual_tasks_url);
}

INSTANTIATE_TEST_SUITE_P(
    InvocationSourceParams,
    LensUrlUtilsTestWithParams,
    testing::Values(
        InvocationSourceParamTestCase{
            lens::LensOverlayInvocationSource::kAppMenu, "menu"},
        InvocationSourceParamTestCase{
            lens::LensOverlayInvocationSource::kContentAreaContextMenuPage,
            "ctxp"},
        InvocationSourceParamTestCase{
            lens::LensOverlayInvocationSource::kContentAreaContextMenuImage,
            "ctxi"},
        InvocationSourceParamTestCase{
            lens::LensOverlayInvocationSource::kToolbar, "tbic"},
        InvocationSourceParamTestCase{
            lens::LensOverlayInvocationSource::kFindInPage, "find"},
        InvocationSourceParamTestCase{
            lens::LensOverlayInvocationSource::kOmnibox, "obic"},
        InvocationSourceParamTestCase{
            lens::LensOverlayInvocationSource::kContentAreaContextMenuVideo,
            "ctxv"},
        InvocationSourceParamTestCase{
            lens::LensOverlayInvocationSource::kNtpContextualQuery, "rb"},
        InvocationSourceParamTestCase{
            lens::LensOverlayInvocationSource::kOmniboxContextualQuery, "obic"},
        InvocationSourceParamTestCase{
            lens::LensOverlayInvocationSource::kCobrowseToolbarButton, "cct"},
        InvocationSourceParamTestCase{
            lens::LensOverlayInvocationSource::kCobrowsePinnedToolbarButton,
            "ccpt"},
        InvocationSourceParamTestCase{
            lens::LensOverlayInvocationSource::kNtpActionChips, "ntpac"},
        InvocationSourceParamTestCase{
            lens::LensOverlayInvocationSource::kAppBarAimButton, "abab"},
        InvocationSourceParamTestCase{
            lens::LensOverlayInvocationSource::kOmniboxEverywhereComposebox,
            "oecb"},
        InvocationSourceParamTestCase{
            lens::LensOverlayInvocationSource::kOmniboxPopupButton, "obpb"}));

}  // namespace lens