Refactor safe browsing service tests.

chrome/browser/safe_browsing/v4_test_utils.cc

Index: chrome/browser/safe_browsing/v4_test_utils.cc
diff --git a/chrome/browser/safe_browsing/v4_test_utils.cc b/chrome/browser/safe_browsing/v4_test_utils.cc
new file mode 100644
index 0000000000000000000000000000000000000000..261fc92ee52165f6279328ed27a0f0cdc71393f6
--- /dev/null
+++ b/chrome/browser/safe_browsing/v4_test_utils.cc
@@ -0,0 +1,106 @@
+// 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.
+
+#include "chrome/browser/safe_browsing/v4_test_utils.h"
+
+#include "base/memory/ptr_util.h"
+#include "chrome/browser/safe_browsing/test_safe_browsing_service.h"
+#include "components/safe_browsing_db/v4_database.h"
+#include "components/safe_browsing_db/v4_get_hash_protocol_manager.h"

[mattm, 2017/02/17 21:49:53]

Don't need to duplicate includes that are in the v4_test_utils.h file

[melandory, 2017/02/27 15:01:48]

Done.

+#include "crypto/sha2.h"
+
+namespace safe_browsing {
+
+TestV4Store::TestV4Store(
+    const scoped_refptr<base::SequencedTaskRunner>& task_runner,
+    const base::FilePath& store_path)
+    : V4Store(task_runner, store_path, 0) {}
+
+bool TestV4Store::HasValidData() const {
+  return true;
+}
+
+void TestV4Store::MarkPrefixAsBad(HashPrefix prefix) {
+  hash_prefix_map_[prefix.size()] = prefix;
+}
+
+TestV4Database::TestV4Database(
+    const scoped_refptr<base::SequencedTaskRunner>& db_task_runner,
+    std::unique_ptr<StoreMap> store_map)
+    : V4Database(db_task_runner, std::move(store_map)) {}
+
+void TestV4Database::MarkPrefixAsBad(ListIdentifier list_id,
+                                     HashPrefix prefix) {
+  V4Store* base_store = store_map_->at(list_id).get();
+  TestV4Store* test_store = static_cast<TestV4Store*>(base_store);
+  test_store->MarkPrefixAsBad(prefix);
+}
+
+V4Store* TestV4StoreFactory::CreateV4Store(
+    const scoped_refptr<base::SequencedTaskRunner>& task_runner,
+    const base::FilePath& store_path) {
+  V4Store* new_store = new TestV4Store(task_runner, store_path);
+  new_store->Initialize();
+  return new_store;
+}
+
+TestV4DatabaseFactory::TestV4DatabaseFactory() : v4_db_(nullptr) {}
+
+std::unique_ptr<V4Database> TestV4DatabaseFactory::Create(
+    const scoped_refptr<base::SequencedTaskRunner>& db_task_runner,
+    std::unique_ptr<StoreMap> store_map) {
+  v4_db_ = new TestV4Database(db_task_runner, std::move(store_map));
+  return base::WrapUnique(v4_db_);
+}
+
+void TestV4DatabaseFactory::MarkPrefixAsBad(ListIdentifier list_id,
+                                            HashPrefix prefix) {
+  v4_db_->MarkPrefixAsBad(list_id, prefix);
+}
+
+TestV4GetHashProtocolManager::TestV4GetHashProtocolManager(
+    net::URLRequestContextGetter* request_context_getter,
+    const StoresToCheck& stores_to_check,
+    const V4ProtocolConfig& config)
+    : V4GetHashProtocolManager(request_context_getter,
+                               stores_to_check,
+                               config) {}
+
+void TestV4GetHashProtocolManager::AddToFullHashCache(FullHashInfo fhi) {
+  full_hash_cache_[fhi.full_hash].full_hash_infos.push_back(fhi);
+}
+
+std::unique_ptr<V4GetHashProtocolManager>
+TestV4GetHashProtocolManagerFactory::CreateProtocolManager(
+    net::URLRequestContextGetter* request_context_getter,
+    const StoresToCheck& stores_to_check,
+    const V4ProtocolConfig& config) {
+  pm_ = new TestV4GetHashProtocolManager(request_context_getter,
+                                         stores_to_check, config);
+  return base::WrapUnique(pm_);
+}
+
+FullHash GetFullHash(const GURL& url) {
+  std::string host;
+  std::string path;
+  V4ProtocolManagerUtil::CanonicalizeUrl(url, &host, &path, nullptr);
+
+  return crypto::SHA256HashString(host + path);
+}
+
+FullHashInfo GetFullHashInfo(const GURL& url, const ListIdentifier& list_id) {
+  return FullHashInfo(GetFullHash(url), list_id,
+                      base::Time::Now() + base::TimeDelta::FromMinutes(5));
+}
+
+FullHashInfo GetFullHashInfoWithMetadata(
+    const GURL& url,
+    const ListIdentifier& list_id,
+    ThreatPatternType threat_pattern_type) {
+  FullHashInfo fhi = GetFullHashInfo(url, list_id);
+  fhi.metadata.threat_pattern_type = threat_pattern_type;
+  return fhi;
+}
+
+}  // namespace safe_browsing