Index: components/precache/core/precache_session_table_unittest.cc |
diff --git a/components/precache/core/precache_session_table_unittest.cc b/components/precache/core/precache_session_table_unittest.cc |
index 7ab84234df0819797e05d205785448277f9adcd6..a354032a5574613e459d4fab2e024c2533b4b938 100644 |
--- a/components/precache/core/precache_session_table_unittest.cc |
+++ b/components/precache/core/precache_session_table_unittest.cc |
@@ -56,9 +56,6 @@ TEST_F(PrecacheSessionTableTest, SaveAndGetUnfinishedWork) { |
s->set_top_resources_count(12); |
s->set_max_bytes_per_resource(501); |
s->set_max_bytes_total(1001); |
- unfinished_work->add_manifest()->set_url("http://a.com/"); |
- unfinished_work->add_manifest()->set_url("http://b.com/"); |
- unfinished_work->add_manifest()->set_url("http://c.com/"); |
unfinished_work->add_resource()->set_url("http://x.com/"); |
unfinished_work->add_resource()->set_url("http://y.com/"); |
unfinished_work->add_resource()->set_url("http://z.com/"); |
@@ -72,6 +69,7 @@ TEST_F(PrecacheSessionTableTest, SaveAndGetUnfinishedWork) { |
std::unique_ptr<PrecacheUnfinishedWork> unfinished_work2 = |
precache_session_table_->GetUnfinishedWork(); |
+ EXPECT_EQ(2, unfinished_work2->top_host_size()); |
EXPECT_EQ("foo.com", unfinished_work2->top_host(0).hostname()); |
EXPECT_EQ("bar.com", unfinished_work2->top_host(1).hostname()); |
EXPECT_EQ(11, unfinished_work2->config_settings().top_sites_count()); |
@@ -81,10 +79,6 @@ TEST_F(PrecacheSessionTableTest, SaveAndGetUnfinishedWork) { |
EXPECT_EQ(501ul, |
unfinished_work2->config_settings().max_bytes_per_resource()); |
EXPECT_EQ(1001ul, unfinished_work2->config_settings().max_bytes_total()); |
- EXPECT_EQ(3, unfinished_work2->manifest_size()); |
- EXPECT_EQ("http://a.com/", unfinished_work2->manifest(0).url()); |
- EXPECT_EQ("http://b.com/", unfinished_work2->manifest(1).url()); |
- EXPECT_EQ("http://c.com/", unfinished_work2->manifest(2).url()); |
EXPECT_EQ(3, unfinished_work2->resource_size()); |
EXPECT_EQ("http://x.com/", unfinished_work2->resource(0).url()); |
EXPECT_EQ("http://y.com/", unfinished_work2->resource(1).url()); |
@@ -100,24 +94,24 @@ TEST_F(PrecacheSessionTableTest, SaveAndGetUnfinishedWork) { |
TEST_F(PrecacheSessionTableTest, SaveAgainAndGet) { |
std::unique_ptr<PrecacheUnfinishedWork> unfinished_work( |
new PrecacheUnfinishedWork()); |
- unfinished_work->add_manifest()->set_url("http://a.com/"); |
+ unfinished_work->add_top_host()->set_hostname("a.com"); |
precache_session_table_->SaveUnfinishedWork(std::move(unfinished_work)); |
std::unique_ptr<PrecacheUnfinishedWork> unfinished_work2( |
new PrecacheUnfinishedWork()); |
- unfinished_work2->add_manifest()->set_url("http://b.com/"); |
+ unfinished_work2->add_top_host()->set_hostname("b.com"); |
precache_session_table_->SaveUnfinishedWork(std::move(unfinished_work2)); |
std::unique_ptr<PrecacheUnfinishedWork> unfinished_work3 = |
precache_session_table_->GetUnfinishedWork(); |
- EXPECT_EQ("http://b.com/", unfinished_work3->manifest(0).url()); |
+ EXPECT_EQ("b.com", unfinished_work3->top_host(0).hostname()); |
} |
// Test that reading does not remove unfinished work from storage. |
TEST_F(PrecacheSessionTableTest, SaveAndGetAgain) { |
std::unique_ptr<PrecacheUnfinishedWork> unfinished_work( |
new PrecacheUnfinishedWork()); |
- unfinished_work->add_manifest()->set_url("http://a.com/"); |
+ unfinished_work->add_top_host()->set_hostname("a.com"); |
precache_session_table_->SaveUnfinishedWork(std::move(unfinished_work)); |
std::unique_ptr<PrecacheUnfinishedWork> unfinished_work2 = |
@@ -126,7 +120,7 @@ TEST_F(PrecacheSessionTableTest, SaveAndGetAgain) { |
std::unique_ptr<PrecacheUnfinishedWork> unfinished_work3 = |
precache_session_table_->GetUnfinishedWork(); |
- EXPECT_EQ("http://a.com/", unfinished_work3->manifest(0).url()); |
+ EXPECT_EQ("a.com", unfinished_work3->top_host(0).hostname()); |
} |
// Test that storing a large proto works. |
@@ -134,29 +128,29 @@ TEST_F(PrecacheSessionTableTest, SaveManyURLs) { |
std::unique_ptr<PrecacheUnfinishedWork> unfinished_work( |
new PrecacheUnfinishedWork()); |
for (int i = 0; i < 1000; ++i) |
- unfinished_work->add_manifest()->set_url("http://a.com/"); |
+ unfinished_work->add_top_host()->set_hostname("a.com"); |
precache_session_table_->SaveUnfinishedWork(std::move(unfinished_work)); |
std::unique_ptr<PrecacheUnfinishedWork> unfinished_work2 = |
precache_session_table_->GetUnfinishedWork(); |
- EXPECT_EQ(1000, unfinished_work2->manifest_size()); |
+ EXPECT_EQ(1000, unfinished_work2->top_host_size()); |
for (int i = 0; i < 1000; ++i) |
- EXPECT_EQ("http://a.com/", unfinished_work2->manifest(i).url()); |
+ EXPECT_EQ("a.com", unfinished_work2->top_host(i).hostname()); |
} |
// Test that reading after deletion returns no unfinished work. |
TEST_F(PrecacheSessionTableTest, SaveDeleteGet) { |
std::unique_ptr<PrecacheUnfinishedWork> unfinished_work( |
new PrecacheUnfinishedWork()); |
- unfinished_work->add_manifest()->set_url("http://a.com/"); |
+ unfinished_work->add_top_host()->set_hostname("a.com"); |
precache_session_table_->SaveUnfinishedWork(std::move(unfinished_work)); |
precache_session_table_->DeleteUnfinishedWork(); |
std::unique_ptr<PrecacheUnfinishedWork> unfinished_work2 = |
precache_session_table_->GetUnfinishedWork(); |
- EXPECT_EQ(0, unfinished_work2->manifest_size()); |
+ EXPECT_EQ(0, unfinished_work2->top_host_size()); |
} |
} // namespace |