Migrate content setting PROMPT_NO_DECISION_COUNT to PERMISSION_AUTOBLOCKER_DATA

chrome/browser/permissions/permission_decision_auto_blocker_unittest.cc

 // Copyright 2016 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/permissions/permission_decision_auto_blocker.h"
 
 #include "base/bind.h"
 #include "base/test/scoped_feature_list.h"
 #include "base/time/time.h"
 #include "chrome/browser/content_settings/host_content_settings_map_factory.h"
+#include "chrome/browser/permissions/permission_util.h"
 #include "chrome/common/chrome_features.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/host_content_settings_map.h"
 #include "components/safe_browsing_db/test_database_manager.h"
 #include "content/public/browser/permission_type.h"
 
 namespace {
 
 bool FilterGoogle(const GURL& url) {
   return url == "https://www.google.com/";
 }
 
 bool FilterAll(const GURL& url) {
@@ skipping 46 matching lines @@
   // Manually placing an origin, permission pair under embargo for blacklisting.
   // To embargo on dismissals, RecordDismissAndEmbargo can be used.
   void PlaceUnderBlacklistEmbargo(content::PermissionType permission,
                                   const GURL& url,
                                   HostContentSettingsMap* map,
                                   base::Time current_time) {
     PermissionDecisionAutoBlocker::PlaceUnderEmbargo(
         permission, url, map, current_time,
         PermissionDecisionAutoBlocker::kPermissionBlacklistEmbargoKey);
   }
+
+  const char* GetDismissKey() {
+    return PermissionDecisionAutoBlocker::kPromptDismissCountKey;
+  }
+
+  const char* GetIgnoreKey() {
+    return PermissionDecisionAutoBlocker::kPromptIgnoreCountKey;
+  }
 };
 
 TEST_F(PermissionDecisionAutoBlockerUnitTest, RemoveCountsByUrl) {
   GURL url1("https://www.google.com");
   GURL url2("https://www.example.com");
   base::test::ScopedFeatureList feature_list;
   feature_list.InitAndEnableFeature(features::kBlockPromptsIfDismissedOften);
 
   // Record some dismissals.
   EXPECT_FALSE(RecordDismissAndEmbargo(
@@ skipping 198 matching lines @@
   time_now += base::TimeDelta::FromDays(3);
   EXPECT_FALSE(PermissionDecisionAutoBlocker::IsUnderEmbargo(
       content::PermissionType::GEOLOCATION, profile(), url,
       time_now + base::TimeDelta::FromDays(5)));
 
   // Check that dismissal embargo is still set, even though the blacklisting
   // embargo has expired.
   EXPECT_TRUE(PermissionDecisionAutoBlocker::IsUnderEmbargo(
       content::PermissionType::GEOLOCATION, profile(), url, time_now));
 }
+
+// TODO(raymes): See crbug.com/681709. Remove after M60.
+TEST_F(PermissionDecisionAutoBlockerUnitTest,
+       MigrateNoDecisionCountToPermissionAutoblockerData) {

[dominickn, 2017/01/25 06:02:33]

Sp: AutoBlocker

+  GURL url("https://www.google.com");
+  auto* map = HostContentSettingsMapFactory::GetForProfile(profile());
+
+  // Write to the old content setting.
+  base::DictionaryValue permissions_dict;
+  permissions_dict.SetInteger(GetDismissKey(), 100);
+  permissions_dict.SetInteger(GetIgnoreKey(), 50);
+
+  base::DictionaryValue origin_dict;
+  origin_dict.Set(
+      PermissionUtil::GetPermissionString(content::PermissionType::GEOLOCATION),
+      permissions_dict.CreateDeepCopy());
+  map->SetWebsiteSettingDefaultScope(
+      url, GURL(), CONTENT_SETTINGS_TYPE_PROMPT_NO_DECISION_COUNT,
+      std::string(), origin_dict.CreateDeepCopy());
+
+  // Nothing should be migrated yet, so the current values should be 0.
+  EXPECT_EQ(0, PermissionDecisionAutoBlocker::GetDismissCount(
+                   url, content::PermissionType::GEOLOCATION, profile()));
+  EXPECT_EQ(0, PermissionDecisionAutoBlocker::GetIgnoreCount(
+                   url, content::PermissionType::GEOLOCATION, profile()));
+
+  // Trigger pref migration which happens at the creation of the
+  // HostContentSetitngsMap.

[dominickn, 2017/01/25 06:02:33]

sp. SettingsMap

+  {
+    scoped_refptr<HostContentSettingsMap> temp_map(new HostContentSettingsMap(
+        profile()->GetPrefs(), false /* is_incognito_profile */,
+        false /* is_guest_profile */));
+    temp_map->ShutdownOnUIThread();
+  }
+
+  // The values should now be migrated.
+  EXPECT_EQ(100, PermissionDecisionAutoBlocker::GetDismissCount(
+                     url, content::PermissionType::GEOLOCATION, profile()));
+  EXPECT_EQ(50, PermissionDecisionAutoBlocker::GetIgnoreCount(
+                    url, content::PermissionType::GEOLOCATION, profile()));
+
+  // The old pref should be deleted.
+  std::unique_ptr<base::DictionaryValue> old_dict =
+      base::DictionaryValue::From(map->GetWebsiteSetting(
+          url, GURL(), CONTENT_SETTINGS_TYPE_PROMPT_NO_DECISION_COUNT,
+          std::string(), nullptr));
+  EXPECT_EQ(nullptr, old_dict);
+}