Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(332)

Issues Search
    My Issues | Starred     Open | Closed | All

Unified Diff: chrome/browser/predictors/resource_prefetch_predictor_unittest.cc

Issue 10416002: Seculative resource prefetching for URLs CL. (Closed) Base URL: http://git.chromium.org/chromium/src.git@master
Patch Set: Addressing comments, adding test. Created 8 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« chrome/browser/predictors/resource_prefetch_predictor.cc ('K') | « chrome/browser/predictors/resource_prefetch_predictor_tables_unittest.cc ('k') | chrome/browser/profiles/profile_dependency_manager.cc » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: chrome/browser/predictors/resource_prefetch_predictor_unittest.cc
diff --git a/chrome/browser/predictors/resource_prefetch_predictor_unittest.cc b/chrome/browser/predictors/resource_prefetch_predictor_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..2c48231bc7eaa8f822247dd4bacc27653ba23292
--- /dev/null
+++ b/chrome/browser/predictors/resource_prefetch_predictor_unittest.cc
@@ -0,0 +1,600 @@
+// Copyright (c) 2012 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 "base/memory/ref_counted.h"
+#include "base/message_loop.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/time.h"
+#include "chrome/browser/history/history.h"
+#include "chrome/browser/history/history_types.h"
+#include "chrome/browser/history/in_memory_database.h"
+#include "chrome/browser/history/url_database.h"
+#include "chrome/browser/predictors/resource_prefetch_predictor.h"
+#include "chrome/browser/predictors/resource_prefetch_predictor_tables.h"
+#include "chrome/test/base/testing_profile.h"
+#include "content/public/test/test_browser_thread.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "testing/gmock/include/gmock/gmock.h"
+
+using testing::ContainerEq;
+using testing::Pointee;
+using testing::SetArgPointee;
+using testing::StrictMock;
+
+namespace predictors {
+
+typedef ResourcePrefetchPredictor::URLRequestSummary URLRequestSummary;
+typedef ResourcePrefetchPredictorTables::UrlTableRow UrlTableRow;
+typedef ResourcePrefetchPredictorTables::UrlTableRows UrlTableRows;
+
+class MockResourcePrefetchPredictorTables
+ : public ResourcePrefetchPredictorTables {
+ public:
+ MockResourcePrefetchPredictorTables() { }
+ ~MockResourcePrefetchPredictorTables() { }
+
+ MOCK_METHOD1(GetAllRows, void(std::vector<UrlTableRow>* url_row_buffer));
+ MOCK_METHOD2(UpdateRowsForUrl, void(const GURL& main_page_url,
+ const UrlTableRows& row_buffer));
+ MOCK_METHOD1(DeleteRowsForUrls, void(const std::vector<GURL>& urls));
+ MOCK_METHOD0(DeleteAllRows, void());
+};
+
+class ResourcePrefetchPredictorTest : public testing::Test {
+ public:
+ ResourcePrefetchPredictorTest();
+ ~ResourcePrefetchPredictorTest();
+ void SetUp() OVERRIDE;
+ void TearDown() OVERRIDE;
+
+ protected:
+ void AddUrlToHistory(const std::string& url,
+ int visit_count,
+ time_t last_visit) {
+ history::URLRow row = history::URLRow(GURL(url));
+ row.set_visit_count(visit_count);
+ row.set_last_visit(base::Time::FromTimeT(last_visit));
+ url_db_->AddURL(row);
+ }
+
+ NavigationID CreateNavigationID(int process_id,
+ int render_view_id,
+ const std::string& main_frame_url) {
+ NavigationID navigation_id;
+ navigation_id.render_process_id = process_id;
+ navigation_id.render_view_id = render_view_id;
+ navigation_id.main_frame_url = GURL(main_frame_url);
+ navigation_id.creation_time = base::TimeTicks::Now();
+ return navigation_id;
+ }
+
+ ResourcePrefetchPredictor::URLRequestSummary CreateURLRequestSummary(
+ int process_id,
+ int render_view_id,
+ const std::string& main_frame_url,
+ const std::string& resource_url,
+ ResourceType::Type resource_type,
+ const std::string& mime_type,
+ bool was_cached) {
+ ResourcePrefetchPredictor::URLRequestSummary summary;
+ summary.navigation_id = CreateNavigationID(process_id, render_view_id,
+ main_frame_url);
+ summary.resource_url = GURL(resource_url);
+ summary.resource_type = resource_type;
+ summary.mime_type = mime_type;
+ summary.was_cached = was_cached;
+ return summary;
+ }
+
+ bool URLRequestSummaryAreEqual(const URLRequestSummary& lhs,
+ const URLRequestSummary& rhs) {
+ return lhs.navigation_id == rhs.navigation_id &&
+ lhs.resource_url == rhs.resource_url &&
+ lhs.resource_type == rhs.resource_type &&
+ lhs.mime_type == rhs.mime_type &&
+ lhs.was_cached == rhs.was_cached;
+ }
+
+ void ResetPredictor() {
+ ResourcePrefetchPredictor::Config config;
+ config.max_urls_to_track = 3;
+ config.min_url_visit_count = 2;
+ config.max_resources_per_entry = 4;
+ config.max_consecutive_misses = 2;
+ predictor_.reset(new ResourcePrefetchPredictor(config, &profile_));
+ predictor_->SetTablesForTesting(mock_tables_);
+ }
+
+ MessageLoop loop_;
+ content::TestBrowserThread ui_thread_;
+ content::TestBrowserThread db_thread_;
+ TestingProfile profile_;
+
+ scoped_ptr<ResourcePrefetchPredictor> predictor_;
+ scoped_refptr<StrictMock<MockResourcePrefetchPredictorTables> >
+ mock_tables_;
+ UrlTableRows test_url_rows_;
+ history::URLDatabase* url_db_;
+};
+
+ResourcePrefetchPredictorTest::ResourcePrefetchPredictorTest()
+ : loop_(MessageLoop::TYPE_DEFAULT),
+ ui_thread_(content::BrowserThread::UI, &loop_),
+ db_thread_(content::BrowserThread::DB, &loop_),
+ predictor_(NULL),
+ mock_tables_(new StrictMock<MockResourcePrefetchPredictorTables>()) {
+ test_url_rows_.push_back(UrlTableRow(
+ "http://www.google.com",
+ "http://google.com/style1.css",
+ ResourceType::STYLESHEET,
+ 3, 2, 1, 1.0));
+ test_url_rows_.push_back(UrlTableRow(
+ "http://www.google.com",
+ "http://google.com/script3.js",
+ ResourceType::SCRIPT,
+ 4, 0, 1, 2.1));
+ test_url_rows_.push_back(UrlTableRow(
+ "http://www.google.com",
+ "http://google.com/script4.js",
+ ResourceType::SCRIPT,
+ 11, 0, 0, 2.1));
+ test_url_rows_.push_back(UrlTableRow(
+ "http://www.google.com",
+ "http://google.com/image1.png",
+ ResourceType::IMAGE,
+ 6, 3, 0, 2.2));
+ test_url_rows_.push_back(UrlTableRow(
+ "http://www.google.com",
+ "http://google.com/a.font",
+ ResourceType::LAST_TYPE,
+ 2, 0, 0, 5.1));
+
+ test_url_rows_.push_back(UrlTableRow(
+ "http://www.reddit.com",
+ "http://reddit-resource.com/script1.js",
+ ResourceType::SCRIPT,
+ 4, 0, 1, 1.0));
+ test_url_rows_.push_back(UrlTableRow(
+ "http://www.reddit.com",
+ "http://reddit-resource.com/script2.js",
+ ResourceType::SCRIPT,
+ 2, 0, 0, 2.1));
+
+ test_url_rows_.push_back(UrlTableRow(
+ "http://www.yahoo.com",
+ "http://google.com/image.png",
+ ResourceType::IMAGE,
+ 20, 1, 0, 10.0));
+}
+
+ResourcePrefetchPredictorTest::~ResourcePrefetchPredictorTest() {
+}
+
+void ResourcePrefetchPredictorTest::SetUp() {
+ profile_.CreateHistoryService(true, false);
+ profile_.BlockUntilHistoryProcessesPendingRequests();
+ EXPECT_TRUE(profile_.GetHistoryService(Profile::EXPLICIT_ACCESS));
+ url_db_ = profile_.GetHistoryService(
+ Profile::EXPLICIT_ACCESS)->InMemoryDatabase();
+
+ // Initialize the predictor with empty data.
+ ResetPredictor();
+ EXPECT_EQ(predictor_->initialization_state_,
+ ResourcePrefetchPredictor::NOT_INITIALIZED);
+ EXPECT_CALL(*mock_tables_, GetAllRows(Pointee(ContainerEq(UrlTableRows()))));
+ predictor_->LazilyInitialize();
+ profile_.BlockUntilHistoryProcessesPendingRequests();
+ EXPECT_TRUE(predictor_->inflight_navigations_.empty());
+ EXPECT_EQ(predictor_->initialization_state_,
+ ResourcePrefetchPredictor::INITIALIZED);
+}
+
+void ResourcePrefetchPredictorTest::TearDown() {
+ predictor_.reset(NULL);
+ profile_.DestroyHistoryService();
+}
+
+TEST_F(ResourcePrefetchPredictorTest, LazilyInitializeEmpty) {
+ // Tests that the predictor initializes correctly without any data.
+ EXPECT_TRUE(predictor_->url_table_cache_.empty());
+}
+
+TEST_F(ResourcePrefetchPredictorTest, LazilyInitializeWithData) {
+ // Tests that the history and the db tables data are loaded correctly.
+ AddUrlToHistory("http://www.google.com", 4, 12345);
+ AddUrlToHistory("http://www.yahoo.com", 2, 12346);
+
+ EXPECT_CALL(*mock_tables_, GetAllRows(Pointee(ContainerEq(UrlTableRows()))))
+ .WillOnce(SetArgPointee<0>(test_url_rows_));
+
+ std::vector<GURL> urls_to_delete;
+ urls_to_delete.push_back(GURL("http://www.reddit.com"));
+ EXPECT_CALL(*mock_tables_, DeleteRowsForUrls(ContainerEq(urls_to_delete)));
+
+ ResetPredictor();
+ predictor_->LazilyInitialize();
+ loop_.RunAllPending();
+
+ // Test that the internal variables correctly initialized.
+ EXPECT_EQ(predictor_->initialization_state_,
+ ResourcePrefetchPredictor::INITIALIZED);
+ EXPECT_TRUE(predictor_->inflight_navigations_.empty());
+ EXPECT_EQ(2, static_cast<int>(predictor_->url_table_cache_.size()));
+
+ UrlTableRows google_rows(test_url_rows_.begin(),
+ test_url_rows_.begin() + 5);
+ std::sort(google_rows.begin(), google_rows.end(),
+ ResourcePrefetchPredictorTables::UrlTableRowSorter());
+ EXPECT_EQ(predictor_->url_table_cache_[GURL("http://www.google.com")].rows,
+ google_rows);
+ EXPECT_EQ(
+ predictor_->url_table_cache_[GURL("http://www.google.com")].last_visit,
+ base::Time::FromTimeT(12345));
+
+ UrlTableRows yahoo_rows(test_url_rows_.begin() + 7,
+ test_url_rows_.begin() + 8);
+ EXPECT_EQ(predictor_->url_table_cache_[GURL("http://www.yahoo.com")].rows,
+ yahoo_rows);
+ EXPECT_EQ(
+ predictor_->url_table_cache_[GURL("http://www.yahoo.com")].last_visit,
+ base::Time::FromTimeT(12346));
+}
+
+TEST_F(ResourcePrefetchPredictorTest, NavigationNotRecorded) {
+ // Single navigation but history count is low, so should not record.
+ AddUrlToHistory("http://www.google.com", 1, 12345);
+
+ URLRequestSummary main_frame = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://www.google.com",
+ ResourceType::MAIN_FRAME, "", false);
+ predictor_->RecordURLRequest(main_frame);
+ EXPECT_EQ(1, static_cast<int>(predictor_->inflight_navigations_.size()));
+
+ // Now add a few subresources.
+ URLRequestSummary resource1 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://google.com/style1.css",
+ ResourceType::STYLESHEET, "text/css", false);
+ predictor_->RecordUrlResponse(resource1);
+ URLRequestSummary resource2 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://google.com/script1.js",
+ ResourceType::SCRIPT, "text/javascript", false);
+ predictor_->RecordUrlResponse(resource2);
+ URLRequestSummary resource3 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://google.com/script2.js",
+ ResourceType::SCRIPT, "text/javascript", false);
+ predictor_->RecordUrlResponse(resource3);
+ predictor_->OnNavigationComplete(main_frame.navigation_id);
+}
+
+TEST_F(ResourcePrefetchPredictorTest, NavigationUrlNotInDB) {
+ // Single navigation that will be recorded. Will check for duplicate
+ // resources and also for number of resources saved.
+ AddUrlToHistory("http://www.google.com", 4, 12345);
+
+ URLRequestSummary main_frame = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://www.google.com",
+ ResourceType::MAIN_FRAME, "", false);
+ predictor_->RecordURLRequest(main_frame);
+ EXPECT_EQ(1, static_cast<int>(predictor_->inflight_navigations_.size()));
+
+ URLRequestSummary resource1 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://google.com/style1.css",
+ ResourceType::STYLESHEET, "text/css", false);
+ predictor_->RecordUrlResponse(resource1);
+ URLRequestSummary resource2 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://google.com/script1.js",
+ ResourceType::SCRIPT, "text/javascript", false);
+ predictor_->RecordUrlResponse(resource2);
+ URLRequestSummary resource3 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://google.com/script2.js",
+ ResourceType::SCRIPT, "text/javascript", false);
+ predictor_->RecordUrlResponse(resource3);
+ URLRequestSummary resource4 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://google.com/script1.js",
+ ResourceType::SCRIPT, "text/javascript", true);
+ predictor_->RecordUrlResponse(resource4);
+ URLRequestSummary resource5 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://google.com/image1.png",
+ ResourceType::IMAGE, "image/png", false);
+ predictor_->RecordUrlResponse(resource5);
+ URLRequestSummary resource6 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://google.com/image2.png",
+ ResourceType::IMAGE, "image/png", false);
+ predictor_->RecordUrlResponse(resource6);
+ URLRequestSummary resource7 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://google.com/style2.css",
+ ResourceType::STYLESHEET, "text/css", false);
+ predictor_->OnSubresourceLoadedFromMemory(
+ resource7.navigation_id,
+ resource7.resource_url,
+ resource7.mime_type,
+ resource7.resource_type);
+
+ UrlTableRows db_rows;
+ db_rows.push_back(UrlTableRow(
+ "http://www.google.com", "http://google.com/style1.css",
+ ResourceType::STYLESHEET, 1, 0, 0, 1.0));
+ db_rows.push_back(UrlTableRow(
+ "http://www.google.com", "http://google.com/script1.js",
+ ResourceType::SCRIPT, 1, 0, 0, 2.0));
+ db_rows.push_back(UrlTableRow(
+ "http://www.google.com", "http://google.com/script2.js",
+ ResourceType::SCRIPT, 1, 0, 0, 3.0));
+ db_rows.push_back(UrlTableRow(
+ "http://www.google.com", "http://google.com/style2.css",
+ ResourceType::STYLESHEET, 1, 0, 0, 7.0));
+ EXPECT_CALL(*mock_tables_,
+ UpdateRowsForUrl(GURL("http://www.google.com"),
+ ContainerEq(db_rows)));
+
+ predictor_->OnNavigationComplete(main_frame.navigation_id);
+}
+
+TEST_F(ResourcePrefetchPredictorTest, NavigationUrlInDB) {
+ // Tests that navigation is recoreded correctly for URL already present in
+ // the database cache.
+ AddUrlToHistory("http://www.google.com", 4, 12345);
+
+ EXPECT_CALL(*mock_tables_, GetAllRows(Pointee(ContainerEq(UrlTableRows()))))
+ .WillOnce(SetArgPointee<0>(test_url_rows_));
+
+ std::vector<GURL> urls_to_delete;
+ urls_to_delete.push_back(GURL("http://www.reddit.com"));
+ urls_to_delete.push_back(GURL("http://www.yahoo.com"));
+ EXPECT_CALL(*mock_tables_, DeleteRowsForUrls(ContainerEq(urls_to_delete)));
+
+ ResetPredictor();
+ predictor_->LazilyInitialize();
+ loop_.RunAllPending();
+
+ URLRequestSummary main_frame = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://www.google.com",
+ ResourceType::MAIN_FRAME, "", false);
+ predictor_->RecordURLRequest(main_frame);
+ EXPECT_EQ(1, static_cast<int>(predictor_->inflight_navigations_.size()));
+
+ URLRequestSummary resource1 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://google.com/style1.css",
+ ResourceType::STYLESHEET, "text/css", false);
+ predictor_->RecordUrlResponse(resource1);
+ URLRequestSummary resource2 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://google.com/script1.js",
+ ResourceType::SCRIPT, "text/javascript", false);
+ predictor_->RecordUrlResponse(resource2);
+ URLRequestSummary resource3 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://google.com/script2.js",
+ ResourceType::SCRIPT, "text/javascript", false);
+ predictor_->RecordUrlResponse(resource3);
+ URLRequestSummary resource4 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://google.com/script1.js",
+ ResourceType::SCRIPT, "text/javascript", true);
+ predictor_->RecordUrlResponse(resource4);
+ URLRequestSummary resource5 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://google.com/image1.png",
+ ResourceType::IMAGE, "image/png", false);
+ predictor_->RecordUrlResponse(resource5);
+ URLRequestSummary resource6 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://google.com/image2.png",
+ ResourceType::IMAGE, "image/png", false);
+ predictor_->RecordUrlResponse(resource6);
+ URLRequestSummary resource7 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://google.com/style2.css",
+ ResourceType::STYLESHEET, "text/css", false);
+ predictor_->OnSubresourceLoadedFromMemory(
+ resource7.navigation_id,
+ resource7.resource_url,
+ resource7.mime_type,
+ resource7.resource_type);
+
+ UrlTableRows db_rows;
+ db_rows.push_back(UrlTableRow(
+ "http://www.google.com", "http://google.com/style1.css",
+ ResourceType::STYLESHEET, 4, 2, 0, 1.0));
+ db_rows.push_back(UrlTableRow(
+ "http://www.google.com", "http://google.com/script1.js",
+ ResourceType::SCRIPT, 1, 0, 0, 2.0));
+ db_rows.push_back(UrlTableRow(
+ "http://www.google.com", "http://google.com/script4.js",
+ ResourceType::SCRIPT, 11, 1, 1, 2.1));
+ db_rows.push_back(UrlTableRow(
+ "http://www.google.com", "http://google.com/script2.js",
+ ResourceType::SCRIPT, 1, 0, 0, 3.0));
+ EXPECT_CALL(*mock_tables_,
+ UpdateRowsForUrl(GURL("http://www.google.com"),
+ ContainerEq(db_rows)));
+
+ predictor_->OnNavigationComplete(main_frame.navigation_id);
+}
+
+TEST_F(ResourcePrefetchPredictorTest, NavigationUrlNotInDBAndDBFull) {
+ // Tests that a URL is deleted before another is added if the cache is full.
+ AddUrlToHistory("http://www.google.com", 4, 12345);
+ AddUrlToHistory("http://www.reddit.com", 4, 12340); // Should be deleted.
+ AddUrlToHistory("http://www.yahoo.com", 4, 12350);
+ AddUrlToHistory("http://www.nike.com", 4, 10000);
+
+ EXPECT_CALL(*mock_tables_, GetAllRows(Pointee(ContainerEq(UrlTableRows()))))
+ .WillOnce(SetArgPointee<0>(test_url_rows_));
+
+ ResetPredictor();
+ predictor_->LazilyInitialize();
+ loop_.RunAllPending();
+ EXPECT_EQ(3, static_cast<int>(predictor_->url_table_cache_.size()));
+
+ URLRequestSummary main_frame = CreateURLRequestSummary(
+ 1, 1, "http://www.nike.com", "http://www.nike.com",
+ ResourceType::MAIN_FRAME, "", false);
+ predictor_->RecordURLRequest(main_frame);
+ EXPECT_EQ(1, static_cast<int>(predictor_->inflight_navigations_.size()));
+
+ URLRequestSummary resource1 = CreateURLRequestSummary(
+ 1, 1, "http://www.nike.com", "http://nike.com/style1.css",
+ ResourceType::STYLESHEET, "text/css", false);
+ predictor_->RecordUrlResponse(resource1);
+ URLRequestSummary resource2 = CreateURLRequestSummary(
+ 1, 1, "http://www.nike.com", "http://nike.com/image2.png",
+ ResourceType::IMAGE, "image/png", false);
+ predictor_->RecordUrlResponse(resource2);
+
+ std::vector<GURL> urls_to_delete;
+ urls_to_delete.push_back(GURL("http://www.reddit.com"));
+ EXPECT_CALL(*mock_tables_, DeleteRowsForUrls(ContainerEq(urls_to_delete)));
+
+ UrlTableRows db_rows;
+ db_rows.push_back(UrlTableRow(
+ "http://www.nike.com", "http://nike.com/style1.css",
+ ResourceType::STYLESHEET, 1, 0, 0, 1.0));
+ db_rows.push_back(UrlTableRow(
+ "http://www.nike.com", "http://nike.com/image2.png",
+ ResourceType::IMAGE, 1, 0, 0, 2.0));
+ EXPECT_CALL(*mock_tables_,
+ UpdateRowsForUrl(GURL("http://www.nike.com"),
+ ContainerEq(db_rows)));
+
+ predictor_->OnNavigationComplete(main_frame.navigation_id);
+}
+
+TEST_F(ResourcePrefetchPredictorTest, DeleteUrls) {
+ // Add some dummy entries to cache.
+ predictor_->url_table_cache_[GURL("http://www.google.com")];
+ predictor_->url_table_cache_[GURL("http://www.yahoo.com")];
+ predictor_->url_table_cache_[GURL("http://www.apple.com")];
+ predictor_->url_table_cache_[GURL("http://www.nike.com")];
+ predictor_->url_table_cache_[GURL("http://www.reddit.com")];
+
+ history::URLRows rows;
+ rows.push_back(history::URLRow(GURL("http://www.apple.com")));
+ rows.push_back(history::URLRow(GURL("http://www.yahoo.com")));
+
+ std::vector<GURL> urls_to_delete;
+ urls_to_delete.push_back(GURL("http://www.apple.com"));
+ urls_to_delete.push_back(GURL("http://www.yahoo.com"));
+ EXPECT_CALL(*mock_tables_, DeleteRowsForUrls(ContainerEq(urls_to_delete)));
+ predictor_->DeleteUrls(rows);
+ EXPECT_EQ(3, static_cast<int>(predictor_->url_table_cache_.size()));
+
+ EXPECT_CALL(*mock_tables_, DeleteAllRows());
+
+ predictor_->DeleteAllUrls();
+ EXPECT_TRUE(predictor_->url_table_cache_.empty());
+}
+
+TEST_F(ResourcePrefetchPredictorTest, OnMainFrameRequest) {
+ URLRequestSummary summary1 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://www.google.com",
+ ResourceType::MAIN_FRAME, "", false);
+ URLRequestSummary summary2 = CreateURLRequestSummary(
+ 1, 2, "http://www.google.com", "http://www.google.com",
+ ResourceType::MAIN_FRAME, "", false);
+ URLRequestSummary summary3 = CreateURLRequestSummary(
+ 2, 1, "http://www.yahoo.com", "http://www.yahoo.com",
+ ResourceType::MAIN_FRAME, "", false);
+
+ predictor_->OnMainFrameRequest(summary1);
+ EXPECT_EQ(1, static_cast<int>(predictor_->inflight_navigations_.size()));
+ predictor_->OnMainFrameRequest(summary2);
+ EXPECT_EQ(2, static_cast<int>(predictor_->inflight_navigations_.size()));
+ predictor_->OnMainFrameRequest(summary3);
+ EXPECT_EQ(3, static_cast<int>(predictor_->inflight_navigations_.size()));
+
+ // Insert anther with same navigation id. It should replace.
+ URLRequestSummary summary4 = CreateURLRequestSummary(
+ 1, 1, "http://www.nike.com", "http://www.nike.com",
+ ResourceType::MAIN_FRAME, "", false);
+ URLRequestSummary summary5 = CreateURLRequestSummary(
+ 1, 2, "http://www.google.com", "http://www.google.com",
+ ResourceType::MAIN_FRAME, "", false);
+
+ predictor_->OnMainFrameRequest(summary4);
+ EXPECT_EQ(3, static_cast<int>(predictor_->inflight_navigations_.size()));
+
+ // Change this creation time so that it will go away on the next insert.
+ summary5.navigation_id.creation_time = base::TimeTicks::Now() -
+ base::TimeDelta::FromDays(1);
+ predictor_->OnMainFrameRequest(summary5);
+ EXPECT_EQ(3, static_cast<int>(predictor_->inflight_navigations_.size()));
+
+ URLRequestSummary summary6 = CreateURLRequestSummary(
+ 3, 1, "http://www.shoes.com", "http://www.shoes.com",
+ ResourceType::MAIN_FRAME, "", false);
+ predictor_->OnMainFrameRequest(summary6);
+ EXPECT_EQ(3, static_cast<int>(predictor_->inflight_navigations_.size()));
+
+ EXPECT_NE(predictor_->inflight_navigations_.find(summary3.navigation_id),
+ predictor_->inflight_navigations_.end());
+ EXPECT_NE(predictor_->inflight_navigations_.find(summary4.navigation_id),
+ predictor_->inflight_navigations_.end());
+ EXPECT_NE(predictor_->inflight_navigations_.find(summary6.navigation_id),
+ predictor_->inflight_navigations_.end());
+}
+
+TEST_F(ResourcePrefetchPredictorTest, OnMainFrameRedirect) {
+ URLRequestSummary summary1 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://www.google.com",
+ ResourceType::MAIN_FRAME, "", false);
+ URLRequestSummary summary2 = CreateURLRequestSummary(
+ 1, 2, "http://www.google.com", "http://www.google.com",
+ ResourceType::MAIN_FRAME, "", false);
+ URLRequestSummary summary3 = CreateURLRequestSummary(
+ 2, 1, "http://www.yahoo.com", "http://www.yahoo.com",
+ ResourceType::MAIN_FRAME, "", false);
+
+ predictor_->OnMainFrameRedirect(summary1);
+ EXPECT_TRUE(predictor_->inflight_navigations_.empty());
+
+ predictor_->OnMainFrameRequest(summary1);
+ EXPECT_EQ(1, static_cast<int>(predictor_->inflight_navigations_.size()));
+ predictor_->OnMainFrameRequest(summary2);
+ EXPECT_EQ(2, static_cast<int>(predictor_->inflight_navigations_.size()));
+
+ predictor_->OnMainFrameRedirect(summary3);
+ EXPECT_EQ(2, static_cast<int>(predictor_->inflight_navigations_.size()));
+ predictor_->OnMainFrameRedirect(summary1);
+ EXPECT_EQ(1, static_cast<int>(predictor_->inflight_navigations_.size()));
+ predictor_->OnMainFrameRedirect(summary2);
+ EXPECT_TRUE(predictor_->inflight_navigations_.empty());
+}
+
+TEST_F(ResourcePrefetchPredictorTest, OnSubresourceResponse) {
+ // If there is no inflight navigation, nothing happens.
+ URLRequestSummary resource1 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://google.com/style1.css",
+ ResourceType::STYLESHEET, "text/css", false);
+ predictor_->OnSubresourceResponse(resource1);
+ EXPECT_TRUE(predictor_->inflight_navigations_.empty());
+
+ // Add an inflight navigation.
+ URLRequestSummary main_frame1 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://www.google.com",
+ ResourceType::MAIN_FRAME, "", false);
+ predictor_->OnMainFrameRequest(main_frame1);
+ EXPECT_EQ(1, static_cast<int>(predictor_->inflight_navigations_.size()));
+
+ // Now add a few subresources.
+ URLRequestSummary resource2 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://google.com/script1.js",
+ ResourceType::SCRIPT, "text/javascript", false);
+ URLRequestSummary resource3 = CreateURLRequestSummary(
+ 1, 1, "http://www.google.com", "http://google.com/script2.js",
+ ResourceType::SCRIPT, "text/javascript", false);
+ predictor_->OnSubresourceResponse(resource1);
+ predictor_->OnSubresourceResponse(resource2);
+ predictor_->OnSubresourceResponse(resource3);
+
+ EXPECT_EQ(1, static_cast<int>(predictor_->inflight_navigations_.size()));
+ EXPECT_EQ(3, static_cast<int>(
+ predictor_->inflight_navigations_[main_frame1.navigation_id].size()));
+ EXPECT_TRUE(URLRequestSummaryAreEqual(
+ resource1,
+ predictor_->inflight_navigations_[main_frame1.navigation_id][0]));
+ EXPECT_TRUE(URLRequestSummaryAreEqual(
+ resource2,
+ predictor_->inflight_navigations_[main_frame1.navigation_id][1]));
+ EXPECT_TRUE(URLRequestSummaryAreEqual(
+ resource3,
+ predictor_->inflight_navigations_[main_frame1.navigation_id][2]));
+}
+
+} // namespace predictors
« chrome/browser/predictors/resource_prefetch_predictor.cc ('K') | « chrome/browser/predictors/resource_prefetch_predictor_tables_unittest.cc ('k') | chrome/browser/profiles/profile_dependency_manager.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698