lens_entrypoints_unittest.cc 6.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154
  1. // Copyright 2021 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "components/lens/lens_entrypoints.h"
  5. #include "base/strings/string_number_conversions.h"
  6. #include "base/time/time.h"
  7. #include "testing/gmock/include/gmock/gmock.h"
  8. #include "testing/gtest/include/gtest/gtest.h"
  9. #include "url/gurl.h"
  10. using ::testing::MatchesRegex;
  11. namespace lens {
  12. TEST(LensEntryPointsTest, GetRegionSearchNewTabQueryParameterTest) {
  13. lens::EntryPoint lens_region_search_ep =
  14. lens::EntryPoint::CHROME_REGION_SEARCH_MENU_ITEM;
  15. std::string query_param = lens::GetQueryParametersForLensRequest(
  16. lens_region_search_ep, /*is_side_panel_request=*/false);
  17. EXPECT_THAT(query_param, MatchesRegex("ep=crs&s=&st=\\d+"));
  18. }
  19. TEST(LensEntryPointsTest, GetImageSearchNewTabQueryParameterTest) {
  20. lens::EntryPoint lens_image_search_ep =
  21. lens::EntryPoint::CHROME_SEARCH_WITH_GOOGLE_LENS_CONTEXT_MENU_ITEM;
  22. std::string query_param = lens::GetQueryParametersForLensRequest(
  23. lens_image_search_ep, /*is_side_panel_request=*/false);
  24. EXPECT_THAT(query_param, MatchesRegex("ep=ccm&s=&st=\\d+"));
  25. }
  26. TEST(LensEntryPointsTest, GetRegionSearchSidePanelQueryParameterTest) {
  27. lens::EntryPoint lens_region_search_ep =
  28. lens::EntryPoint::CHROME_REGION_SEARCH_MENU_ITEM;
  29. std::string query_param = lens::GetQueryParametersForLensRequest(
  30. lens_region_search_ep, /*is_side_panel_request=*/true);
  31. EXPECT_THAT(query_param, MatchesRegex("ep=crs&s=csp&st=\\d+"));
  32. }
  33. TEST(LensEntryPointsTest, GetImageSearchSidePanelQueryParameterTest) {
  34. lens::EntryPoint lens_image_search_ep =
  35. lens::EntryPoint::CHROME_SEARCH_WITH_GOOGLE_LENS_CONTEXT_MENU_ITEM;
  36. std::string query_param = lens::GetQueryParametersForLensRequest(
  37. lens_image_search_ep, /*is_side_panel_request=*/true);
  38. EXPECT_THAT(query_param, MatchesRegex("ep=ccm&s=csp&st=\\d+"));
  39. }
  40. TEST(LensEntryPointsTest, GetOpenNewTabSidePanelParameterTest) {
  41. lens::EntryPoint lens_open_new_tab_side_panel_ep =
  42. lens::EntryPoint::CHROME_OPEN_NEW_TAB_SIDE_PANEL;
  43. std::string query_param = lens::GetQueryParametersForLensRequest(
  44. lens_open_new_tab_side_panel_ep, /*is_side_panel_request=*/false);
  45. EXPECT_THAT(query_param, MatchesRegex("ep=cnts&s=&st=\\d+"));
  46. }
  47. TEST(LensEntryPointsTest, GetFullscreenSearchQueryParameterTest) {
  48. lens::EntryPoint lens_ep =
  49. lens::EntryPoint::CHROME_FULLSCREEN_SEARCH_MENU_ITEM;
  50. std::string query_param = lens::GetQueryParametersForLensRequest(
  51. lens_ep, /*is_side_panel_request=*/false);
  52. EXPECT_THAT(query_param, MatchesRegex("ep=cfs&s=&st=\\d+"));
  53. }
  54. TEST(LensEntryPointsTest, GetScreenshotSearchQueryParameterTest) {
  55. lens::EntryPoint lens_ep = lens::EntryPoint::CHROME_SCREENSHOT_SEARCH;
  56. std::string query_param = lens::GetQueryParametersForLensRequest(
  57. lens_ep, /*is_side_panel_request=*/false);
  58. EXPECT_THAT(query_param, MatchesRegex("ep=css&s=&st=\\d+"));
  59. }
  60. TEST(LensEntryPointsTest, GetUnknownEntryPointTest) {
  61. std::string query_param = lens::GetQueryParametersForLensRequest(
  62. lens::EntryPoint::UNKNOWN, /*is_side_panel_request=*/false);
  63. EXPECT_THAT(query_param, MatchesRegex("s=&st=\\d+"));
  64. }
  65. TEST(LensEntryPointsTest, GetUnknownEntryPointSidePanelTest) {
  66. std::string query_param = lens::GetQueryParametersForLensRequest(
  67. lens::EntryPoint::UNKNOWN, /*is_side_panel_request=*/true);
  68. EXPECT_THAT(query_param, MatchesRegex("s=csp&st=\\d+"));
  69. }
  70. TEST(LensEntryPointsTest, AppendRegionSearchNewTabQueryParameterTest) {
  71. lens::EntryPoint lens_region_search_ep =
  72. lens::EntryPoint::CHROME_REGION_SEARCH_MENU_ITEM;
  73. GURL original_url = GURL("https://lens.google.com/");
  74. GURL url = lens::AppendOrReplaceQueryParametersForLensRequest(
  75. original_url, lens_region_search_ep, /*is_side_panel_request=*/false);
  76. EXPECT_THAT(url.query(), MatchesRegex("ep=crs&s=&st=\\d+"));
  77. }
  78. TEST(LensEntryPointsTest, AppendImageSearchNewTabQueryParameterTest) {
  79. lens::EntryPoint lens_image_search_ep =
  80. lens::EntryPoint::CHROME_SEARCH_WITH_GOOGLE_LENS_CONTEXT_MENU_ITEM;
  81. GURL original_url = GURL("https://lens.google.com/");
  82. GURL url = lens::AppendOrReplaceQueryParametersForLensRequest(
  83. original_url, lens_image_search_ep, /*is_side_panel_request=*/false);
  84. EXPECT_THAT(url.query(), MatchesRegex("ep=ccm&s=&st=\\d+"));
  85. }
  86. TEST(LensEntryPointsTest, AppendRegionSearchSidePanelQueryParameterTest) {
  87. lens::EntryPoint lens_region_search_ep =
  88. lens::EntryPoint::CHROME_REGION_SEARCH_MENU_ITEM;
  89. GURL original_url = GURL("https://lens.google.com/");
  90. GURL url = lens::AppendOrReplaceQueryParametersForLensRequest(
  91. original_url, lens_region_search_ep, /*is_side_panel_request=*/true);
  92. EXPECT_THAT(url.query(), MatchesRegex("ep=crs&s=csp&st=\\d+"));
  93. }
  94. TEST(LensEntryPointsTest, AppendImageSearchSidePanelQueryParameterTest) {
  95. lens::EntryPoint lens_image_search_ep =
  96. lens::EntryPoint::CHROME_SEARCH_WITH_GOOGLE_LENS_CONTEXT_MENU_ITEM;
  97. GURL original_url = GURL("https://lens.google.com/");
  98. GURL url = lens::AppendOrReplaceQueryParametersForLensRequest(
  99. original_url, lens_image_search_ep, /*is_side_panel_request=*/true);
  100. EXPECT_THAT(url.query(), MatchesRegex("ep=ccm&s=csp&st=\\d+"));
  101. }
  102. TEST(LensEntryPointsTest, AppendOpenNewTabSidePanelParameterTest) {
  103. lens::EntryPoint lens_open_new_tab_side_panel_ep =
  104. lens::EntryPoint::CHROME_OPEN_NEW_TAB_SIDE_PANEL;
  105. GURL original_url = GURL("https://lens.google.com/");
  106. GURL url = lens::AppendOrReplaceQueryParametersForLensRequest(
  107. original_url, lens_open_new_tab_side_panel_ep,
  108. /*is_side_panel_request=*/false);
  109. EXPECT_THAT(url.query(), MatchesRegex("ep=cnts&s=&st=\\d+"));
  110. }
  111. TEST(LensEntryPointsTest, AppendFullscreenSearchQueryParameterTest) {
  112. lens::EntryPoint lens_ep =
  113. lens::EntryPoint::CHROME_FULLSCREEN_SEARCH_MENU_ITEM;
  114. GURL original_url = GURL("https://lens.google.com/");
  115. GURL url = lens::AppendOrReplaceQueryParametersForLensRequest(
  116. original_url, lens_ep, /*is_side_panel_request=*/false);
  117. EXPECT_THAT(url.query(), MatchesRegex("ep=cfs&s=&st=\\d+"));
  118. }
  119. TEST(LensEntryPointsTest, AppendScreenshotSearchQueryParameterTest) {
  120. lens::EntryPoint lens_ep = lens::EntryPoint::CHROME_SCREENSHOT_SEARCH;
  121. GURL original_url = GURL("https://lens.google.com/");
  122. GURL url = lens::AppendOrReplaceQueryParametersForLensRequest(
  123. original_url, lens_ep, /*is_side_panel_request=*/false);
  124. EXPECT_THAT(url.query(), MatchesRegex("ep=css&s=&st=\\d+"));
  125. }
  126. TEST(LensEntryPointsTest, AppendUnknownEntryPointTest) {
  127. GURL original_url = GURL("https://lens.google.com/");
  128. GURL url = lens::AppendOrReplaceQueryParametersForLensRequest(
  129. original_url, lens::EntryPoint::UNKNOWN, /*is_side_panel_request=*/false);
  130. EXPECT_THAT(url.query(), MatchesRegex("s=&st=\\d+"));
  131. }
  132. } // namespace lens