predictors: Pass manifests from Bork to store in ResourcePrefetchPredictor.

chrome/browser/predictors/resource_prefetch_predictor_unittest.cc

 // Copyright 2014 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 "chrome/browser/predictors/resource_prefetch_predictor.h"
 
 #include <iostream>
 #include <memory>
 #include <utility>
 
@@ skipping 124 matching lines @@
                void(PrefetchDataMap* url_data_map,
                     PrefetchDataMap* host_data_map,
                     RedirectDataMap* url_redirect_data_map,
                     RedirectDataMap* host_redirect_data_map,
                     ManifestDataMap* manifest_data_map));
   MOCK_METHOD4(UpdateData,
                void(const PrefetchData& url_data,
                     const PrefetchData& host_data,
                     const RedirectData& url_redirect_data,
                     const RedirectData& host_redirect_data));
+  MOCK_METHOD2(UpdateManifestData,
+               void(const std::string& host,
+                    const precache::PrecacheManifest& manifest_data));
   MOCK_METHOD2(DeleteResourceData,
                void(const std::vector<std::string>& urls,
                     const std::vector<std::string>& hosts));
   MOCK_METHOD2(DeleteSingleResourceDataPoint,
                void(const std::string& key, PrefetchKeyType key_type));
   MOCK_METHOD2(DeleteRedirectData,
                void(const std::vector<std::string>& urls,
                     const std::vector<std::string>& hosts));
   MOCK_METHOD2(DeleteSingleRedirectDataPoint,
                void(const std::string& key, PrefetchKeyType key_type));
+  MOCK_METHOD1(DeleteManifestData, void(const std::vector<std::string>& hosts));
   MOCK_METHOD0(DeleteAllData, void());
 
  protected:
   ~MockResourcePrefetchPredictorTables() { }
 };
 
 class MockResourcePrefetchPredictorObserver : public TestObserver {
  public:
   explicit MockResourcePrefetchPredictorObserver(
       ResourcePrefetchPredictor* predictor)
@@ skipping 252 matching lines @@
 
     RedirectData microsoft = CreateRedirectData("microsoft.com", 10);
     InitializeRedirectStat(microsoft.add_redirect_endpoints(),
                            "www.microsoft.com", 10, 0, 0);
 
     test_host_redirect_data_.clear();
     test_host_redirect_data_.insert(std::make_pair(bbc.primary_key(), bbc));
     test_host_redirect_data_.insert(
         std::make_pair(microsoft.primary_key(), microsoft));
   }
+
+  {  // Manifest data.
+    precache::PrecacheManifest google = CreateManifestData(11);
+    InitializePrecacheResource(google.add_resource(),
+                               "http://google.com/script.js", 0.5);
+    InitializePrecacheResource(google.add_resource(),
+                               "http://static.google.com/style.css", 0.333);
+
+    precache::PrecacheManifest facebook = CreateManifestData(12);
+    InitializePrecacheResource(facebook.add_resource(),
+                               "http://fb.com/static.css", 0.99);
+
+    test_manifest_data_.insert(std::make_pair("google.com", google));
+    test_manifest_data_.insert(std::make_pair("facebook.com", facebook));
+  }
 }
 
 void ResourcePrefetchPredictorTest::TestRedirectStatusHistogram(
     const std::string& predictor_initial_key,
     const std::string& predictor_key,
     const std::string& navigation_initial_url,
     const std::string& navigation_url,
     ResourcePrefetchPredictor::RedirectStatus expected_status) {
   // Database initialization.
   const std::string& script_url = "https://cdn.google.com/script.js";
@@ skipping 50 matching lines @@
 static_assert(static_cast<int>(net::MAXIMUM_PRIORITY) ==
               static_cast<int>(ResourceData_Priority_REQUEST_PRIORITY_HIGHEST),
               "Database/Net priority mismatch: Maximum");
 
 // Tests that the predictor initializes correctly without any data.
 TEST_F(ResourcePrefetchPredictorTest, LazilyInitializeEmpty) {
   EXPECT_TRUE(predictor_->url_table_cache_->empty());
   EXPECT_TRUE(predictor_->host_table_cache_->empty());
   EXPECT_TRUE(predictor_->url_redirect_table_cache_->empty());
   EXPECT_TRUE(predictor_->host_redirect_table_cache_->empty());
+  EXPECT_TRUE(predictor_->manifest_table_cache_->empty());
 }
 
 // Tests that the history and the db tables data are loaded correctly.
 TEST_F(ResourcePrefetchPredictorTest, LazilyInitializeWithData) {
   AddUrlToHistory("http://www.google.com/", 4);
   AddUrlToHistory("http://www.yahoo.com/", 2);
 
   EXPECT_CALL(*mock_tables_.get(),
               GetAllData(Pointee(ContainerEq(PrefetchDataMap())),
                          Pointee(ContainerEq(PrefetchDataMap())),
@@ skipping 449 matching lines @@
   InitializeRedirectStat(host_redirect_data.add_redirect_endpoints(),
                          "facebook.com", 1, 0, 0);
   EXPECT_CALL(*mock_tables_.get(),
               UpdateData(empty_resource_data_, empty_resource_data_,
                          empty_redirect_data_, host_redirect_data));
 
   predictor_->RecordMainFrameLoadComplete(fb_end);
   profile_->BlockUntilHistoryProcessesPendingRequests();
 }
 
+TEST_F(ResourcePrefetchPredictorTest, ManifestHostNotInDB) {
+  precache::PrecacheManifest manifest = CreateManifestData(1);
+  InitializePrecacheResource(manifest.add_resource(),
+                             "http://cdn.google.com/script.js", 0.9);
+  InitializePrecacheResource(manifest.add_resource(),
+                             "http://cdn.google.com/style.css", 0.75);
+
+  EXPECT_CALL(*mock_tables_.get(), UpdateManifestData("google.com", manifest));
+
+  predictor_->OnManifestFetched("google.com", manifest);
+}
+
+TEST_F(ResourcePrefetchPredictorTest, ManifestHostInDB) {
+  EXPECT_CALL(*mock_tables_.get(),
+              GetAllData(Pointee(ContainerEq(PrefetchDataMap())),
+                         Pointee(ContainerEq(PrefetchDataMap())),
+                         Pointee(ContainerEq(RedirectDataMap())),
+                         Pointee(ContainerEq(RedirectDataMap())),
+                         Pointee(ContainerEq(ManifestDataMap()))))
+      .WillOnce(SetArgPointee<4>(test_manifest_data_));
+  ResetPredictor();
+  InitializePredictor();
+  EXPECT_EQ(2U, predictor_->manifest_table_cache_->size());
+
+  precache::PrecacheManifest manifest = CreateManifestData(1);
+  InitializePrecacheResource(manifest.add_resource(),
+                             "http://google.com/image.jpg", 0.1);
+
+  EXPECT_CALL(*mock_tables_.get(), UpdateManifestData("google.com", manifest));
+
+  predictor_->OnManifestFetched("google.com", manifest);
+}
+
+TEST_F(ResourcePrefetchPredictorTest, ManifestHostNotInDBAndDBFull) {
+  EXPECT_CALL(*mock_tables_.get(),
+              GetAllData(Pointee(ContainerEq(PrefetchDataMap())),
+                         Pointee(ContainerEq(PrefetchDataMap())),
+                         Pointee(ContainerEq(RedirectDataMap())),
+                         Pointee(ContainerEq(RedirectDataMap())),
+                         Pointee(ContainerEq(ManifestDataMap()))))
+      .WillOnce(SetArgPointee<4>(test_manifest_data_));
+  ResetPredictor();
+  InitializePredictor();
+  EXPECT_EQ(2U, predictor_->manifest_table_cache_->size());
+
+  precache::PrecacheManifest manifest = CreateManifestData(1);
+  InitializePrecacheResource(manifest.add_resource(),
+                             "http://en.wikipedia.org/logo.png", 1.0);
+
+  EXPECT_CALL(*mock_tables_.get(),
+              DeleteManifestData(std::vector<std::string>({"google.com"})));
+
+  EXPECT_CALL(*mock_tables_.get(),
+              UpdateManifestData("en.wikipedia.org", manifest));
+
+  predictor_->OnManifestFetched("en.wikipedia.org", manifest);
+}
+
 TEST_F(ResourcePrefetchPredictorTest, DeleteUrls) {
   // Add some dummy entries to cache.
   predictor_->url_table_cache_->insert(
       std::make_pair("http://www.google.com/page1.html",
                      CreatePrefetchData("http://www.google.com/page1.html")));
   predictor_->url_table_cache_->insert(
       std::make_pair("http://www.google.com/page2.html",
                      CreatePrefetchData("http://www.google.com/page2.html")));
   predictor_->url_table_cache_->insert(std::make_pair(
       "http://www.yahoo.com/", CreatePrefetchData("http://www.yahoo.com/")));
@@ skipping 20 matching lines @@
   predictor_->url_redirect_table_cache_->insert(
       std::make_pair("http://nyt.com/", CreateRedirectData("http://nyt.com/")));
 
   predictor_->host_redirect_table_cache_->insert(
       std::make_pair("www.google.com", CreateRedirectData("www.google.com")));
   predictor_->host_redirect_table_cache_->insert(
       std::make_pair("www.nike.com", CreateRedirectData("www.nike.com")));
   predictor_->host_redirect_table_cache_->insert(std::make_pair(
       "www.wikipedia.org", CreateRedirectData("www.wikipedia.org")));
 
+  predictor_->manifest_table_cache_->insert(
+      std::make_pair("google.com", CreateManifestData()));
+  predictor_->manifest_table_cache_->insert(
+      std::make_pair("apple.com", CreateManifestData()));
+  predictor_->manifest_table_cache_->insert(
+      std::make_pair("en.wikipedia.org", CreateManifestData()));
+
   history::URLRows rows;
   rows.push_back(history::URLRow(GURL("http://www.google.com/page2.html")));
   rows.push_back(history::URLRow(GURL("http://www.apple.com")));
   rows.push_back(history::URLRow(GURL("http://www.nike.com")));
 
   std::vector<std::string> urls_to_delete, hosts_to_delete,
-      url_redirects_to_delete, host_redirects_to_delete;
+      url_redirects_to_delete, host_redirects_to_delete, manifests_to_delete;
   urls_to_delete.push_back("http://www.google.com/page2.html");
   urls_to_delete.push_back("http://www.apple.com/");
   urls_to_delete.push_back("http://www.nike.com/");
   hosts_to_delete.push_back("www.google.com");
   hosts_to_delete.push_back("www.apple.com");
   url_redirects_to_delete.push_back("http://www.google.com/page2.html");
   url_redirects_to_delete.push_back("http://www.apple.com/");
   host_redirects_to_delete.push_back("www.google.com");
   host_redirects_to_delete.push_back("www.nike.com");
+  manifests_to_delete.push_back("google.com");
+  manifests_to_delete.push_back("apple.com");
 
   EXPECT_CALL(*mock_tables_.get(),
               DeleteResourceData(ContainerEq(urls_to_delete),
                                  ContainerEq(hosts_to_delete)));
   EXPECT_CALL(*mock_tables_.get(),
               DeleteRedirectData(ContainerEq(url_redirects_to_delete),
                                  ContainerEq(host_redirects_to_delete)));
+  EXPECT_CALL(*mock_tables_.get(),
+              DeleteManifestData(ContainerEq(manifests_to_delete)));
 
   predictor_->DeleteUrls(rows);
   EXPECT_EQ(2U, predictor_->url_table_cache_->size());
   EXPECT_EQ(1U, predictor_->host_table_cache_->size());
   EXPECT_EQ(2U, predictor_->url_redirect_table_cache_->size());
   EXPECT_EQ(1U, predictor_->host_redirect_table_cache_->size());
+  EXPECT_EQ(1U, predictor_->manifest_table_cache_->size());
 
   EXPECT_CALL(*mock_tables_.get(), DeleteAllData());
 
   predictor_->DeleteAllUrls();
   EXPECT_TRUE(predictor_->url_table_cache_->empty());
   EXPECT_TRUE(predictor_->host_table_cache_->empty());
   EXPECT_TRUE(predictor_->url_redirect_table_cache_->empty());
   EXPECT_TRUE(predictor_->host_redirect_table_cache_->empty());
+  EXPECT_TRUE(predictor_->manifest_table_cache_->empty());
 }
 
 TEST_F(ResourcePrefetchPredictorTest, OnMainFrameRequest) {
   URLRequestSummary summary1 = CreateURLRequestSummary(
       1, "http://www.google.com", "http://www.google.com",
       content::RESOURCE_TYPE_MAIN_FRAME, net::MEDIUM, std::string(), false);
   URLRequestSummary summary2 = CreateURLRequestSummary(
       2, "http://www.google.com", "http://www.google.com",
       content::RESOURCE_TYPE_MAIN_FRAME, net::MEDIUM, std::string(), false);
   URLRequestSummary summary3 = CreateURLRequestSummary(
@@ skipping 690 matching lines @@
   predictor_->host_table_cache_->insert(
       std::make_pair(google.primary_key(), google));
 
   predictor_->StartPrefetching(GURL(main_frame_url), PrefetchOrigin::EXTERNAL);
   predictor_->StopPrefetching(GURL(main_frame_url));
   histogram_tester_->ExpectTotalCount(
       internal::kResourcePrefetchPredictorPrefetchingDurationHistogram, 1);
 }
 
 }  // namespace predictors