Add revocation metrics from OIB and content setting

chrome/browser/permissions/permission_util.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/permissions/permission_util.h"
 
 #include "base/logging.h"
 #include "chrome/browser/content_settings/host_content_settings_map_factory.h"
 #include "chrome/browser/permissions/permission_uma_util.h"
 #include "components/content_settings/core/browser/host_content_settings_map.h"
@@ skipping 67 matching lines @@
   return true;
 }
 
 PermissionUtil::ScopedRevocationReporter::ScopedRevocationReporter(
     Profile* profile,
     const GURL& primary_url,
     const GURL& secondary_url,
     ContentSettingsType content_type,
     PermissionSourceUI source_ui)
     : profile_(profile),
+      settings_map_(HostContentSettingsMapFactory::GetForProfile(profile_)),
       primary_url_(primary_url),
       secondary_url_(secondary_url),
       content_type_(content_type),
       source_ui_(source_ui) {
   if (!primary_url_.is_valid() ||
       (!secondary_url_.is_valid() && !secondary_url_.is_empty())) {
     is_initially_allowed_ = false;
     return;
   }
-  HostContentSettingsMap* map =
-      HostContentSettingsMapFactory::GetForProfile(profile_);
-  ContentSetting initial_content_setting = map->GetContentSetting(
+  ContentSetting initial_content_setting = settings_map_->GetContentSetting(
       primary_url_, secondary_url_, content_type_, std::string());
   is_initially_allowed_ = initial_content_setting == CONTENT_SETTING_ALLOW;
 }
 
+PermissionUtil::ScopedRevocationReporter::ScopedRevocationReporter(
+    Profile* profile,
+    const ContentSettingsPattern& primary_pattern,
+    const ContentSettingsPattern& secondary_pattern,
+    ContentSettingsType content_type,
+    PermissionSourceUI source_ui)
+    : ScopedRevocationReporter(
+          profile,
+          GURL(primary_pattern.ToString()),
+          GURL((secondary_pattern == ContentSettingsPattern::Wildcard())
+                   ? primary_pattern.ToString()
+                   : secondary_pattern.ToString()),
+          content_type,
+          source_ui) {}
+
 PermissionUtil::ScopedRevocationReporter::~ScopedRevocationReporter() {
   if (!is_initially_allowed_)
     return;
-  HostContentSettingsMap* map =
-      HostContentSettingsMapFactory::GetForProfile(profile_);
-  ContentSetting final_content_setting = map->GetContentSetting(
+  ContentSetting final_content_setting = settings_map_->GetContentSetting(
       primary_url_, secondary_url_, content_type_, std::string());
   if (final_content_setting != CONTENT_SETTING_ALLOW) {
     PermissionType permission_type;
     if (PermissionUtil::GetPermissionType(content_type_, &permission_type)) {
       PermissionUmaUtil::PermissionRevoked(permission_type, source_ui_,
                                            primary_url_, profile_);
     }
   }
 }
+
+void PermissionUtil::ScopedRevocationReporter::SetCustomSettingsMap(
+    HostContentSettingsMap* settings_map) {
+  this->settings_map_ = settings_map;
+}