Move prefix_set and parts of s_b_util into a new component safe_browsing_db.

components/safe_browsing_db/prefix_set_unittest.cc

 // Copyright (c) 2012 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/prefix_set.h"
+#include "components/safe_browsing_db/prefix_set.h"
 
 #include <algorithm>
 #include <iterator>
 #include <set>
 #include <string>
 
 #include "base/files/file_util.h"
 #include "base/files/scoped_file.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/logging.h"
 #include "base/md5.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/path_service.h"
 #include "base/rand_util.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/string_util.h"
-#include "chrome/common/chrome_paths.h"
+#include "components/safe_browsing_db/safe_browsing_db_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/platform_test.h"
 
 namespace {
 
 const SBPrefix kHighBitClear = 1000u * 1000u * 1000u;
 const SBPrefix kHighBitSet = 3u * 1000u * 1000u * 1000u;
 
 }  // namespace
 
@@ skipping 136 matching lines @@
     base::ScopedFILE file(base::OpenFile(filename, "r+b"));
     IncrementIntAt(file.get(), offset, inc);
     CleanChecksum(file.get());
     file.reset();
 
     int64 new_size_64;
     ASSERT_TRUE(base::GetFileSize(filename, &new_size_64));
     ASSERT_EQ(new_size_64, size_64);
   }
 
+  base::FilePath TestFilePath() {
+    base::FilePath path;
+    PathService::Get(base::DIR_SOURCE_ROOT, &path);
+    return path.AppendASCII("components")
+        .AppendASCII("test")
+        .AppendASCII("data")
+        .AppendASCII("SafeBrowsingDb");
+  }
+
   // Fill |prefixes| with values read from a reference file.  The reference file
   // was generated from a specific |shared_prefixes_|.
   bool ReadReferencePrefixes(std::vector<SBPrefix>* prefixes) {
     const char kRefname[] = "PrefixSetRef";
-    base::FilePath ref_path;
-    if (!PathService::Get(chrome::DIR_TEST_DATA, &ref_path))
-      return false;
-    ref_path = ref_path.AppendASCII("SafeBrowsing");
+    base::FilePath ref_path = TestFilePath();
     ref_path = ref_path.AppendASCII(kRefname);
 
     base::ScopedFILE file(base::OpenFile(ref_path, "r"));
     if (!file.get())
       return false;
     char buf[1024];
     while (fgets(buf, sizeof(buf), file.get())) {
       std::string trimmed;
       if (base::TRIM_TRAILING !=
           base::TrimWhitespace(buf, base::TRIM_ALL, &trimmed))
@@ skipping 481 matching lines @@
 
 // Test that a golden v2 file is discarded on read.  All platforms generating v2
 // files are little-endian, so there is no point to testing this transition
 // if/when a big-endian port is added.
 #if defined(ARCH_CPU_LITTLE_ENDIAN)
 TEST_F(PrefixSetTest, Version2) {
   std::vector<SBPrefix> ref_prefixes;
   ASSERT_TRUE(ReadReferencePrefixes(&ref_prefixes));
 
   const char kBasename[] = "PrefixSetVersion2";
-  base::FilePath golden_path;
-  ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &golden_path));
-  golden_path = golden_path.AppendASCII("SafeBrowsing");
+  base::FilePath golden_path = TestFilePath();
   golden_path = golden_path.AppendASCII(kBasename);
 
   scoped_ptr<const PrefixSet> prefix_set(PrefixSet::LoadFile(golden_path));
   ASSERT_FALSE(prefix_set.get());
 }
 #endif
 
 // Test that a golden v3 file can be read by the current code.  All platforms
 // generating v3 files are little-endian, so there is no point to testing this
 // transition if/when a big-endian port is added.
 #if defined(ARCH_CPU_LITTLE_ENDIAN)
 TEST_F(PrefixSetTest, Version3) {
   std::vector<SBPrefix> ref_prefixes;
   ASSERT_TRUE(ReadReferencePrefixes(&ref_prefixes));
 
   const char kBasename[] = "PrefixSetVersion3";
-  base::FilePath golden_path;
-  ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &golden_path));
-  golden_path = golden_path.AppendASCII("SafeBrowsing");
+  base::FilePath golden_path = TestFilePath();
   golden_path = golden_path.AppendASCII(kBasename);
 
   scoped_ptr<const PrefixSet> prefix_set(PrefixSet::LoadFile(golden_path));
   ASSERT_TRUE(prefix_set.get());
   CheckPrefixes(*prefix_set, ref_prefixes);
 
   const SBFullHash kHash1 = SBFullHashForString("www.evil.com/malware.html");
   const SBFullHash kHash2 = SBFullHashForString("www.evil.com/phishing.html");
 
   EXPECT_TRUE(prefix_set->Exists(kHash1));
   EXPECT_TRUE(prefix_set->Exists(kHash2));
   EXPECT_FALSE(prefix_set->PrefixExists(kHash1.prefix));
   EXPECT_FALSE(prefix_set->PrefixExists(kHash2.prefix));
 }
 #endif
 
 }  // namespace safe_browsing