[HBD] Gate the advertising of Flash on Site Engagement.

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 11 matching lines @@
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/chrome_render_view_host_test_harness.h"
 #include "chrome/test/base/testing_profile.h"
 #include "components/content_settings/core/browser/content_settings_observer.h"
 #include "components/content_settings/core/browser/host_content_settings_map.h"
 #include "components/history/core/browser/history_database_params.h"
 #include "components/history/core/browser/history_service.h"
 #include "components/history/core/test/test_history_database.h"
 #include "components/prefs/pref_service.h"
+#include "content/public/browser/browser_thread.h"
 #include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/page_navigator.h"
 #include "content/public/browser/web_contents.h"
+#include "content/public/test/test_browser_thread_bundle.h"
 #include "content/public/test/web_contents_tester.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
 base::FilePath g_temp_history_dir;
 
 const int kMoreAccumulationsThanNeededToMaxDailyEngagement = 40;
 const int kMoreDaysThanNeededToMaxTotalEngagement = 40;
 const int kMorePeriodsThanNeededToDecayMaxScore = 40;
@@ skipping 92 matching lines @@
   GURL url_;
   double score_;
   bool callback_called_;
   base::RunLoop run_loop_;
 
   DISALLOW_COPY_AND_ASSIGN(ObserverTester);
 };
 
 class SiteEngagementServiceTest : public ChromeRenderViewHostTestHarness {
  public:
+  SiteEngagementServiceTest() : ChromeRenderViewHostTestHarness() {
+    SetThreadBundleOptions(content::TestBrowserThreadBundle::REAL_IO_THREAD |
+                           content::TestBrowserThreadBundle::REAL_DB_THREAD);
+  }
+
   void SetUp() override {
     ChromeRenderViewHostTestHarness::SetUp();
     ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
     g_temp_history_dir = temp_dir_.GetPath();
     HistoryServiceFactory::GetInstance()->SetTestingFactory(
         profile(), &BuildTestHistoryService);
     SiteEngagementScore::SetParamValuesForTesting();
   }
 
   void NavigateWithTransitionAndExpectHigherScore(
@@ skipping 23 matching lines @@
   }
 
   void SetParamValue(SiteEngagementScore::Variation variation, double value) {
     SiteEngagementScore::GetParamValues()[variation].second = value;
   }
 
   void AssertInRange(double expected, double actual) {
     EXPECT_NEAR(expected, actual, kMaxRoundingDeviation);
   }
 
+  double CheckScoreFromSettingsOnThread(
+      content::BrowserThread::ID thread_id,
+      HostContentSettingsMap* settings_map,
+      const GURL& url) {
+    double score = 0;
+    base::RunLoop run_loop;
+    content::BrowserThread::PostTaskAndReply(
+        thread_id, FROM_HERE,
+        base::Bind(&SiteEngagementServiceTest::CheckScoreFromSettings,
+                   base::Unretained(this), settings_map, url, &score),
+                   run_loop.QuitClosure());
+    run_loop.Run();
+    return score;
+  }
+
  private:
+  void CheckScoreFromSettings(HostContentSettingsMap* settings_map,
+                              const GURL& url,
+                              double *score) {
+    *score = SiteEngagementService::GetScoreFromSettings(settings_map, url);
+  }
+
   base::ScopedTempDir temp_dir_;
 };
 
 TEST_F(SiteEngagementServiceTest, GetMedianEngagement) {
   SiteEngagementService* service = SiteEngagementService::Get(profile());
   ASSERT_TRUE(service);
 
   GURL url1("http://www.google.com/");
   GURL url2("https://www.google.com/");
   GURL url3("https://drive.google.com/");
@@ skipping 1306 matching lines @@
 
   service->AddPoints(url3, 2);
   EXPECT_EQ(1, incognito_service->GetScore(url3));
   EXPECT_EQ(2, service->GetScore(url3));
 
   EXPECT_EQ(0, incognito_service->GetScore(url4));
   service->AddPoints(url4, 2);
   EXPECT_EQ(2, incognito_service->GetScore(url4));
   EXPECT_EQ(2, service->GetScore(url4));
 }
+
+TEST_F(SiteEngagementServiceTest, GetScoreFromSettings) {
+  GURL url1("http://www.google.com/");
+  GURL url2("https://www.google.com/");
+
+  HostContentSettingsMap* settings_map =
+      HostContentSettingsMapFactory::GetForProfile(profile());
+  HostContentSettingsMap* incognito_settings_map =
+      HostContentSettingsMapFactory::GetForProfile(
+          profile()->GetOffTheRecordProfile());
+
+  // All scores are 0 to start.
+  EXPECT_EQ(0, CheckScoreFromSettingsOnThread(content::BrowserThread::IO,
+                                              settings_map, url1));
+  EXPECT_EQ(0, CheckScoreFromSettingsOnThread(content::BrowserThread::FILE,
+                                              settings_map, url2));
+  EXPECT_EQ(0, CheckScoreFromSettingsOnThread(content::BrowserThread::FILE,
+                                              incognito_settings_map, url1));
+  EXPECT_EQ(0, CheckScoreFromSettingsOnThread(content::BrowserThread::IO,
+                                              incognito_settings_map, url2));
+
+  SiteEngagementService* service = SiteEngagementService::Get(profile());
+  ASSERT_TRUE(service);
+  service->AddPoints(url1, 1);
+  service->AddPoints(url2, 2);
+
+  EXPECT_EQ(1, CheckScoreFromSettingsOnThread(content::BrowserThread::FILE,
+                                              settings_map, url1));
+  EXPECT_EQ(2, CheckScoreFromSettingsOnThread(content::BrowserThread::IO,
+                                              settings_map, url2));
+  EXPECT_EQ(1, CheckScoreFromSettingsOnThread(content::BrowserThread::FILE,
+                                              incognito_settings_map, url1));
+  EXPECT_EQ(2, CheckScoreFromSettingsOnThread(content::BrowserThread::IO,
+                                              incognito_settings_map, url2));
+
+  SiteEngagementService* incognito_service =
+      SiteEngagementService::Get(profile()->GetOffTheRecordProfile());
+  ASSERT_TRUE(incognito_service);
+  incognito_service->AddPoints(url1, 3);
+  incognito_service->AddPoints(url2, 1);
+
+  EXPECT_EQ(1, CheckScoreFromSettingsOnThread(content::BrowserThread::IO,
+                                              settings_map, url1));
+  EXPECT_EQ(2, CheckScoreFromSettingsOnThread(content::BrowserThread::FILE,
+                                              settings_map, url2));
+  EXPECT_EQ(4, CheckScoreFromSettingsOnThread(content::BrowserThread::IO,
+                                              incognito_settings_map, url1));
+  EXPECT_EQ(3, CheckScoreFromSettingsOnThread(content::BrowserThread::FILE,
+                                              incognito_settings_map, url2));
+
+  service->AddPoints(url1, 2);
+  service->AddPoints(url2, 1);
+
+  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));
+}