Grant origins engagement for having interactions on their notifications.

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 373 matching lines @@
   service->HandleUserInput(web_contents(),
                            SiteEngagementMetrics::ENGAGEMENT_SCROLL);
   NavigateAndCommit(url3);
   service->HandleUserInput(web_contents(),
                            SiteEngagementMetrics::ENGAGEMENT_TOUCH_GESTURE);
   EXPECT_DOUBLE_EQ(0.15, service->GetScore(url2));
   EXPECT_DOUBLE_EQ(0.1, service->GetScore(url3));
   EXPECT_DOUBLE_EQ(0.4, service->GetTotalEngagementPoints());
 }
 
+TEST_F(SiteEngagementServiceTest, GetTotalNotificationPoints) {
+  SiteEngagementService* service = SiteEngagementService::Get(profile());
+  ASSERT_TRUE(service);
+  base::HistogramTester histograms;
+
+  // The https and http versions of www.google.com should be separate.
+  GURL url1("https://www.google.com/");
+  GURL url2("http://www.google.com/");
+  GURL url3("http://drive.google.com/");
+
+  EXPECT_EQ(0, service->GetScore(url1));
+  EXPECT_EQ(0, service->GetScore(url2));
+  EXPECT_EQ(0, service->GetScore(url3));
+
+  service->HandleNotificationInteraction(url1);
+  EXPECT_DOUBLE_EQ(1.0, service->GetScore(url1));
+  EXPECT_DOUBLE_EQ(1.0, service->GetTotalEngagementPoints());
+  histograms.ExpectBucketCount(
+      SiteEngagementMetrics::kEngagementTypeHistogram,
+      SiteEngagementMetrics::ENGAGEMENT_NOTIFICATION_INTERACTION, 1);
+
+  service->HandleNotificationInteraction(url2);
+  EXPECT_DOUBLE_EQ(1.0, service->GetScore(url2));
+  EXPECT_DOUBLE_EQ(2.0, service->GetTotalEngagementPoints());
+  histograms.ExpectBucketCount(
+      SiteEngagementMetrics::kEngagementTypeHistogram,
+      SiteEngagementMetrics::ENGAGEMENT_NOTIFICATION_INTERACTION, 2);
+
+  service->HandleNotificationInteraction(url1);
+  EXPECT_DOUBLE_EQ(2.0, service->GetScore(url1));
+  EXPECT_DOUBLE_EQ(3.0, service->GetTotalEngagementPoints());
+  histograms.ExpectBucketCount(
+      SiteEngagementMetrics::kEngagementTypeHistogram,
+      SiteEngagementMetrics::ENGAGEMENT_NOTIFICATION_INTERACTION, 3);
+
+  service->HandleNotificationInteraction(url3);
+  EXPECT_DOUBLE_EQ(1.0, service->GetScore(url3));
+  EXPECT_DOUBLE_EQ(4.0, service->GetTotalEngagementPoints());
+  histograms.ExpectBucketCount(
+      SiteEngagementMetrics::kEngagementTypeHistogram,
+      SiteEngagementMetrics::ENGAGEMENT_NOTIFICATION_INTERACTION, 4);
+}
+
 TEST_F(SiteEngagementServiceTest, RestrictedToHTTPAndHTTPS) {
   SiteEngagementService* service = SiteEngagementService::Get(profile());
   ASSERT_TRUE(service);
 
   // The https and http versions of www.google.com should be separate.
   GURL url1("ftp://www.google.com/");
   GURL url2("file://blah");
   GURL url3("chrome://");
   GURL url4("about://config");
 
@@ skipping 979 matching lines @@
   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_score_4("http://maps.google.com/");
   GURL url_not_called("https://www.google.com/");
 
   // Create an observer and Observe(nullptr).
   ObserverTester tester_not_called(service, web_contents(), url_not_called, 1);
   tester_not_called.Observe(nullptr);
 
   {
     // Create an observer for navigation.
     ObserverTester tester(service, web_contents(), url_score_1, 0.5);
     NavigateAndCommit(url_score_1);
@@ skipping 35 matching lines @@
   // Add an observer for media playing in the background.
   {
     ObserverTester tester(service, web_contents(), url_score_3, 0.58);
     service->HandleMediaPlaying(web_contents(), true);
     tester.Wait();
 
     EXPECT_TRUE(tester.callback_called());
     EXPECT_FALSE(tester_not_called.callback_called());
     tester.Observe(nullptr);
   }
+
+  // Add an observer for notifications.
+  {
+    ObserverTester tester(service, nullptr, url_score_4, 1.0);
+    service->HandleNotificationInteraction(url_score_4);
+    tester.Wait();
+
+    EXPECT_TRUE(tester.callback_called());
+    EXPECT_FALSE(tester_not_called.callback_called());
+    tester.Observe(nullptr);
+  }
 }
 
 TEST_F(SiteEngagementServiceTest, ScoreDecayHistograms) {
   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);
   base::HistogramTester histograms;
@@ skipping 351 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));
 }