Add a feature which, when enabled, blocks permissions after X prompt dismissals.

chrome/browser/permissions/permission_prompt_decision_log_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_prompt_decision_log.h"
+
+#include "base/bind.h"
+#include "chrome/test/base/chrome_render_view_host_test_harness.h"
+#include "chrome/test/base/testing_profile.h"
+#include "content/public/browser/permission_type.h"
+
+namespace {
+
+bool FilterGoogle(const GURL& url) {
+  return url == GURL("https://www.google.com");
+}
+
+bool FilterAll(const GURL& url) {
+  return true;
+}
+
+}  // namespace
+
+class PermissionPromptDecisionLogUnitTest
+    : public ChromeRenderViewHostTestHarness {
+ protected:
+  int GetDismissalCount(const GURL& url,
+                        content::PermissionType permission) {
+    return PermissionPromptDecisionLog::GetDismissalCount(profile(), url,
+                                                          permission);
+  }
+
+  int RecordDismissalCount(const GURL& url,
+                           content::PermissionType permission) {
+    return PermissionPromptDecisionLog::RecordDismissalCount(profile(), url,

[raymes, 2016/08/05 03:27:48]

nit: I think it's usually better to use the public API in tests where possible.
In this case it seems like you could just call ShouldChangeDismissalToBlock and
use GetDismissalCount to check the counts where needed.

[dominickn, 2016/08/05 04:29:30]

Done.

+                                                             permission);
+  }
+
+};
+
+TEST_F(PermissionPromptDecisionLogUnitTest, RemoveCountsByUrl) {
+  GURL url1("https://www.google.com");
+  GURL url2("https://www.example.com");
+  EXPECT_EQ(1,
+            RecordDismissalCount(url1, content::PermissionType::GEOLOCATION));
+  EXPECT_EQ(2,
+            RecordDismissalCount(url1, content::PermissionType::GEOLOCATION));
+  EXPECT_EQ(3,
+            RecordDismissalCount(url1, content::PermissionType::GEOLOCATION));
+
+  EXPECT_EQ(1,
+            RecordDismissalCount(url1, content::PermissionType::NOTIFICATIONS));
+
+  EXPECT_EQ(1,
+            RecordDismissalCount(url2, content::PermissionType::GEOLOCATION));
+
+  PermissionPromptDecisionLog::RemoveCountsByUrl(profile(),
+                                                 base::Bind(&FilterGoogle));
+
+  // Expect that url1's dismissals are gone, but url2's remain.
+  EXPECT_EQ(0, GetDismissalCount(url1, content::PermissionType::GEOLOCATION));
+  EXPECT_EQ(0, GetDismissalCount(url1, content::PermissionType::NOTIFICATIONS));
+  EXPECT_EQ(1, GetDismissalCount(url2, content::PermissionType::GEOLOCATION));
+
+  // Add some more dismissals.
+  EXPECT_EQ(1,
+            RecordDismissalCount(url1, content::PermissionType::GEOLOCATION));
+  EXPECT_EQ(1,
+            RecordDismissalCount(url1, content::PermissionType::NOTIFICATIONS));
+  EXPECT_EQ(2,
+            RecordDismissalCount(url2, content::PermissionType::GEOLOCATION));
+
+  // Remove everything and expect that it's all gone.
+  PermissionPromptDecisionLog::RemoveCountsByUrl(profile(),
+                                                 base::Bind(&FilterAll));
+
+  EXPECT_EQ(0, GetDismissalCount(url1, content::PermissionType::GEOLOCATION));
+  EXPECT_EQ(0, GetDismissalCount(url1, content::PermissionType::NOTIFICATIONS));
+  EXPECT_EQ(0, GetDismissalCount(url2, content::PermissionType::GEOLOCATION));
+}