Plumb site engagement to the renderer process.

chrome/browser/engagement/site_engagement_service_unittest.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/engagement/site_engagement_service.h"
 
 #include <utility>
 
 #include "base/files/scoped_temp_dir.h"
 #include "base/macros.h"
@@ skipping 1178 matching lines @@
     // origin1 should be removed. origin3 and origin4 remain untouched.
     EXPECT_EQ(0, engagement->GetScore(origin1));
     EXPECT_EQ(0, engagement->GetScore(origin2));
     AssertInRange(5.0, engagement->GetScore(origin3));
     AssertInRange(2.5, engagement->GetScore(origin4));
     AssertInRange(7.5, engagement->GetTotalEngagementPoints());
   }
 }
 
 TEST_F(SiteEngagementServiceTest, EngagementLevel) {
-  static_assert(SiteEngagementService::ENGAGEMENT_LEVEL_NONE !=
-                    SiteEngagementService::ENGAGEMENT_LEVEL_LOW,
+  static_assert(blink::mojom::EngagementLevel::NONE !=
+                    blink::mojom::EngagementLevel::LOW,
                 "enum values should not be equal");
-  static_assert(SiteEngagementService::ENGAGEMENT_LEVEL_LOW !=
-                    SiteEngagementService::ENGAGEMENT_LEVEL_MEDIUM,
+  static_assert(blink::mojom::EngagementLevel::LOW !=
+                    blink::mojom::EngagementLevel::MEDIUM,
                 "enum values should not be equal");
-  static_assert(SiteEngagementService::ENGAGEMENT_LEVEL_MEDIUM !=
-                    SiteEngagementService::ENGAGEMENT_LEVEL_HIGH,
+  static_assert(blink::mojom::EngagementLevel::MEDIUM !=
+                    blink::mojom::EngagementLevel::HIGH,
                 "enum values should not be equal");
-  static_assert(SiteEngagementService::ENGAGEMENT_LEVEL_HIGH !=
-                    SiteEngagementService::ENGAGEMENT_LEVEL_MAX,
+  static_assert(blink::mojom::EngagementLevel::HIGH !=
+                    blink::mojom::EngagementLevel::MAX,
                 "enum values should not be equal");
 
   base::SimpleTestClock* clock = new base::SimpleTestClock();
   std::unique_ptr<SiteEngagementService> service(
       new SiteEngagementService(profile(), base::WrapUnique(clock)));
 
   base::Time current_day = GetReferenceTime();
   clock->SetNow(current_day);
 
   GURL url1("https://www.google.com/");
   GURL url2("http://www.google.com/");
 
-  EXPECT_EQ(SiteEngagementService::ENGAGEMENT_LEVEL_NONE,
+  EXPECT_EQ(blink::mojom::EngagementLevel::NONE,
             service->GetEngagementLevel(url1));
-  EXPECT_EQ(SiteEngagementService::ENGAGEMENT_LEVEL_NONE,
+  EXPECT_EQ(blink::mojom::EngagementLevel::NONE,
             service->GetEngagementLevel(url2));
+  EXPECT_TRUE(
+      service->IsEngagementAtLeast(url1, blink::mojom::EngagementLevel::NONE));
+  EXPECT_FALSE(service->IsEngagementAtLeast(
+      url1, blink::mojom::EngagementLevel::MINIMAL));
+  EXPECT_FALSE(
+      service->IsEngagementAtLeast(url1, blink::mojom::EngagementLevel::LOW));
+  EXPECT_FALSE(service->IsEngagementAtLeast(
+      url1, blink::mojom::EngagementLevel::MEDIUM));
+  EXPECT_FALSE(
+      service->IsEngagementAtLeast(url1, blink::mojom::EngagementLevel::HIGH));
+  EXPECT_FALSE(
+      service->IsEngagementAtLeast(url1, blink::mojom::EngagementLevel::MAX));
+
+  // Bring url2 to MINIMAL engagement.
+  service->AddPoints(url2, 0.5);
+  EXPECT_EQ(blink::mojom::EngagementLevel::NONE,
+            service->GetEngagementLevel(url1));
+  EXPECT_EQ(blink::mojom::EngagementLevel::MINIMAL,
+            service->GetEngagementLevel(url2));
+  EXPECT_TRUE(
+      service->IsEngagementAtLeast(url2, blink::mojom::EngagementLevel::NONE));
   EXPECT_TRUE(service->IsEngagementAtLeast(
-      url1, SiteEngagementService::ENGAGEMENT_LEVEL_NONE));
+      url2, blink::mojom::EngagementLevel::MINIMAL));
+  EXPECT_FALSE(
+      service->IsEngagementAtLeast(url2, blink::mojom::EngagementLevel::LOW));
   EXPECT_FALSE(service->IsEngagementAtLeast(
-      url1, SiteEngagementService::ENGAGEMENT_LEVEL_LOW));
-  EXPECT_FALSE(service->IsEngagementAtLeast(
-      url1, SiteEngagementService::ENGAGEMENT_LEVEL_MEDIUM));
-  EXPECT_FALSE(service->IsEngagementAtLeast(
-      url1, SiteEngagementService::ENGAGEMENT_LEVEL_HIGH));
-  EXPECT_FALSE(service->IsEngagementAtLeast(
-      url1, SiteEngagementService::ENGAGEMENT_LEVEL_MAX));
+      url2, blink::mojom::EngagementLevel::MEDIUM));
+  EXPECT_FALSE(
+      service->IsEngagementAtLeast(url2, blink::mojom::EngagementLevel::HIGH));
+  EXPECT_FALSE(
+      service->IsEngagementAtLeast(url2, blink::mojom::EngagementLevel::MAX));
 
   // Bring url1 to LOW engagement.
   service->AddPoints(url1, 1.0);
-  EXPECT_EQ(SiteEngagementService::ENGAGEMENT_LEVEL_LOW,
+  EXPECT_EQ(blink::mojom::EngagementLevel::LOW,
             service->GetEngagementLevel(url1));
-  EXPECT_EQ(SiteEngagementService::ENGAGEMENT_LEVEL_NONE,
+  EXPECT_EQ(blink::mojom::EngagementLevel::MINIMAL,
             service->GetEngagementLevel(url2));
+  EXPECT_TRUE(
+      service->IsEngagementAtLeast(url1, blink::mojom::EngagementLevel::NONE));
   EXPECT_TRUE(service->IsEngagementAtLeast(
-      url1, SiteEngagementService::ENGAGEMENT_LEVEL_NONE));
-  EXPECT_TRUE(service->IsEngagementAtLeast(
-      url1, SiteEngagementService::ENGAGEMENT_LEVEL_LOW));
+      url1, blink::mojom::EngagementLevel::MINIMAL));
+  EXPECT_TRUE(
+      service->IsEngagementAtLeast(url1, blink::mojom::EngagementLevel::LOW));
   EXPECT_FALSE(service->IsEngagementAtLeast(
-      url1, SiteEngagementService::ENGAGEMENT_LEVEL_MEDIUM));
-  EXPECT_FALSE(service->IsEngagementAtLeast(
-      url1, SiteEngagementService::ENGAGEMENT_LEVEL_HIGH));
-  EXPECT_FALSE(service->IsEngagementAtLeast(
-      url1, SiteEngagementService::ENGAGEMENT_LEVEL_MAX));
+      url1, blink::mojom::EngagementLevel::MEDIUM));
+  EXPECT_FALSE(
+      service->IsEngagementAtLeast(url1, blink::mojom::EngagementLevel::HIGH));
+  EXPECT_FALSE(
+      service->IsEngagementAtLeast(url1, blink::mojom::EngagementLevel::MAX));
 
   // Bring url2 to MEDIUM engagement.
-  service->AddPoints(url2, 5.0);
-  EXPECT_EQ(SiteEngagementService::ENGAGEMENT_LEVEL_LOW,
+  service->AddPoints(url2, 4.5);
+  EXPECT_EQ(blink::mojom::EngagementLevel::LOW,
             service->GetEngagementLevel(url1));
-  EXPECT_EQ(SiteEngagementService::ENGAGEMENT_LEVEL_MEDIUM,
+  EXPECT_EQ(blink::mojom::EngagementLevel::MEDIUM,
             service->GetEngagementLevel(url2));
+  EXPECT_TRUE(
+      service->IsEngagementAtLeast(url2, blink::mojom::EngagementLevel::NONE));
   EXPECT_TRUE(service->IsEngagementAtLeast(
-      url2, SiteEngagementService::ENGAGEMENT_LEVEL_NONE));
+      url2, blink::mojom::EngagementLevel::MINIMAL));
+  EXPECT_TRUE(
+      service->IsEngagementAtLeast(url2, blink::mojom::EngagementLevel::LOW));
   EXPECT_TRUE(service->IsEngagementAtLeast(
-      url2, SiteEngagementService::ENGAGEMENT_LEVEL_LOW));
-  EXPECT_TRUE(service->IsEngagementAtLeast(
-      url2, SiteEngagementService::ENGAGEMENT_LEVEL_MEDIUM));
-  EXPECT_FALSE(service->IsEngagementAtLeast(
-      url2, SiteEngagementService::ENGAGEMENT_LEVEL_HIGH));
-  EXPECT_FALSE(service->IsEngagementAtLeast(
-      url2, SiteEngagementService::ENGAGEMENT_LEVEL_MAX));
+      url2, blink::mojom::EngagementLevel::MEDIUM));
+  EXPECT_FALSE(
+      service->IsEngagementAtLeast(url2, blink::mojom::EngagementLevel::HIGH));
+  EXPECT_FALSE(
+      service->IsEngagementAtLeast(url2, blink::mojom::EngagementLevel::MAX));
 
   // Bring url2 to HIGH engagement.
   for (int i = 0; i < 9; ++i) {
     current_day += base::TimeDelta::FromDays(1);
     clock->SetNow(current_day);
     service->AddPoints(url2, 5.0);
   }
-  EXPECT_EQ(SiteEngagementService::ENGAGEMENT_LEVEL_HIGH,
+  EXPECT_EQ(blink::mojom::EngagementLevel::HIGH,
             service->GetEngagementLevel(url2));
 
+  EXPECT_TRUE(
+      service->IsEngagementAtLeast(url2, blink::mojom::EngagementLevel::NONE));
   EXPECT_TRUE(service->IsEngagementAtLeast(
-      url2, SiteEngagementService::ENGAGEMENT_LEVEL_NONE));
+      url2, blink::mojom::EngagementLevel::MINIMAL));
+  EXPECT_TRUE(
+      service->IsEngagementAtLeast(url2, blink::mojom::EngagementLevel::LOW));
   EXPECT_TRUE(service->IsEngagementAtLeast(
-      url2, SiteEngagementService::ENGAGEMENT_LEVEL_LOW));
-  EXPECT_TRUE(service->IsEngagementAtLeast(
-      url2, SiteEngagementService::ENGAGEMENT_LEVEL_MEDIUM));
-  EXPECT_TRUE(service->IsEngagementAtLeast(
-      url2, SiteEngagementService::ENGAGEMENT_LEVEL_HIGH));
-  EXPECT_FALSE(service->IsEngagementAtLeast(
-      url2, SiteEngagementService::ENGAGEMENT_LEVEL_MAX));
+      url2, blink::mojom::EngagementLevel::MEDIUM));
+  EXPECT_TRUE(
+      service->IsEngagementAtLeast(url2, blink::mojom::EngagementLevel::HIGH));
+  EXPECT_FALSE(
+      service->IsEngagementAtLeast(url2, blink::mojom::EngagementLevel::MAX));
 
   // Bring url2 to MAX engagement.
   for (int i = 0; i < 10; ++i) {
     current_day += base::TimeDelta::FromDays(1);
     clock->SetNow(current_day);
     service->AddPoints(url2, 5.0);
   }
-  EXPECT_EQ(SiteEngagementService::ENGAGEMENT_LEVEL_MAX,
+  EXPECT_EQ(blink::mojom::EngagementLevel::MAX,
             service->GetEngagementLevel(url2));
+  EXPECT_TRUE(
+      service->IsEngagementAtLeast(url2, blink::mojom::EngagementLevel::NONE));
   EXPECT_TRUE(service->IsEngagementAtLeast(
-      url2, SiteEngagementService::ENGAGEMENT_LEVEL_NONE));
+      url2, blink::mojom::EngagementLevel::MINIMAL));
+  EXPECT_TRUE(
+      service->IsEngagementAtLeast(url2, blink::mojom::EngagementLevel::LOW));
   EXPECT_TRUE(service->IsEngagementAtLeast(
-      url2, SiteEngagementService::ENGAGEMENT_LEVEL_LOW));
-  EXPECT_TRUE(service->IsEngagementAtLeast(
-      url2, SiteEngagementService::ENGAGEMENT_LEVEL_MEDIUM));
-  EXPECT_TRUE(service->IsEngagementAtLeast(
-      url2, SiteEngagementService::ENGAGEMENT_LEVEL_HIGH));
-  EXPECT_TRUE(service->IsEngagementAtLeast(
-      url2, SiteEngagementService::ENGAGEMENT_LEVEL_MAX));
+      url2, blink::mojom::EngagementLevel::MEDIUM));
+  EXPECT_TRUE(
+      service->IsEngagementAtLeast(url2, blink::mojom::EngagementLevel::HIGH));
+  EXPECT_TRUE(
+      service->IsEngagementAtLeast(url2, blink::mojom::EngagementLevel::MAX));
 }
 
 TEST_F(SiteEngagementServiceTest, Observers) {
   SiteEngagementService* service = SiteEngagementService::Get(profile());
 
   GURL url_score_1("http://www.google.com/maps");
   GURL url_score_2("http://www.google.com/drive");
   GURL url_score_3("http://www.google.com/");
   GURL url_not_called("https://www.google.com/");
 
@@ skipping 414 matching lines @@
 
   EXPECT_EQ(3, CheckScoreFromSettingsOnThread(content::BrowserThread::IO,
                                               settings_map, url1));
   EXPECT_EQ(3, CheckScoreFromSettingsOnThread(content::BrowserThread::FILE,
                                               settings_map, url2));
   EXPECT_EQ(4, CheckScoreFromSettingsOnThread(content::BrowserThread::FILE,
                                               incognito_settings_map, url1));
   EXPECT_EQ(3, CheckScoreFromSettingsOnThread(content::BrowserThread::IO,
                                               incognito_settings_map, url2));
 }