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

Side by Side Diff: chrome/browser/safe_browsing/safe_browsing_store_unittest.cc

Issue 1420053005: Move code in components/safe_browsing_db and chrome/browser/s_b/ under the safe_browsing namespace. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@02_components_move
Patch Set: Other minor fixes incl. using "using safe_browsing::ClassName" instead of safe_browsing::ClassName everywhere. Created 5 years, 1 month 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 unified diff | Download patch
OLDNEW
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. 1 // Copyright (c) 2011 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/safe_browsing_store.h" 5 #include "chrome/browser/safe_browsing/safe_browsing_store.h"
6 #include "components/safe_browsing_db/util.h" 6 #include "components/safe_browsing_db/util.h"
7 7
8 #include "testing/gtest/include/gtest/gtest.h" 8 #include "testing/gtest/include/gtest/gtest.h"
9 9
10 namespace { 10 namespace safe_browsing {
11 11
12 const SBFullHash kHash1 = safe_browsing::SBFullHashForString("one"); 12 const SBFullHash kHash1 = SBFullHashForString("one");
13 const SBFullHash kHash2 = safe_browsing::SBFullHashForString("two"); 13 const SBFullHash kHash2 = SBFullHashForString("two");
14 const SBFullHash kHash3 = safe_browsing::SBFullHashForString("three"); 14 const SBFullHash kHash3 = SBFullHashForString("three");
15 const SBFullHash kHash4 = safe_browsing::SBFullHashForString("four"); 15 const SBFullHash kHash4 = SBFullHashForString("four");
16 const SBFullHash kHash5 = safe_browsing::SBFullHashForString("five"); 16 const SBFullHash kHash5 = SBFullHashForString("five");
17 const SBFullHash kHash6 = safe_browsing::SBFullHashForString("six"); 17 const SBFullHash kHash6 = SBFullHashForString("six");
18 const SBFullHash kHash7 = safe_browsing::SBFullHashForString("seven"); 18 const SBFullHash kHash7 = SBFullHashForString("seven");
19 19
20 const int kAddChunk1 = 1; // Use different chunk numbers just in case. 20 const int kAddChunk1 = 1; // Use different chunk numbers just in case.
21 const int kSubChunk1 = 2; 21 const int kSubChunk1 = 2;
22 const int kAddChunk2 = 3; 22 const int kAddChunk2 = 3;
23 const int kSubChunk2 = 4; 23 const int kSubChunk2 = 4;
24 const int kAddChunk3 = 5; 24 const int kAddChunk3 = 5;
25 const int kSubChunk3 = 6; 25 const int kSubChunk3 = 6;
26 const int kAddChunk4 = 7; 26 const int kAddChunk4 = 7;
27 const int kSubChunk4 = 8; 27 const int kSubChunk4 = 8;
28 const int kAddChunk5 = 9; 28 const int kAddChunk5 = 9;
(...skipping 16 matching lines...) Expand all
45 SBAddPrefixLess<SBAddPrefix,SBAddPrefix>); 45 SBAddPrefixLess<SBAddPrefix,SBAddPrefix>);
46 std::sort(sub_prefixes->begin(), sub_prefixes->end(), 46 std::sort(sub_prefixes->begin(), sub_prefixes->end(),
47 SBAddPrefixLess<SBSubPrefix,SBSubPrefix>); 47 SBAddPrefixLess<SBSubPrefix,SBSubPrefix>);
48 std::sort(add_full_hashes->begin(), add_full_hashes->end(), 48 std::sort(add_full_hashes->begin(), add_full_hashes->end(),
49 SBAddPrefixHashLess<SBAddFullHash,SBAddFullHash>); 49 SBAddPrefixHashLess<SBAddFullHash,SBAddFullHash>);
50 std::sort(sub_full_hashes->begin(), sub_full_hashes->end(), 50 std::sort(sub_full_hashes->begin(), sub_full_hashes->end(),
51 SBAddPrefixHashLess<SBSubFullHash,SBSubFullHash>); 51 SBAddPrefixHashLess<SBSubFullHash,SBSubFullHash>);
52 52
53 SBProcessSubs(add_prefixes, sub_prefixes, add_full_hashes, sub_full_hashes, 53 SBProcessSubs(add_prefixes, sub_prefixes, add_full_hashes, sub_full_hashes,
54 add_chunks_deleted, sub_chunks_deleted); 54 add_chunks_deleted, sub_chunks_deleted);
55 } 55 }
mattm 2015/11/11 01:10:16 Keep the constants and helpers inside anon namespa
vakh (old account. dont use) 2015/11/11 18:59:53 Done.
56 56
57 TEST(SafeBrowsingStoreTest, SBAddPrefixLess) { 57 TEST(SafeBrowsingStoreTest, SBAddPrefixLess) {
58 // prefix dominates. 58 // prefix dominates.
59 EXPECT_TRUE(SBAddPrefixLess(SBAddPrefix(11, 1), SBAddPrefix(10, 2))); 59 EXPECT_TRUE(SBAddPrefixLess(SBAddPrefix(11, 1), SBAddPrefix(10, 2)));
60 EXPECT_FALSE(SBAddPrefixLess(SBAddPrefix(10, 2), SBAddPrefix(11, 1))); 60 EXPECT_FALSE(SBAddPrefixLess(SBAddPrefix(10, 2), SBAddPrefix(11, 1)));
61 61
62 // After prefix, chunk_id. 62 // After prefix, chunk_id.
63 EXPECT_TRUE(SBAddPrefixLess(SBAddPrefix(10, 1), SBAddPrefix(11, 1))); 63 EXPECT_TRUE(SBAddPrefixLess(SBAddPrefix(10, 1), SBAddPrefix(11, 1)));
64 EXPECT_FALSE(SBAddPrefixLess(SBAddPrefix(11, 1), SBAddPrefix(10, 1))); 64 EXPECT_FALSE(SBAddPrefixLess(SBAddPrefix(11, 1), SBAddPrefix(10, 1)));
65 65
(...skipping 145 matching lines...) Expand 10 before | Expand all | Expand 10 after
211 const base::hash_set<int32> no_deletions; 211 const base::hash_set<int32> no_deletions;
212 ProcessHelper(&add_prefixes, &sub_prefixes, &add_hashes, &sub_hashes, 212 ProcessHelper(&add_prefixes, &sub_prefixes, &add_hashes, &sub_hashes,
213 no_deletions, no_deletions); 213 no_deletions, no_deletions);
214 214
215 ASSERT_EQ(1U, add_prefixes.size()); 215 ASSERT_EQ(1U, add_prefixes.size());
216 EXPECT_EQ(kAddChunk7, add_prefixes[0].chunk_id); 216 EXPECT_EQ(kAddChunk7, add_prefixes[0].chunk_id);
217 EXPECT_EQ(kHash2.prefix, add_prefixes[0].prefix); 217 EXPECT_EQ(kHash2.prefix, add_prefixes[0].prefix);
218 218
219 ASSERT_EQ(2U, add_hashes.size()); 219 ASSERT_EQ(2U, add_hashes.size());
220 EXPECT_EQ(kAddChunk5, add_hashes[0].chunk_id); 220 EXPECT_EQ(kAddChunk5, add_hashes[0].chunk_id);
221 EXPECT_TRUE(safe_browsing::SBFullHashEqual(kHash4, add_hashes[0].full_hash)); 221 EXPECT_TRUE(SBFullHashEqual(kHash4, add_hashes[0].full_hash));
222 EXPECT_EQ(kAddChunk6, add_hashes[1].chunk_id); 222 EXPECT_EQ(kAddChunk6, add_hashes[1].chunk_id);
223 EXPECT_TRUE(safe_browsing::SBFullHashEqual(kHash6, add_hashes[1].full_hash)); 223 EXPECT_TRUE(SBFullHashEqual(kHash6, add_hashes[1].full_hash));
224 224
225 ASSERT_EQ(1U, sub_prefixes.size()); 225 ASSERT_EQ(1U, sub_prefixes.size());
226 EXPECT_EQ(kSubChunk4, sub_prefixes[0].chunk_id); 226 EXPECT_EQ(kSubChunk4, sub_prefixes[0].chunk_id);
227 EXPECT_EQ(kAddChunk4, sub_prefixes[0].add_chunk_id); 227 EXPECT_EQ(kAddChunk4, sub_prefixes[0].add_chunk_id);
228 EXPECT_EQ(kHash3.prefix, sub_prefixes[0].add_prefix); 228 EXPECT_EQ(kHash3.prefix, sub_prefixes[0].add_prefix);
229 229
230 ASSERT_EQ(1U, sub_hashes.size()); 230 ASSERT_EQ(1U, sub_hashes.size());
231 EXPECT_EQ(kSubChunk3, sub_hashes[0].chunk_id); 231 EXPECT_EQ(kSubChunk3, sub_hashes[0].chunk_id);
232 EXPECT_EQ(kAddChunk3, sub_hashes[0].add_chunk_id); 232 EXPECT_EQ(kAddChunk3, sub_hashes[0].add_chunk_id);
233 EXPECT_TRUE(safe_browsing::SBFullHashEqual(kHash7, sub_hashes[0].full_hash)); 233 EXPECT_TRUE(SBFullHashEqual(kHash7, sub_hashes[0].full_hash));
234 } 234 }
235 235
236 // Test chunk deletions, and ordering of deletions WRT subs knocking 236 // Test chunk deletions, and ordering of deletions WRT subs knocking
237 // out adds. 237 // out adds.
238 TEST(SafeBrowsingStoreTest, SBProcessSubsDeleteChunk) { 238 TEST(SafeBrowsingStoreTest, SBProcessSubsDeleteChunk) {
239 // A full hash which shares prefix with another. 239 // A full hash which shares prefix with another.
240 const SBFullHash kHash1mod = ModifyHashAfterPrefix(kHash1, 1); 240 const SBFullHash kHash1mod = ModifyHashAfterPrefix(kHash1, 1);
241 241
242 SBAddPrefixes add_prefixes; 242 SBAddPrefixes add_prefixes;
243 std::vector<SBAddFullHash> add_hashes; 243 std::vector<SBAddFullHash> add_hashes;
(...skipping 24 matching lines...) Expand all
268 sub_deletions.insert(kSubChunk1); 268 sub_deletions.insert(kSubChunk1);
269 ProcessHelper(&add_prefixes, &sub_prefixes, &add_hashes, &sub_hashes, 269 ProcessHelper(&add_prefixes, &sub_prefixes, &add_hashes, &sub_hashes,
270 no_deletions, sub_deletions); 270 no_deletions, sub_deletions);
271 271
272 ASSERT_EQ(1U, add_prefixes.size()); 272 ASSERT_EQ(1U, add_prefixes.size());
273 EXPECT_EQ(kAddChunk1, add_prefixes[0].chunk_id); 273 EXPECT_EQ(kAddChunk1, add_prefixes[0].chunk_id);
274 EXPECT_EQ(kHash2.prefix, add_prefixes[0].prefix); 274 EXPECT_EQ(kHash2.prefix, add_prefixes[0].prefix);
275 275
276 ASSERT_EQ(1U, add_hashes.size()); 276 ASSERT_EQ(1U, add_hashes.size());
277 EXPECT_EQ(kAddChunk1, add_hashes[0].chunk_id); 277 EXPECT_EQ(kAddChunk1, add_hashes[0].chunk_id);
278 EXPECT_TRUE(safe_browsing::SBFullHashEqual(kHash6, add_hashes[0].full_hash)); 278 EXPECT_TRUE(SBFullHashEqual(kHash6, add_hashes[0].full_hash));
279 279
280 EXPECT_TRUE(sub_prefixes.empty()); 280 EXPECT_TRUE(sub_prefixes.empty());
281 EXPECT_TRUE(sub_hashes.empty()); 281 EXPECT_TRUE(sub_hashes.empty());
282 282
283 // Delete the adds, also. 283 // Delete the adds, also.
284 base::hash_set<int32> add_deletions; 284 base::hash_set<int32> add_deletions;
285 add_deletions.insert(kAddChunk1); 285 add_deletions.insert(kAddChunk1);
286 ProcessHelper(&add_prefixes, &sub_prefixes, &add_hashes, &sub_hashes, 286 ProcessHelper(&add_prefixes, &sub_prefixes, &add_hashes, &sub_hashes,
287 add_deletions, no_deletions); 287 add_deletions, no_deletions);
288 288
289 EXPECT_TRUE(add_prefixes.empty()); 289 EXPECT_TRUE(add_prefixes.empty());
290 EXPECT_TRUE(add_hashes.empty()); 290 EXPECT_TRUE(add_hashes.empty());
291 EXPECT_TRUE(sub_prefixes.empty()); 291 EXPECT_TRUE(sub_prefixes.empty());
292 EXPECT_TRUE(sub_hashes.empty()); 292 EXPECT_TRUE(sub_hashes.empty());
293 } 293 }
294 294
295 TEST(SafeBrowsingStoreTest, Y2K38) { 295 TEST(SafeBrowsingStoreTest, Y2K38) {
296 const base::Time now = base::Time::Now(); 296 const base::Time now = base::Time::Now();
297 const base::Time future = now + base::TimeDelta::FromDays(3*365); 297 const base::Time future = now + base::TimeDelta::FromDays(3*365);
298 298
299 // TODO: Fix file format before 2035. 299 // TODO: Fix file format before 2035.
300 EXPECT_GT(static_cast<int32>(future.ToTimeT()), 0) 300 EXPECT_GT(static_cast<int32>(future.ToTimeT()), 0)
301 << " (int32)time_t is running out."; 301 << " (int32)time_t is running out.";
302 } 302 }
303 303
304 } // namespace 304 } // namespace safe_browsing
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698