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 <map>
 
 #include "base/bind.h"
 #include "base/run_loop.h"
 #include "base/test/scoped_feature_list.h"
 #include "base/test/simple_test_clock.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 96 matching lines @@
     if (quit_closure) {
       quit_closure.Run();
       quit_closure.Reset();
     }
   }
 
   bool last_embargoed_status() { return last_embargoed_status_; }
 
   base::SimpleTestClock* clock() { return clock_; }
 
+  const char* GetDismissKey() {
+    return PermissionDecisionAutoBlocker::kPromptDismissCountKey;
+  }
+
+  const char* GetIgnoreKey() {
+    return PermissionDecisionAutoBlocker::kPromptIgnoreCountKey;
+  }
+
  private:
   PermissionDecisionAutoBlocker* autoblocker_;
   base::test::ScopedFeatureList feature_list_;
   base::SimpleTestClock* clock_;
   bool last_embargoed_status_;
 };
 
 TEST_F(PermissionDecisionAutoBlockerUnitTest, RemoveCountsByUrl) {
   GURL url1("https://www.google.com");
   GURL url2("https://www.example.com");
@@ skipping 241 matching lines @@
                                                      2000 /* timeout in ms */);
 
   clock()->Advance(base::TimeDelta::FromDays(1));
   UpdateEmbargoedStatus(content::PermissionType::GEOLOCATION, url);
   EXPECT_TRUE(last_embargoed_status());
 
   clock()->Advance(base::TimeDelta::FromDays(1));
   EXPECT_TRUE(
       autoblocker()->IsUnderEmbargo(content::PermissionType::GEOLOCATION, url));
 }
+
+// TODO(raymes): See crbug.com/681709. Remove after M60.
+TEST_F(PermissionDecisionAutoBlockerUnitTest,
+       MigrateNoDecisionCountToPermissionAutoBlockerData) {
+  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, autoblocker()->GetDismissCount(
+                   url, content::PermissionType::GEOLOCATION));
+  EXPECT_EQ(0, autoblocker()->GetIgnoreCount(
+                   url, content::PermissionType::GEOLOCATION));
+
+  // Trigger pref migration which happens at the creation of the
+  // HostContentSettingsMap.
+  {
+    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, autoblocker()->GetDismissCount(
+                     url, content::PermissionType::GEOLOCATION));
+  EXPECT_EQ(50, autoblocker()->GetIgnoreCount(
+                    url, content::PermissionType::GEOLOCATION));
+
+  // 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);
+
+  // Write to the old content setting again, but with different numbers.
+  permissions_dict.SetInteger(GetDismissKey(), 99);
+  permissions_dict.SetInteger(GetIgnoreKey(), 99);
+
+  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());
+
+  // Ensure that migrating again does nothing.
+  {
+    scoped_refptr<HostContentSettingsMap> temp_map(new HostContentSettingsMap(
+        profile()->GetPrefs(), false /* is_incognito_profile */,
+        false /* is_guest_profile */));
+    temp_map->ShutdownOnUIThread();
+  }
+
+  EXPECT_EQ(100, autoblocker()->GetDismissCount(
+                     url, content::PermissionType::GEOLOCATION));
+  EXPECT_EQ(50, autoblocker()->GetIgnoreCount(
+                    url, content::PermissionType::GEOLOCATION));
+}