OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/safe_browsing/prefix_set.h" | 5 #include "chrome/browser/safe_browsing/prefix_set.h" |
6 | 6 |
7 #include <algorithm> | 7 #include <algorithm> |
8 #include <iterator> | 8 #include <iterator> |
9 | 9 |
10 #include "base/file_util.h" | 10 #include "base/file_util.h" |
| 11 #include "base/files/scoped_temp_dir.h" |
11 #include "base/logging.h" | 12 #include "base/logging.h" |
12 #include "base/md5.h" | 13 #include "base/md5.h" |
13 #include "base/memory/scoped_ptr.h" | 14 #include "base/memory/scoped_ptr.h" |
14 #include "base/rand_util.h" | 15 #include "base/rand_util.h" |
15 #include "base/scoped_temp_dir.h" | |
16 #include "testing/gtest/include/gtest/gtest.h" | 16 #include "testing/gtest/include/gtest/gtest.h" |
17 #include "testing/platform_test.h" | 17 #include "testing/platform_test.h" |
18 | 18 |
19 namespace { | 19 namespace { |
20 | 20 |
21 class PrefixSetTest : public PlatformTest { | 21 class PrefixSetTest : public PlatformTest { |
22 protected: | 22 protected: |
23 // Constants for the v1 format. | 23 // Constants for the v1 format. |
24 static const size_t kMagicOffset = 0 * sizeof(uint32); | 24 static const size_t kMagicOffset = 0 * sizeof(uint32); |
25 static const size_t kVersionOffset = 1 * sizeof(uint32); | 25 static const size_t kVersionOffset = 1 * sizeof(uint32); |
(...skipping 113 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
139 file.reset(); | 139 file.reset(); |
140 | 140 |
141 int64 new_size_64; | 141 int64 new_size_64; |
142 ASSERT_TRUE(file_util::GetFileSize(filename, &new_size_64)); | 142 ASSERT_TRUE(file_util::GetFileSize(filename, &new_size_64)); |
143 ASSERT_EQ(new_size_64, size_64); | 143 ASSERT_EQ(new_size_64, size_64); |
144 } | 144 } |
145 | 145 |
146 // Tests should not modify this shared resource. | 146 // Tests should not modify this shared resource. |
147 static std::vector<SBPrefix> shared_prefixes_; | 147 static std::vector<SBPrefix> shared_prefixes_; |
148 | 148 |
149 ScopedTempDir temp_dir_; | 149 base::ScopedTempDir temp_dir_; |
150 }; | 150 }; |
151 | 151 |
152 std::vector<SBPrefix> PrefixSetTest::shared_prefixes_; | 152 std::vector<SBPrefix> PrefixSetTest::shared_prefixes_; |
153 | 153 |
154 // Test that a small sparse random input works. | 154 // Test that a small sparse random input works. |
155 TEST_F(PrefixSetTest, Baseline) { | 155 TEST_F(PrefixSetTest, Baseline) { |
156 safe_browsing::PrefixSet prefix_set(shared_prefixes_); | 156 safe_browsing::PrefixSet prefix_set(shared_prefixes_); |
157 CheckPrefixes(prefix_set, shared_prefixes_); | 157 CheckPrefixes(prefix_set, shared_prefixes_); |
158 } | 158 } |
159 | 159 |
(...skipping 298 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
458 file_util::ScopedFILE file(file_util::OpenFile(filename, "ab")); | 458 file_util::ScopedFILE file(file_util::OpenFile(filename, "ab")); |
459 const char buf[] = "im in ur base, killing ur d00dz."; | 459 const char buf[] = "im in ur base, killing ur d00dz."; |
460 ASSERT_EQ(strlen(buf), fwrite(buf, 1, strlen(buf), file.get())); | 460 ASSERT_EQ(strlen(buf), fwrite(buf, 1, strlen(buf), file.get())); |
461 file.reset(); | 461 file.reset(); |
462 scoped_ptr<safe_browsing::PrefixSet> | 462 scoped_ptr<safe_browsing::PrefixSet> |
463 prefix_set(safe_browsing::PrefixSet::LoadFile(filename)); | 463 prefix_set(safe_browsing::PrefixSet::LoadFile(filename)); |
464 ASSERT_FALSE(prefix_set.get()); | 464 ASSERT_FALSE(prefix_set.get()); |
465 } | 465 } |
466 | 466 |
467 } // namespace | 467 } // namespace |
OLD | NEW |