Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(436)

Unified Diff: components/subresource_filter/content/browser/fake_safe_browsing_database_manager.h

Issue 2820933002: [subresource_filter] add //chrome level unit test harness (Closed)
Patch Set: remove dep Created 3 years, 8 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: components/subresource_filter/content/browser/fake_safe_browsing_database_manager.h
diff --git a/components/subresource_filter/content/browser/fake_safe_browsing_database_manager.h b/components/subresource_filter/content/browser/fake_safe_browsing_database_manager.h
new file mode 100644
index 0000000000000000000000000000000000000000..ed4ccda41d160fa2103cb2d009e0072278416c13
--- /dev/null
+++ b/components/subresource_filter/content/browser/fake_safe_browsing_database_manager.h
@@ -0,0 +1,50 @@
+// Copyright 2017 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.
+
+#ifndef COMPONENTS_SUBRESOURCE_FILTER_CONTENT_BROWSER_FAKE_SAFE_BROWSING_DATABASE_MANAGER_H_
+#define COMPONENTS_SUBRESOURCE_FILTER_CONTENT_BROWSER_FAKE_SAFE_BROWSING_DATABASE_MANAGER_H_
+
+#include <map>
+
+#include "base/macros.h"
+#include "components/safe_browsing_db/test_database_manager.h"
+#include "content/public/common/resource_type.h"
+
+class GURL;
+
+// Database manager that allows any URL to be configured as blacklisted for
+// testing.
+class FakeSafeBrowsingDatabaseManager
+ : public safe_browsing::TestSafeBrowsingDatabaseManager {
+ public:
+ FakeSafeBrowsingDatabaseManager();
+
+ void AddBlacklistedUrl(const GURL& url,
+ safe_browsing::SBThreatType threat_type);
+
+ void SimulateTimeout();
+
+ protected:
+ ~FakeSafeBrowsingDatabaseManager() override;
+
+ // safe_browsing::TestSafeBrowsingDatabaseManager:
+ bool CheckUrlForSubresourceFilter(const GURL& url, Client* client) override;
+ bool CheckResourceUrl(const GURL& url, Client* client) override;
+ bool IsSupported() const override;
+ void CancelCheck(Client* client) override;
+ bool ChecksAreAlwaysAsync() const override;
+ bool CanCheckResourceType(
+ content::ResourceType /* resource_type */) const override;
+ safe_browsing::ThreatSource GetThreatSource() const override;
+ bool CheckExtensionIDs(const std::set<std::string>& extension_ids,
+ Client* client) override;
+
+ private:
+ std::map<GURL, safe_browsing::SBThreatType> url_to_threat_type_;
+ bool simulate_timeout_;
+
+ DISALLOW_COPY_AND_ASSIGN(FakeSafeBrowsingDatabaseManager);
+};
+
+#endif // COMPONENTS_SUBRESOURCE_FILTER_CONTENT_BROWSER_FAKE_SAFE_BROWSING_DATABASE_MANAGER_H_

Powered by Google App Engine
This is Rietveld 408576698