OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 "components/precache/core/precache_url_table.h" | 5 #include "components/precache/core/precache_url_table.h" |
6 | 6 |
7 #include <map> | 7 #include <map> |
8 #include <memory> | 8 #include <memory> |
9 | 9 |
10 #include "base/compiler_specific.h" | 10 #include "base/compiler_specific.h" |
(...skipping 17 matching lines...) Expand all Loading... |
28 ASSERT_TRUE(db_->OpenInMemory()); | 28 ASSERT_TRUE(db_->OpenInMemory()); |
29 precache_url_table_->Init(db_.get()); | 29 precache_url_table_->Init(db_.get()); |
30 } | 30 } |
31 | 31 |
32 std::unique_ptr<PrecacheURLTable> precache_url_table_; | 32 std::unique_ptr<PrecacheURLTable> precache_url_table_; |
33 std::unique_ptr<sql::Connection> db_; | 33 std::unique_ptr<sql::Connection> db_; |
34 }; | 34 }; |
35 | 35 |
36 TEST_F(PrecacheURLTableTest, AddURLWithNoExistingRow) { | 36 TEST_F(PrecacheURLTableTest, AddURLWithNoExistingRow) { |
37 const base::Time kTime = base::Time::FromInternalValue(100); | 37 const base::Time kTime = base::Time::FromInternalValue(100); |
38 precache_url_table_->AddURL(GURL("http://url.com"), kTime); | 38 precache_url_table_->AddURL(GURL("http://url.com"), 1, true, kTime); |
39 | 39 |
40 std::map<GURL, base::Time> expected_map; | 40 std::map<GURL, base::Time> expected_map; |
41 expected_map[GURL("http://url.com")] = kTime; | 41 expected_map[GURL("http://url.com")] = kTime; |
42 | 42 |
43 std::map<GURL, base::Time> actual_map; | 43 std::map<GURL, base::Time> actual_map; |
44 precache_url_table_->GetAllDataForTesting(&actual_map); | 44 precache_url_table_->GetAllDataForTesting(&actual_map); |
45 EXPECT_EQ(expected_map, actual_map); | 45 EXPECT_EQ(expected_map, actual_map); |
46 } | 46 } |
47 | 47 |
48 TEST_F(PrecacheURLTableTest, AddURLWithExistingRow) { | 48 TEST_F(PrecacheURLTableTest, AddURLWithExistingRow) { |
49 const base::Time kOldTime = base::Time::FromInternalValue(50); | 49 const base::Time kOldTime = base::Time::FromInternalValue(50); |
50 const base::Time kNewTime = base::Time::FromInternalValue(100); | 50 const base::Time kNewTime = base::Time::FromInternalValue(100); |
51 precache_url_table_->AddURL(GURL("http://url.com"), kOldTime); | 51 precache_url_table_->AddURL(GURL("http://url.com"), 1, true, kOldTime); |
52 precache_url_table_->AddURL(GURL("http://url.com"), kNewTime); | 52 precache_url_table_->AddURL(GURL("http://url.com"), 1, true, kNewTime); |
53 | 53 |
54 std::map<GURL, base::Time> expected_map; | 54 std::map<GURL, base::Time> expected_map; |
55 expected_map[GURL("http://url.com")] = kNewTime; | 55 expected_map[GURL("http://url.com")] = kNewTime; |
56 | 56 |
57 std::map<GURL, base::Time> actual_map; | 57 std::map<GURL, base::Time> actual_map; |
58 precache_url_table_->GetAllDataForTesting(&actual_map); | 58 precache_url_table_->GetAllDataForTesting(&actual_map); |
59 EXPECT_EQ(expected_map, actual_map); | 59 EXPECT_EQ(expected_map, actual_map); |
60 } | 60 } |
61 | 61 |
62 TEST_F(PrecacheURLTableTest, DeleteURL) { | 62 TEST_F(PrecacheURLTableTest, DeleteURL) { |
63 const base::Time kStaysTime = base::Time::FromInternalValue(50); | 63 const base::Time kStaysTime = base::Time::FromInternalValue(50); |
64 const base::Time kDeletedTime = base::Time::FromInternalValue(100); | 64 const base::Time kDeletedTime = base::Time::FromInternalValue(100); |
65 | 65 |
66 precache_url_table_->AddURL(GURL("http://stays.com"), kStaysTime); | 66 precache_url_table_->AddURL(GURL("http://stays.com"), 1, true, kStaysTime); |
67 precache_url_table_->AddURL(GURL("http://deleted.com"), kDeletedTime); | 67 precache_url_table_->AddURL(GURL("http://deleted.com"), 1, true, |
| 68 kDeletedTime); |
68 | 69 |
69 precache_url_table_->DeleteURL(GURL("http://deleted.com")); | 70 precache_url_table_->DeleteURL(GURL("http://deleted.com")); |
70 | 71 |
71 std::map<GURL, base::Time> expected_map; | 72 std::map<GURL, base::Time> expected_map; |
72 expected_map[GURL("http://stays.com")] = kStaysTime; | 73 expected_map[GURL("http://stays.com")] = kStaysTime; |
73 | 74 |
74 std::map<GURL, base::Time> actual_map; | 75 std::map<GURL, base::Time> actual_map; |
75 precache_url_table_->GetAllDataForTesting(&actual_map); | 76 precache_url_table_->GetAllDataForTesting(&actual_map); |
76 EXPECT_EQ(expected_map, actual_map); | 77 EXPECT_EQ(expected_map, actual_map); |
77 } | 78 } |
78 | 79 |
79 TEST_F(PrecacheURLTableTest, HasURL) { | 80 TEST_F(PrecacheURLTableTest, IsURLPrecached) { |
80 EXPECT_FALSE(precache_url_table_->HasURL(GURL("http://url.com"))); | 81 EXPECT_FALSE(precache_url_table_->IsURLPrecached(GURL("http://url.com"))); |
81 | 82 |
82 precache_url_table_->AddURL(GURL("http://url.com"), | 83 precache_url_table_->AddURL(GURL("http://url.com"), 1, true, |
83 base::Time::FromInternalValue(100)); | 84 base::Time::FromInternalValue(100)); |
84 | 85 |
85 EXPECT_TRUE(precache_url_table_->HasURL(GURL("http://url.com"))); | 86 EXPECT_TRUE(precache_url_table_->IsURLPrecached(GURL("http://url.com"))); |
86 | 87 |
87 precache_url_table_->DeleteURL(GURL("http://url.com")); | 88 precache_url_table_->DeleteURL(GURL("http://url.com")); |
88 | 89 |
89 EXPECT_FALSE(precache_url_table_->HasURL(GURL("http://url.com"))); | 90 EXPECT_FALSE(precache_url_table_->IsURLPrecached(GURL("http://url.com"))); |
90 } | 91 } |
91 | 92 |
92 TEST_F(PrecacheURLTableTest, DeleteAllPrecachedBefore) { | 93 TEST_F(PrecacheURLTableTest, DeleteAllPrecachedBefore) { |
93 const base::Time kOldTime = base::Time::FromInternalValue(10); | 94 const base::Time kOldTime = base::Time::FromInternalValue(10); |
94 const base::Time kBeforeTime = base::Time::FromInternalValue(20); | 95 const base::Time kBeforeTime = base::Time::FromInternalValue(20); |
95 const base::Time kEndTime = base::Time::FromInternalValue(30); | 96 const base::Time kEndTime = base::Time::FromInternalValue(30); |
96 const base::Time kAfterTime = base::Time::FromInternalValue(40); | 97 const base::Time kAfterTime = base::Time::FromInternalValue(40); |
97 | 98 |
98 precache_url_table_->AddURL(GURL("http://old.com"), kOldTime); | 99 precache_url_table_->AddURL(GURL("http://old.com"), 1, true, kOldTime); |
99 precache_url_table_->AddURL(GURL("http://before.com"), kBeforeTime); | 100 precache_url_table_->AddURL(GURL("http://before.com"), 1, true, kBeforeTime); |
100 precache_url_table_->AddURL(GURL("http://end.com"), kEndTime); | 101 precache_url_table_->AddURL(GURL("http://end.com"), 1, true, kEndTime); |
101 precache_url_table_->AddURL(GURL("http://after.com"), kAfterTime); | 102 precache_url_table_->AddURL(GURL("http://after.com"), 1, true, kAfterTime); |
102 | 103 |
103 precache_url_table_->DeleteAllPrecachedBefore(kEndTime); | 104 precache_url_table_->DeleteAllPrecachedBefore(kEndTime); |
104 | 105 |
105 std::map<GURL, base::Time> expected_map; | 106 std::map<GURL, base::Time> expected_map; |
106 expected_map[GURL("http://end.com")] = kEndTime; | 107 expected_map[GURL("http://end.com")] = kEndTime; |
107 expected_map[GURL("http://after.com")] = kAfterTime; | 108 expected_map[GURL("http://after.com")] = kAfterTime; |
108 | 109 |
109 std::map<GURL, base::Time> actual_map; | 110 std::map<GURL, base::Time> actual_map; |
110 precache_url_table_->GetAllDataForTesting(&actual_map); | 111 precache_url_table_->GetAllDataForTesting(&actual_map); |
111 EXPECT_EQ(expected_map, actual_map); | 112 EXPECT_EQ(expected_map, actual_map); |
112 } | 113 } |
113 | 114 |
114 } // namespace | 115 } // namespace |
115 | 116 |
116 } // namespace precache | 117 } // namespace precache |
OLD | NEW |