OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include <iostream> | 5 #include <iostream> |
6 #include "base/memory/ref_counted.h" | 6 #include "base/memory/ref_counted.h" |
7 #include "base/memory/scoped_ptr.h" | 7 #include "base/memory/scoped_ptr.h" |
8 #include "base/message_loop.h" | 8 #include "base/message_loop.h" |
9 #include "base/run_loop.h" | 9 #include "base/run_loop.h" |
10 #include "base/time.h" | 10 #include "base/time.h" |
(...skipping 137 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
148 | 148 |
149 // TODO(shishir): Enable the prefetching mode in the tests. | 149 // TODO(shishir): Enable the prefetching mode in the tests. |
150 config.mode |= ResourcePrefetchPredictorConfig::URL_LEARNING; | 150 config.mode |= ResourcePrefetchPredictorConfig::URL_LEARNING; |
151 config.mode |= ResourcePrefetchPredictorConfig::HOST_LEARNING; | 151 config.mode |= ResourcePrefetchPredictorConfig::HOST_LEARNING; |
152 predictor_.reset(new ResourcePrefetchPredictor(config, profile_.get())); | 152 predictor_.reset(new ResourcePrefetchPredictor(config, profile_.get())); |
153 predictor_->set_mock_tables(mock_tables_); | 153 predictor_->set_mock_tables(mock_tables_); |
154 } | 154 } |
155 | 155 |
156 void InitializeSampleData(); | 156 void InitializeSampleData(); |
157 | 157 |
158 MessageLoop loop_; | 158 base::MessageLoop loop_; |
159 content::TestBrowserThread ui_thread_; | 159 content::TestBrowserThread ui_thread_; |
160 content::TestBrowserThread db_thread_; | 160 content::TestBrowserThread db_thread_; |
161 scoped_ptr<TestingProfile> profile_; | 161 scoped_ptr<TestingProfile> profile_; |
162 | 162 |
163 scoped_ptr<ResourcePrefetchPredictor> predictor_; | 163 scoped_ptr<ResourcePrefetchPredictor> predictor_; |
164 scoped_refptr<StrictMock<MockResourcePrefetchPredictorTables> > mock_tables_; | 164 scoped_refptr<StrictMock<MockResourcePrefetchPredictorTables> > mock_tables_; |
165 | 165 |
166 PrefetchDataMap test_url_data_; | 166 PrefetchDataMap test_url_data_; |
167 PrefetchDataMap test_host_data_; | 167 PrefetchDataMap test_host_data_; |
168 PrefetchData empty_url_data_; | 168 PrefetchData empty_url_data_; |
169 PrefetchData empty_host_data_; | 169 PrefetchData empty_host_data_; |
170 }; | 170 }; |
171 | 171 |
172 ResourcePrefetchPredictorTest::ResourcePrefetchPredictorTest() | 172 ResourcePrefetchPredictorTest::ResourcePrefetchPredictorTest() |
173 : loop_(MessageLoop::TYPE_DEFAULT), | 173 : loop_(base::MessageLoop::TYPE_DEFAULT), |
174 ui_thread_(content::BrowserThread::UI, &loop_), | 174 ui_thread_(content::BrowserThread::UI, &loop_), |
175 db_thread_(content::BrowserThread::DB, &loop_), | 175 db_thread_(content::BrowserThread::DB, &loop_), |
176 profile_(new TestingProfile()), | 176 profile_(new TestingProfile()), |
177 predictor_(NULL), | 177 predictor_(NULL), |
178 mock_tables_(new StrictMock<MockResourcePrefetchPredictorTables>()), | 178 mock_tables_(new StrictMock<MockResourcePrefetchPredictorTables>()), |
179 empty_url_data_(PREFETCH_KEY_TYPE_URL, std::string()), | 179 empty_url_data_(PREFETCH_KEY_TYPE_URL, std::string()), |
180 empty_host_data_(PREFETCH_KEY_TYPE_HOST, std::string()) {} | 180 empty_host_data_(PREFETCH_KEY_TYPE_HOST, std::string()) {} |
181 | 181 |
182 ResourcePrefetchPredictorTest::~ResourcePrefetchPredictorTest() { | 182 ResourcePrefetchPredictorTest::~ResourcePrefetchPredictorTest() { |
183 profile_.reset(NULL); | 183 profile_.reset(NULL); |
(...skipping 734 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
918 predictor_->inflight_navigations_[main_frame1.navigation_id]->at(0))); | 918 predictor_->inflight_navigations_[main_frame1.navigation_id]->at(0))); |
919 EXPECT_TRUE(URLRequestSummaryAreEqual( | 919 EXPECT_TRUE(URLRequestSummaryAreEqual( |
920 resource2, | 920 resource2, |
921 predictor_->inflight_navigations_[main_frame1.navigation_id]->at(1))); | 921 predictor_->inflight_navigations_[main_frame1.navigation_id]->at(1))); |
922 EXPECT_TRUE(URLRequestSummaryAreEqual( | 922 EXPECT_TRUE(URLRequestSummaryAreEqual( |
923 resource3, | 923 resource3, |
924 predictor_->inflight_navigations_[main_frame1.navigation_id]->at(2))); | 924 predictor_->inflight_navigations_[main_frame1.navigation_id]->at(2))); |
925 } | 925 } |
926 | 926 |
927 } // namespace predictors | 927 } // namespace predictors |
OLD | NEW |