content_visibility_cluster_finalizer_unittest.cc 3.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798
  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/history_clusters/core/content_visibility_cluster_finalizer.h"
  5. #include "base/test/metrics/histogram_tester.h"
  6. #include "base/test/task_environment.h"
  7. #include "components/history_clusters/core/clustering_test_utils.h"
  8. #include "testing/gmock/include/gmock/gmock.h"
  9. #include "testing/gtest/include/gtest/gtest.h"
  10. namespace history_clusters {
  11. namespace {
  12. class ContentVisibilityClusterFinalizerTest : public ::testing::Test {
  13. public:
  14. void SetUp() override {
  15. cluster_finalizer_ = std::make_unique<ContentVisibilityClusterFinalizer>();
  16. }
  17. void TearDown() override { cluster_finalizer_.reset(); }
  18. void FinalizeCluster(history::Cluster& cluster) {
  19. cluster_finalizer_->FinalizeCluster(cluster);
  20. }
  21. private:
  22. std::unique_ptr<ContentVisibilityClusterFinalizer> cluster_finalizer_;
  23. base::test::TaskEnvironment task_environment_;
  24. };
  25. TEST_F(ContentVisibilityClusterFinalizerTest, OneVisitNoVisibilityScore) {
  26. // Visit2 has the same URL as Visit1.
  27. history::ClusterVisit visit = testing::CreateClusterVisit(
  28. testing::CreateDefaultAnnotatedVisit(1, GURL("https://google.com/")));
  29. history::Cluster cluster;
  30. cluster.visits = {visit};
  31. FinalizeCluster(cluster);
  32. EXPECT_TRUE(cluster.should_show_on_prominent_ui_surfaces);
  33. }
  34. TEST_F(ContentVisibilityClusterFinalizerTest,
  35. OneVisitNoVisibilityScoreButExplicitlyNotShownBefore) {
  36. // Visit2 has the same URL as Visit1.
  37. history::ClusterVisit visit = testing::CreateClusterVisit(
  38. testing::CreateDefaultAnnotatedVisit(1, GURL("https://google.com/")));
  39. history::Cluster cluster;
  40. cluster.visits = {visit};
  41. cluster.should_show_on_prominent_ui_surfaces = false;
  42. FinalizeCluster(cluster);
  43. EXPECT_FALSE(cluster.should_show_on_prominent_ui_surfaces);
  44. }
  45. TEST_F(ContentVisibilityClusterFinalizerTest, MultipleVisitsOneBelowThreshold) {
  46. base::HistogramTester histogram_tester;
  47. history::ClusterVisit visit = testing::CreateClusterVisit(
  48. testing::CreateDefaultAnnotatedVisit(1, GURL("https://google.com/")));
  49. visit.annotated_visit.content_annotations.model_annotations.visibility_score =
  50. 1.0;
  51. history::ClusterVisit visit2 = testing::CreateClusterVisit(
  52. testing::CreateDefaultAnnotatedVisit(2, GURL("https://foo.com/")));
  53. visit2.annotated_visit.content_annotations.model_annotations
  54. .visibility_score = 0.1;
  55. history::Cluster cluster;
  56. cluster.visits = {visit, visit2};
  57. FinalizeCluster(cluster);
  58. EXPECT_FALSE(cluster.should_show_on_prominent_ui_surfaces);
  59. histogram_tester.ExpectUniqueSample(
  60. "History.Clusters.Backend.WasClusterFiltered.VisibilityScore", true, 1);
  61. }
  62. TEST_F(ContentVisibilityClusterFinalizerTest, AllVisitsAboveThreshold) {
  63. base::HistogramTester histogram_tester;
  64. history::ClusterVisit visit = testing::CreateClusterVisit(
  65. testing::CreateDefaultAnnotatedVisit(1, GURL("https://google.com/")));
  66. visit.annotated_visit.content_annotations.model_annotations.visibility_score =
  67. 1.0;
  68. history::ClusterVisit visit2 = testing::CreateClusterVisit(
  69. testing::CreateDefaultAnnotatedVisit(2, GURL("https://foo.com/")));
  70. visit2.annotated_visit.content_annotations.model_annotations
  71. .visibility_score = 1.0;
  72. history::Cluster cluster;
  73. cluster.visits = {visit, visit2};
  74. FinalizeCluster(cluster);
  75. EXPECT_TRUE(cluster.should_show_on_prominent_ui_surfaces);
  76. histogram_tester.ExpectUniqueSample(
  77. "History.Clusters.Backend.WasClusterFiltered.VisibilityScore", false, 1);
  78. }
  79. } // namespace
  80. } // namespace history_clusters