Delete the omahaproxy-backed extension blacklist and clear its entries from the

preferences file.

BUG=295882
R=shess@chromium.org, yoz@chromium.org

Review URL: https://codereview.chromium.org/23591050

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@224680 0039d316-1c4b-4281-b951-d872f2087c98

1 files changed, 38 insertions, 2 deletions

diff --git a/chrome/browser/extensions/fake_safe_browsing_database_manager.cc b/chrome/browser/extensions/fake_safe_browsing_database_manager.cc
index 6abb86e..714a3fa 100644
--- a/chrome/browser/extensions/fake_safe_browsing_database_manager.cc
+++ b/chrome/browser/extensions/fake_safe_browsing_database_manager.cc
@@ -16,15 +16,51 @@
 
 namespace extensions {
 
-FakeSafeBrowsingDatabaseManager::FakeSafeBrowsingDatabaseManager()
+FakeSafeBrowsingDatabaseManager::FakeSafeBrowsingDatabaseManager(bool enabled)
     : SafeBrowsingDatabaseManager(
           make_scoped_refptr(SafeBrowsingService::CreateSafeBrowsingService())),
-      enabled_(false) {
+      enabled_(enabled) {
 }
 
 FakeSafeBrowsingDatabaseManager::~FakeSafeBrowsingDatabaseManager() {
 }
 
+FakeSafeBrowsingDatabaseManager& FakeSafeBrowsingDatabaseManager::Enable() {
+  enabled_ = true;
+  return *this;
+}
+
+FakeSafeBrowsingDatabaseManager&
+FakeSafeBrowsingDatabaseManager::ClearUnsafe() {
+  unsafe_ids_.clear();
+  return *this;
+}
+
+FakeSafeBrowsingDatabaseManager& FakeSafeBrowsingDatabaseManager::SetUnsafe(
+    const std::string& a) {
+  ClearUnsafe();
+  unsafe_ids_.insert(a);
+  return *this;
+}
+
+FakeSafeBrowsingDatabaseManager& FakeSafeBrowsingDatabaseManager::SetUnsafe(
+    const std::string& a, const std::string& b) {
+  SetUnsafe(a);
+  unsafe_ids_.insert(b);
+  return *this;
+}
+
+FakeSafeBrowsingDatabaseManager& FakeSafeBrowsingDatabaseManager::SetUnsafe(
+    const std::string& a, const std::string& b, const std::string& c) {
+  SetUnsafe(a, b);
+  unsafe_ids_.insert(c);
+  return *this;
+}
+
+void FakeSafeBrowsingDatabaseManager::NotifyUpdate() {
+  SafeBrowsingDatabaseManager::NotifyDatabaseUpdateFinished(true);
+}
+
 bool FakeSafeBrowsingDatabaseManager::CheckExtensionIDs(
     const std::set<std::string>& extension_ids,
     Client* client) {