Manage all the testing classes for Blacklist in TestBlacklist.

chrome/browser/extensions/fake_safe_browsing_database_manager.cc

 // Copyright 2013 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/extensions/fake_safe_browsing_database_manager.h"
 
 #include <algorithm>
 #include <iterator>
 #include <vector>
 
@@ skipping 13 matching lines @@
 }
 
 FakeSafeBrowsingDatabaseManager::~FakeSafeBrowsingDatabaseManager() {
 }
 
 FakeSafeBrowsingDatabaseManager& FakeSafeBrowsingDatabaseManager::Enable() {
   enabled_ = true;
   return *this;
 }
 
+FakeSafeBrowsingDatabaseManager& FakeSafeBrowsingDatabaseManager::Disable() {
+  enabled_ = false;
+  return *this;
+}
+
 FakeSafeBrowsingDatabaseManager&
 FakeSafeBrowsingDatabaseManager::ClearUnsafe() {
   unsafe_ids_.clear();
   return *this;
 }
 
 FakeSafeBrowsingDatabaseManager& FakeSafeBrowsingDatabaseManager::SetUnsafe(
     const std::string& a) {
   ClearUnsafe();
   unsafe_ids_.insert(a);
@@ skipping 15 matching lines @@
 }
 
 FakeSafeBrowsingDatabaseManager& FakeSafeBrowsingDatabaseManager::SetUnsafe(
     const std::string& a, const std::string& b, const std::string& c,
     const std::string& d) {
   SetUnsafe(a, b, c);
   unsafe_ids_.insert(d);
   return *this;
 }
 
+FakeSafeBrowsingDatabaseManager& FakeSafeBrowsingDatabaseManager::AddUnsafe(
+    const std::string& a) {
+  unsafe_ids_.insert(a);
+  return *this;
+}
+
+FakeSafeBrowsingDatabaseManager& FakeSafeBrowsingDatabaseManager::RemoveUnsafe(
+    const std::string& a) {
+  unsafe_ids_.erase(a);
+  return *this;
+}
+
 void FakeSafeBrowsingDatabaseManager::NotifyUpdate() {
   SafeBrowsingDatabaseManager::NotifyDatabaseUpdateFinished(true);
 }
 
 bool FakeSafeBrowsingDatabaseManager::CheckExtensionIDs(
     const std::set<std::string>& extension_ids,
     Client* client) {
   if (!enabled_)
     return true;
 
@@ skipping 29 matching lines @@
   return false;
 }
 
 void FakeSafeBrowsingDatabaseManager::OnSafeBrowsingResult(
     scoped_ptr<SafeBrowsingCheck> result,
     Client* client) {
   client->OnSafeBrowsingResult(*result);
 }
 
 }  // namespace extensions