Add pause/resume functionality to precache

components/precache/core/precache_database_unittest.cc

Index: components/precache/core/precache_database_unittest.cc
diff --git a/components/precache/core/precache_database_unittest.cc b/components/precache/core/precache_database_unittest.cc
index 3f9253c744af757c3272e89fc7f0eceeb8b34204..a489dcebd7df88bedecc6b98fdc338266c9e389c 100644
--- a/components/precache/core/precache_database_unittest.cc
+++ b/components/precache/core/precache_database_unittest.cc
@@ -17,6 +17,7 @@
 #include "base/test/histogram_tester.h"
 #include "base/time/time.h"
 #include "components/history/core/browser/history_constants.h"
+#include "components/precache/core/proto/unfinished_work.pb.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "url/gurl.h"
@@ -381,6 +382,57 @@ TEST_F(PrecacheDatabaseTest, SampleInteraction) {
               ElementsAre(Bucket(kSize1, 1)));
 }
 
+TEST_F(PrecacheDatabaseTest, SaveAndGetUnfinishedWork) {

[sclittle, 2016/05/19 01:59:45]

You could probably just remove this test, since there are already session table
tests for this stuff.

[bengr, 2016/05/19 23:19:31]

Done.

+  std::unique_ptr<PrecacheUnfinishedWork> unfinished_work(
+      new PrecacheUnfinishedWork());
+  unfinished_work->add_top_host()->set_hostname("foo.com");
+  unfinished_work->add_top_host()->set_hostname("bar.com");
+  auto s = unfinished_work->mutable_config_settings();
+  s->set_top_sites_count(11);
+  s->add_forced_site("baz.com");
+  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/");
+  unfinished_work->set_total_bytes(13);
+  unfinished_work->set_network_bytes(14);
+  unfinished_work->set_num_manifest_urls(15);
+  base::Time sometime = base::Time::UnixEpoch();
+  unfinished_work->set_start_time(sometime.ToInternalValue());
+
+  precache_database_->SaveUnfinishedWork(std::move(unfinished_work));
+  std::unique_ptr<PrecacheUnfinishedWork> unfinished_work2 =
+      precache_database_->GetUnfinishedWork();
+
+  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());
+  EXPECT_EQ(1, unfinished_work2->config_settings().forced_site_size());
+  EXPECT_EQ("baz.com", unfinished_work2->config_settings().forced_site(0));
+  EXPECT_EQ(12, unfinished_work2->config_settings().top_resources_count());
+  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());
+  EXPECT_EQ("http://z.com/", unfinished_work2->resource(2).url());
+  EXPECT_EQ(13ul, unfinished_work2->total_bytes());
+  EXPECT_EQ(14ul, unfinished_work2->network_bytes());
+  EXPECT_EQ(15ul, unfinished_work2->num_manifest_urls());
+  EXPECT_EQ(base::Time::UnixEpoch(),
+            base::Time::FromInternalValue(unfinished_work2->start_time()));
+}
+
 }  // namespace
 
 }  // namespace precache