| 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 |