| 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 6abb86ecdb821fdda53240aa82a7cc047385357e..a8f213d86461cf0abd3f61ef14c010fecab2cf77 100644
|
| --- a/chrome/browser/extensions/fake_safe_browsing_database_manager.cc
|
| +++ b/chrome/browser/extensions/fake_safe_browsing_database_manager.cc
|
| @@ -11,20 +11,57 @@
|
| #include "base/bind_helpers.h"
|
| #include "base/memory/ref_counted.h"
|
| #include "base/message_loop/message_loop_proxy.h"
|
| +#include "base/run_loop.h"
|
| #include "chrome/browser/safe_browsing/safe_browsing_service.h"
|
| #include "chrome/browser/safe_browsing/safe_browsing_util.h"
|
|
|
| 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) {
|
|
|