OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 <stdint.h> | 5 #include <stdint.h> |
6 | 6 |
7 #include "base/compiler_specific.h" | 7 #include "base/compiler_specific.h" |
8 #include "base/time/time.h" | 8 #include "base/time/time.h" |
9 #include "components/precache/core/precache_session_table.h" | 9 #include "components/precache/core/precache_session_table.h" |
10 #include "components/precache/core/proto/unfinished_work.pb.h" | 10 #include "components/precache/core/proto/unfinished_work.pb.h" |
(...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
49 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work( | 49 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work( |
50 new PrecacheUnfinishedWork()); | 50 new PrecacheUnfinishedWork()); |
51 unfinished_work->add_top_host()->set_hostname("foo.com"); | 51 unfinished_work->add_top_host()->set_hostname("foo.com"); |
52 unfinished_work->add_top_host()->set_hostname("bar.com"); | 52 unfinished_work->add_top_host()->set_hostname("bar.com"); |
53 auto* s = unfinished_work->mutable_config_settings(); | 53 auto* s = unfinished_work->mutable_config_settings(); |
54 s->set_top_sites_count(11); | 54 s->set_top_sites_count(11); |
55 s->add_forced_site("baz.com"); | 55 s->add_forced_site("baz.com"); |
56 s->set_top_resources_count(12); | 56 s->set_top_resources_count(12); |
57 s->set_max_bytes_per_resource(501); | 57 s->set_max_bytes_per_resource(501); |
58 s->set_max_bytes_total(1001); | 58 s->set_max_bytes_total(1001); |
59 unfinished_work->add_manifest()->set_url("http://a.com/"); | |
60 unfinished_work->add_manifest()->set_url("http://b.com/"); | |
61 unfinished_work->add_manifest()->set_url("http://c.com/"); | |
62 unfinished_work->add_resource()->set_url("http://x.com/"); | 59 unfinished_work->add_resource()->set_url("http://x.com/"); |
63 unfinished_work->add_resource()->set_url("http://y.com/"); | 60 unfinished_work->add_resource()->set_url("http://y.com/"); |
64 unfinished_work->add_resource()->set_url("http://z.com/"); | 61 unfinished_work->add_resource()->set_url("http://z.com/"); |
65 unfinished_work->set_total_bytes(13); | 62 unfinished_work->set_total_bytes(13); |
66 unfinished_work->set_network_bytes(14); | 63 unfinished_work->set_network_bytes(14); |
67 unfinished_work->set_num_manifest_urls(15); | 64 unfinished_work->set_num_manifest_urls(15); |
68 base::Time sometime = base::Time::UnixEpoch(); | 65 base::Time sometime = base::Time::UnixEpoch(); |
69 unfinished_work->set_start_time(sometime.ToInternalValue()); | 66 unfinished_work->set_start_time(sometime.ToInternalValue()); |
70 | 67 |
71 precache_session_table_->SaveUnfinishedWork(std::move(unfinished_work)); | 68 precache_session_table_->SaveUnfinishedWork(std::move(unfinished_work)); |
72 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work2 = | 69 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work2 = |
73 precache_session_table_->GetUnfinishedWork(); | 70 precache_session_table_->GetUnfinishedWork(); |
74 | 71 |
| 72 EXPECT_EQ(2, unfinished_work2->top_host_size()); |
75 EXPECT_EQ("foo.com", unfinished_work2->top_host(0).hostname()); | 73 EXPECT_EQ("foo.com", unfinished_work2->top_host(0).hostname()); |
76 EXPECT_EQ("bar.com", unfinished_work2->top_host(1).hostname()); | 74 EXPECT_EQ("bar.com", unfinished_work2->top_host(1).hostname()); |
77 EXPECT_EQ(11, unfinished_work2->config_settings().top_sites_count()); | 75 EXPECT_EQ(11, unfinished_work2->config_settings().top_sites_count()); |
78 EXPECT_EQ(1, unfinished_work2->config_settings().forced_site_size()); | 76 EXPECT_EQ(1, unfinished_work2->config_settings().forced_site_size()); |
79 EXPECT_EQ("baz.com", unfinished_work2->config_settings().forced_site(0)); | 77 EXPECT_EQ("baz.com", unfinished_work2->config_settings().forced_site(0)); |
80 EXPECT_EQ(12, unfinished_work2->config_settings().top_resources_count()); | 78 EXPECT_EQ(12, unfinished_work2->config_settings().top_resources_count()); |
81 EXPECT_EQ(501ul, | 79 EXPECT_EQ(501ul, |
82 unfinished_work2->config_settings().max_bytes_per_resource()); | 80 unfinished_work2->config_settings().max_bytes_per_resource()); |
83 EXPECT_EQ(1001ul, unfinished_work2->config_settings().max_bytes_total()); | 81 EXPECT_EQ(1001ul, unfinished_work2->config_settings().max_bytes_total()); |
84 EXPECT_EQ(3, unfinished_work2->manifest_size()); | |
85 EXPECT_EQ("http://a.com/", unfinished_work2->manifest(0).url()); | |
86 EXPECT_EQ("http://b.com/", unfinished_work2->manifest(1).url()); | |
87 EXPECT_EQ("http://c.com/", unfinished_work2->manifest(2).url()); | |
88 EXPECT_EQ(3, unfinished_work2->resource_size()); | 82 EXPECT_EQ(3, unfinished_work2->resource_size()); |
89 EXPECT_EQ("http://x.com/", unfinished_work2->resource(0).url()); | 83 EXPECT_EQ("http://x.com/", unfinished_work2->resource(0).url()); |
90 EXPECT_EQ("http://y.com/", unfinished_work2->resource(1).url()); | 84 EXPECT_EQ("http://y.com/", unfinished_work2->resource(1).url()); |
91 EXPECT_EQ("http://z.com/", unfinished_work2->resource(2).url()); | 85 EXPECT_EQ("http://z.com/", unfinished_work2->resource(2).url()); |
92 EXPECT_EQ(13ul, unfinished_work2->total_bytes()); | 86 EXPECT_EQ(13ul, unfinished_work2->total_bytes()); |
93 EXPECT_EQ(14ul, unfinished_work2->network_bytes()); | 87 EXPECT_EQ(14ul, unfinished_work2->network_bytes()); |
94 EXPECT_EQ(15ul, unfinished_work2->num_manifest_urls()); | 88 EXPECT_EQ(15ul, unfinished_work2->num_manifest_urls()); |
95 EXPECT_EQ(base::Time::UnixEpoch(), | 89 EXPECT_EQ(base::Time::UnixEpoch(), |
96 base::Time::FromInternalValue(unfinished_work2->start_time())); | 90 base::Time::FromInternalValue(unfinished_work2->start_time())); |
97 } | 91 } |
98 | 92 |
99 // Test that storing overwrites previous unfinished work. | 93 // Test that storing overwrites previous unfinished work. |
100 TEST_F(PrecacheSessionTableTest, SaveAgainAndGet) { | 94 TEST_F(PrecacheSessionTableTest, SaveAgainAndGet) { |
101 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work( | 95 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work( |
102 new PrecacheUnfinishedWork()); | 96 new PrecacheUnfinishedWork()); |
103 unfinished_work->add_manifest()->set_url("http://a.com/"); | 97 unfinished_work->add_top_host()->set_hostname("a.com"); |
104 precache_session_table_->SaveUnfinishedWork(std::move(unfinished_work)); | 98 precache_session_table_->SaveUnfinishedWork(std::move(unfinished_work)); |
105 | 99 |
106 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work2( | 100 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work2( |
107 new PrecacheUnfinishedWork()); | 101 new PrecacheUnfinishedWork()); |
108 unfinished_work2->add_manifest()->set_url("http://b.com/"); | 102 unfinished_work2->add_top_host()->set_hostname("b.com"); |
109 precache_session_table_->SaveUnfinishedWork(std::move(unfinished_work2)); | 103 precache_session_table_->SaveUnfinishedWork(std::move(unfinished_work2)); |
110 | 104 |
111 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work3 = | 105 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work3 = |
112 precache_session_table_->GetUnfinishedWork(); | 106 precache_session_table_->GetUnfinishedWork(); |
113 EXPECT_EQ("http://b.com/", unfinished_work3->manifest(0).url()); | 107 EXPECT_EQ("b.com", unfinished_work3->top_host(0).hostname()); |
114 } | 108 } |
115 | 109 |
116 // Test that reading does not remove unfinished work from storage. | 110 // Test that reading does not remove unfinished work from storage. |
117 TEST_F(PrecacheSessionTableTest, SaveAndGetAgain) { | 111 TEST_F(PrecacheSessionTableTest, SaveAndGetAgain) { |
118 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work( | 112 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work( |
119 new PrecacheUnfinishedWork()); | 113 new PrecacheUnfinishedWork()); |
120 unfinished_work->add_manifest()->set_url("http://a.com/"); | 114 unfinished_work->add_top_host()->set_hostname("a.com"); |
121 precache_session_table_->SaveUnfinishedWork(std::move(unfinished_work)); | 115 precache_session_table_->SaveUnfinishedWork(std::move(unfinished_work)); |
122 | 116 |
123 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work2 = | 117 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work2 = |
124 precache_session_table_->GetUnfinishedWork(); | 118 precache_session_table_->GetUnfinishedWork(); |
125 | 119 |
126 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work3 = | 120 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work3 = |
127 precache_session_table_->GetUnfinishedWork(); | 121 precache_session_table_->GetUnfinishedWork(); |
128 | 122 |
129 EXPECT_EQ("http://a.com/", unfinished_work3->manifest(0).url()); | 123 EXPECT_EQ("a.com", unfinished_work3->top_host(0).hostname()); |
130 } | 124 } |
131 | 125 |
132 // Test that storing a large proto works. | 126 // Test that storing a large proto works. |
133 TEST_F(PrecacheSessionTableTest, SaveManyURLs) { | 127 TEST_F(PrecacheSessionTableTest, SaveManyURLs) { |
134 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work( | 128 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work( |
135 new PrecacheUnfinishedWork()); | 129 new PrecacheUnfinishedWork()); |
136 for (int i = 0; i < 1000; ++i) | 130 for (int i = 0; i < 1000; ++i) |
137 unfinished_work->add_manifest()->set_url("http://a.com/"); | 131 unfinished_work->add_top_host()->set_hostname("a.com"); |
138 precache_session_table_->SaveUnfinishedWork(std::move(unfinished_work)); | 132 precache_session_table_->SaveUnfinishedWork(std::move(unfinished_work)); |
139 | 133 |
140 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work2 = | 134 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work2 = |
141 precache_session_table_->GetUnfinishedWork(); | 135 precache_session_table_->GetUnfinishedWork(); |
142 | 136 |
143 EXPECT_EQ(1000, unfinished_work2->manifest_size()); | 137 EXPECT_EQ(1000, unfinished_work2->top_host_size()); |
144 for (int i = 0; i < 1000; ++i) | 138 for (int i = 0; i < 1000; ++i) |
145 EXPECT_EQ("http://a.com/", unfinished_work2->manifest(i).url()); | 139 EXPECT_EQ("a.com", unfinished_work2->top_host(i).hostname()); |
146 } | 140 } |
147 | 141 |
148 // Test that reading after deletion returns no unfinished work. | 142 // Test that reading after deletion returns no unfinished work. |
149 TEST_F(PrecacheSessionTableTest, SaveDeleteGet) { | 143 TEST_F(PrecacheSessionTableTest, SaveDeleteGet) { |
150 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work( | 144 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work( |
151 new PrecacheUnfinishedWork()); | 145 new PrecacheUnfinishedWork()); |
152 unfinished_work->add_manifest()->set_url("http://a.com/"); | 146 unfinished_work->add_top_host()->set_hostname("a.com"); |
153 precache_session_table_->SaveUnfinishedWork(std::move(unfinished_work)); | 147 precache_session_table_->SaveUnfinishedWork(std::move(unfinished_work)); |
154 precache_session_table_->DeleteUnfinishedWork(); | 148 precache_session_table_->DeleteUnfinishedWork(); |
155 | 149 |
156 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work2 = | 150 std::unique_ptr<PrecacheUnfinishedWork> unfinished_work2 = |
157 precache_session_table_->GetUnfinishedWork(); | 151 precache_session_table_->GetUnfinishedWork(); |
158 | 152 |
159 EXPECT_EQ(0, unfinished_work2->manifest_size()); | 153 EXPECT_EQ(0, unfinished_work2->top_host_size()); |
160 } | 154 } |
161 | 155 |
162 } // namespace | 156 } // namespace |
163 | 157 |
164 } // namespace precache | 158 } // namespace precache |
OLD | NEW |