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 |
index f36c437d4ba1b1bf294b6fa08569440b20cbd65d..693ab5292ae05d2395534425c76ad6da209ae9e3 100644 |
--- a/components/subresource_filter/content/browser/fake_safe_browsing_database_manager.h |
+++ b/components/subresource_filter/content/browser/fake_safe_browsing_database_manager.h |
@@ -22,7 +22,9 @@ class FakeSafeBrowsingDatabaseManager |
FakeSafeBrowsingDatabaseManager(); |
void AddBlacklistedUrl(const GURL& url, |
- safe_browsing::SBThreatType threat_type); |
+ safe_browsing::SBThreatType threat_type, |
+ safe_browsing::ThreatPatternType pattern_type = |
+ safe_browsing::ThreatPatternType::NONE); |
void SimulateTimeout(); |
@@ -44,8 +46,11 @@ class FakeSafeBrowsingDatabaseManager |
private: |
void OnCheckUrlForSubresourceFilterComplete(Client* client, const GURL& url); |
- std::map<GURL, safe_browsing::SBThreatType> url_to_threat_type_; |
std::set<Client*> checks_; |
+ std::map< |
+ GURL, |
+ std::pair<safe_browsing::SBThreatType, safe_browsing::ThreatPatternType>> |
+ url_to_threat_type_; |
bool simulate_timeout_; |
DISALLOW_COPY_AND_ASSIGN(FakeSafeBrowsingDatabaseManager); |