Index: chrome/browser/extensions/fake_safe_browsing_database_manager.cc |
diff --git a/chrome/browser/extensions/fake_safe_browsing_database_manager.cc b/chrome/browser/extensions/fake_safe_browsing_database_manager.cc |
index 8b6f52f66e347823b43f76f7a1d7929de0139ff8..f3af7fba8e0f46a1eae96660d691ba2b9cbd0a24 100644 |
--- a/chrome/browser/extensions/fake_safe_browsing_database_manager.cc |
+++ b/chrome/browser/extensions/fake_safe_browsing_database_manager.cc |
@@ -31,6 +31,11 @@ FakeSafeBrowsingDatabaseManager& FakeSafeBrowsingDatabaseManager::Enable() { |
return *this; |
} |
+FakeSafeBrowsingDatabaseManager& FakeSafeBrowsingDatabaseManager::Disable() { |
+ enabled_ = false; |
+ return *this; |
+} |
+ |
FakeSafeBrowsingDatabaseManager& |
FakeSafeBrowsingDatabaseManager::ClearUnsafe() { |
unsafe_ids_.clear(); |
@@ -66,6 +71,18 @@ FakeSafeBrowsingDatabaseManager& FakeSafeBrowsingDatabaseManager::SetUnsafe( |
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); |
} |