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

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

Issue 2804283002: predictors: Refactor UpdateData() function. (Closed)
Patch Set: Oops fix. Created 3 years, 8 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
« no previous file with comments | « chrome/browser/predictors/resource_prefetch_predictor_tables_unittest.cc ('k') | no next file » | 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
index 9f89ef8a03153c1caa0f41ddfab57c0fff33eea3..71ed0e38141286942d9d176447463852e9ac141e 100644
--- a/chrome/browser/predictors/resource_prefetch_predictor_unittest.cc
+++ b/chrome/browser/predictors/resource_prefetch_predictor_unittest.cc
@@ -137,11 +137,10 @@ class MockResourcePrefetchPredictorTables
RedirectDataMap* host_redirect_data_map,
ManifestDataMap* manifest_data_map,
OriginDataMap* origin_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(UpdateResourceData,
+ void(const PrefetchData& data, PrefetchKeyType key_type));
+ MOCK_METHOD2(UpdateRedirectData,
+ void(const RedirectData& data, PrefetchKeyType key_type));
MOCK_METHOD2(UpdateManifestData,
void(const std::string& host,
const precache::PrecacheManifest& manifest_data));
@@ -270,8 +269,6 @@ class ResourcePrefetchPredictorTest : public testing::Test {
RedirectDataMap test_host_redirect_data_;
ManifestDataMap test_manifest_data_;
OriginDataMap test_origin_data_;
- PrefetchData empty_resource_data_;
- RedirectData empty_redirect_data_;
MockURLRequestJobFactory url_request_job_factory_;
EmptyURLRequestDelegate url_request_delegate_;
@@ -282,9 +279,7 @@ class ResourcePrefetchPredictorTest : public testing::Test {
ResourcePrefetchPredictorTest::ResourcePrefetchPredictorTest()
: thread_bundle_(),
profile_(new TestingProfile()),
- mock_tables_(new StrictMock<MockResourcePrefetchPredictorTables>()),
- empty_resource_data_(),
- empty_redirect_data_() {}
+ mock_tables_(new StrictMock<MockResourcePrefetchPredictorTables>()) {}
ResourcePrefetchPredictorTest::~ResourcePrefetchPredictorTest() {
profile_.reset(NULL);
@@ -494,10 +489,9 @@ void ResourcePrefetchPredictorTest::TestRedirectStatusHistogram(
// Navigation simulation.
using testing::_;
- EXPECT_CALL(*mock_tables_.get(), UpdateData(_, _, _, _))
- .Times(testing::AtLeast(1));
+ EXPECT_CALL(*mock_tables_.get(), UpdateResourceData(_, _));
+ EXPECT_CALL(*mock_tables_.get(), UpdateRedirectData(_, _));
EXPECT_CALL(*mock_tables_.get(), UpdateOriginData(_));
-
URLRequestSummary initial =
CreateURLRequestSummary(1, navigation_initial_url);
predictor_->RecordURLRequest(initial);
@@ -628,8 +622,8 @@ TEST_F(ResourcePrefetchPredictorTest, NavigationNotRecorded) {
host_data.add_resources(), "https://google.com/script2.js",
content::RESOURCE_TYPE_SCRIPT, 1, 0, 0, 3.0, net::MEDIUM, false, false);
EXPECT_CALL(*mock_tables_.get(),
- UpdateData(empty_resource_data_, host_data, empty_redirect_data_,
- empty_redirect_data_));
+ UpdateResourceData(host_data, PREFETCH_KEY_TYPE_HOST));
+
OriginData origin_data = CreateOriginData("www.google.com");
InitializeOriginStat(origin_data.add_origins(), "https://google.com/", 1, 0,
0, 1., false, true);
@@ -639,8 +633,7 @@ TEST_F(ResourcePrefetchPredictorTest, NavigationNotRecorded) {
InitializeRedirectStat(host_redirect_data.add_redirect_endpoints(),
"www.google.com", 1, 0, 0);
EXPECT_CALL(*mock_tables_.get(),
- UpdateData(empty_resource_data_, empty_resource_data_,
- empty_redirect_data_, host_redirect_data));
+ UpdateRedirectData(host_redirect_data, PREFETCH_KEY_TYPE_HOST));
predictor_->RecordMainFrameLoadComplete(main_frame.navigation_id);
profile_->BlockUntilHistoryProcessesPendingRequests();
@@ -730,8 +723,7 @@ TEST_F(ResourcePrefetchPredictorTest, NavigationUrlNotInDB) {
content::RESOURCE_TYPE_STYLESHEET, 1, 0, 0, 7.0,
net::MEDIUM, false, false);
EXPECT_CALL(*mock_tables_.get(),
- UpdateData(url_data, empty_resource_data_, empty_redirect_data_,
- empty_redirect_data_));
+ UpdateResourceData(url_data, PREFETCH_KEY_TYPE_URL));
OriginData origin_data = CreateOriginData("www.google.com");
InitializeOriginStat(origin_data.add_origins(), "http://static.google.com/",
@@ -747,22 +739,19 @@ TEST_F(ResourcePrefetchPredictorTest, NavigationUrlNotInDB) {
PrefetchData host_data = CreatePrefetchData("www.google.com");
host_data.mutable_resources()->CopyFrom(url_data.resources());
EXPECT_CALL(*mock_tables_.get(),
- UpdateData(empty_resource_data_, host_data, empty_redirect_data_,
- empty_redirect_data_));
+ UpdateResourceData(host_data, PREFETCH_KEY_TYPE_HOST));
RedirectData url_redirect_data = CreateRedirectData("http://www.google.com/");
InitializeRedirectStat(url_redirect_data.add_redirect_endpoints(),
"http://www.google.com/", 1, 0, 0);
EXPECT_CALL(*mock_tables_.get(),
- UpdateData(empty_resource_data_, empty_resource_data_,
- url_redirect_data, empty_redirect_data_));
+ UpdateRedirectData(url_redirect_data, PREFETCH_KEY_TYPE_URL));
RedirectData host_redirect_data = CreateRedirectData("www.google.com");
InitializeRedirectStat(host_redirect_data.add_redirect_endpoints(),
"www.google.com", 1, 0, 0);
EXPECT_CALL(*mock_tables_.get(),
- UpdateData(empty_resource_data_, empty_resource_data_,
- empty_redirect_data_, host_redirect_data));
+ UpdateRedirectData(host_redirect_data, PREFETCH_KEY_TYPE_HOST));
predictor_->RecordMainFrameLoadComplete(main_frame.navigation_id);
profile_->BlockUntilHistoryProcessesPendingRequests();
@@ -853,8 +842,7 @@ TEST_F(ResourcePrefetchPredictorTest, NavigationUrlInDB) {
url_data.add_resources(), "http://google.com/script2.js",
content::RESOURCE_TYPE_SCRIPT, 1, 0, 0, 3.0, net::MEDIUM, false, false);
EXPECT_CALL(*mock_tables_.get(),
- UpdateData(url_data, empty_resource_data_, empty_redirect_data_,
- empty_redirect_data_));
+ UpdateResourceData(url_data, PREFETCH_KEY_TYPE_URL));
EXPECT_CALL(*mock_tables_.get(),
DeleteSingleResourceDataPoint("www.facebook.com",
PREFETCH_KEY_TYPE_HOST));
@@ -875,22 +863,19 @@ TEST_F(ResourcePrefetchPredictorTest, NavigationUrlInDB) {
content::RESOURCE_TYPE_STYLESHEET, 1, 0, 0, 7.0,
net::MEDIUM, false, false);
EXPECT_CALL(*mock_tables_.get(),
- UpdateData(empty_resource_data_, host_data, empty_redirect_data_,
- empty_redirect_data_));
+ UpdateResourceData(host_data, PREFETCH_KEY_TYPE_HOST));
RedirectData url_redirect_data = CreateRedirectData("http://www.google.com/");
InitializeRedirectStat(url_redirect_data.add_redirect_endpoints(),
"http://www.google.com/", 1, 0, 0);
EXPECT_CALL(*mock_tables_.get(),
- UpdateData(empty_resource_data_, empty_resource_data_,
- url_redirect_data, empty_redirect_data_));
+ UpdateRedirectData(url_redirect_data, PREFETCH_KEY_TYPE_URL));
RedirectData host_redirect_data = CreateRedirectData("www.google.com");
InitializeRedirectStat(host_redirect_data.add_redirect_endpoints(),
"www.google.com", 1, 0, 0);
EXPECT_CALL(*mock_tables_.get(),
- UpdateData(empty_resource_data_, empty_resource_data_,
- empty_redirect_data_, host_redirect_data));
+ UpdateRedirectData(host_redirect_data, PREFETCH_KEY_TYPE_HOST));
OriginData origin_data = CreateOriginData("www.google.com");
InitializeOriginStat(origin_data.add_origins(), "http://static.google.com/",
@@ -964,28 +949,24 @@ TEST_F(ResourcePrefetchPredictorTest, NavigationUrlNotInDBAndDBFull) {
content::RESOURCE_TYPE_IMAGE, 1, 0, 0, 2.0,
net::MEDIUM, false, false);
EXPECT_CALL(*mock_tables_.get(),
- UpdateData(url_data, empty_resource_data_, empty_redirect_data_,
- empty_redirect_data_));
+ UpdateResourceData(url_data, PREFETCH_KEY_TYPE_URL));
PrefetchData host_data = CreatePrefetchData("www.nike.com");
host_data.mutable_resources()->CopyFrom(url_data.resources());
EXPECT_CALL(*mock_tables_.get(),
- UpdateData(empty_resource_data_, host_data, empty_redirect_data_,
- empty_redirect_data_));
+ UpdateResourceData(host_data, PREFETCH_KEY_TYPE_HOST));
RedirectData url_redirect_data = CreateRedirectData("http://www.nike.com/");
InitializeRedirectStat(url_redirect_data.add_redirect_endpoints(),
"http://www.nike.com/", 1, 0, 0);
EXPECT_CALL(*mock_tables_.get(),
- UpdateData(empty_resource_data_, empty_resource_data_,
- url_redirect_data, empty_redirect_data_));
+ UpdateRedirectData(url_redirect_data, PREFETCH_KEY_TYPE_URL));
RedirectData host_redirect_data = CreateRedirectData("www.nike.com");
InitializeRedirectStat(host_redirect_data.add_redirect_endpoints(),
"www.nike.com", 1, 0, 0);
EXPECT_CALL(*mock_tables_.get(),
- UpdateData(empty_resource_data_, empty_resource_data_,
- empty_redirect_data_, host_redirect_data));
+ UpdateRedirectData(host_redirect_data, PREFETCH_KEY_TYPE_HOST));
EXPECT_CALL(*mock_tables_.get(), UpdateOriginData(testing::_));
@@ -1030,15 +1011,13 @@ TEST_F(ResourcePrefetchPredictorTest, RedirectUrlNotInDB) {
InitializeRedirectStat(url_redirect_data.add_redirect_endpoints(),
"https://facebook.com/google", 1, 0, 0);
EXPECT_CALL(*mock_tables_.get(),
- UpdateData(empty_resource_data_, empty_resource_data_,
- url_redirect_data, empty_redirect_data_));
+ UpdateRedirectData(url_redirect_data, PREFETCH_KEY_TYPE_URL));
RedirectData host_redirect_data = CreateRedirectData("fb.com");
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));
+ UpdateRedirectData(host_redirect_data, PREFETCH_KEY_TYPE_HOST));
predictor_->RecordMainFrameLoadComplete(fb_end);
profile_->BlockUntilHistoryProcessesPendingRequests();
@@ -1103,15 +1082,13 @@ TEST_F(ResourcePrefetchPredictorTest, RedirectUrlInDB) {
// Existing redirect to https://facebook.com/login will be deleted because of
// too many consecutive misses.
EXPECT_CALL(*mock_tables_.get(),
- UpdateData(empty_resource_data_, empty_resource_data_,
- url_redirect_data, empty_redirect_data_));
+ UpdateRedirectData(url_redirect_data, PREFETCH_KEY_TYPE_URL));
RedirectData host_redirect_data = CreateRedirectData("fb.com");
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));
+ UpdateRedirectData(host_redirect_data, PREFETCH_KEY_TYPE_HOST));
predictor_->RecordMainFrameLoadComplete(fb_end);
profile_->BlockUntilHistoryProcessesPendingRequests();
@@ -1906,7 +1883,8 @@ TEST_F(ResourcePrefetchPredictorTest, GetPrefetchData) {
TEST_F(ResourcePrefetchPredictorTest, TestPrecisionRecallHistograms) {
using testing::_;
- EXPECT_CALL(*mock_tables_.get(), UpdateData(_, _, _, _)).Times(2);
+ EXPECT_CALL(*mock_tables_.get(), UpdateResourceData(_, _));
+ EXPECT_CALL(*mock_tables_.get(), UpdateRedirectData(_, _));
EXPECT_CALL(*mock_tables_.get(), UpdateOriginData(_));
// Fill the database with 3 resources: 1 useful, 2 useless.
« no previous file with comments | « chrome/browser/predictors/resource_prefetch_predictor_tables_unittest.cc ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698