Add pause/resume functionality to precache

components/precache/core/precache_database_unittest.cc

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/precache/core/precache_database.h"
 
 #include <stdint.h>
 
 #include <map>
 #include <memory>
 
 #include "base/containers/hash_tables.h"
 #include "base/files/file_path.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/message_loop/message_loop.h"
 #include "base/metrics/histogram_base.h"
 #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"
 
 namespace {
 
 using ::testing::ContainerEq;
 using ::testing::ElementsAre;
 using base::Bucket;
 
@@ skipping 344 matching lines @@
   EXPECT_THAT(histograms_.GetAllSamples("Precache.Latency.NonPrefetch"),
               ElementsAre(Bucket(0, 6), Bucket(kLatency.InMilliseconds(), 3)));
 
   EXPECT_THAT(histograms_.GetAllSamples("Precache.Saved"),
               ElementsAre(Bucket(kSize1, 1), Bucket(kSize3, 1)));
 
   EXPECT_THAT(histograms_.GetAllSamples("Precache.Saved.Cellular"),
               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