Chromium Code Reviews| Index: chrome/browser/prerender/prerender_unittest.cc |
| diff --git a/chrome/browser/prerender/prerender_unittest.cc b/chrome/browser/prerender/prerender_unittest.cc |
| index b530a2874eee902997b53ea0819d2c99c6f65258..13617a4d8800961dda3b9e3b22a52fb752a45599 100644 |
| --- a/chrome/browser/prerender/prerender_unittest.cc |
| +++ b/chrome/browser/prerender/prerender_unittest.cc |
| @@ -59,7 +59,7 @@ class DummyPrerenderContents : public PrerenderContents { |
| } |
| virtual bool GetRouteId(int* route_id) const OVERRIDE { |
| - *route_id = 0; |
| + *route_id = route_id_; |
| return true; |
| } |
| @@ -70,10 +70,15 @@ class DummyPrerenderContents : public PrerenderContents { |
| } |
| private: |
| + static int g_next_route_id_; |
| + int route_id_; |
| + |
| UnitTestPrerenderManager* test_prerender_manager_; |
| FinalStatus expected_final_status_; |
| }; |
| +int DummyPrerenderContents::g_next_route_id_ = 0; |
| + |
| const gfx::Size kSize(640, 480); |
| } // namespace |
| @@ -129,7 +134,7 @@ class UnitTestPrerenderManager : public PrerenderManager { |
| PrerenderContents* prerender_contents = prerender_data->ReleaseContents(); |
| active_prerenders_.erase(to_erase); |
| - prerender_contents->set_final_status(FINAL_STATUS_USED); |
| + prerender_contents->SetFinalStatus(FINAL_STATUS_USED); |
| prerender_contents->StartPendingPrerenders(); |
| return prerender_contents; |
| } |
| @@ -244,9 +249,10 @@ DummyPrerenderContents::DummyPrerenderContents( |
| const GURL& url, |
| Origin origin, |
| FinalStatus expected_final_status) |
| - : PrerenderContents(test_prerender_manager, prerender_tracker, |
| + : PrerenderContents(test_prerender_manager, |
| NULL, url, Referrer(), origin, |
| PrerenderManager::kNoExperiment), |
| + route_id_(g_next_route_id_++), |
| test_prerender_manager_(test_prerender_manager), |
| expected_final_status_(expected_final_status) { |
| } |
| @@ -263,6 +269,7 @@ void DummyPrerenderContents::StartPrerendering( |
| if (!is_control_group) |
| prerendering_has_started_ = true; |
| load_start_time_ = test_prerender_manager_->GetCurrentTimeTicks(); |
| + NotifyPrerenderStart(); |
|
mmenke
2012/12/06 19:54:22
We should not be running this in the control group
gavinp
2012/12/07 17:37:04
Done.
|
| } |
|
mmenke
2012/12/06 19:54:22
These two methods should be up in the anonymous na
gavinp
2012/12/07 17:37:04
Tricky. The functions reference the UnitTestPreren
mmenke
2012/12/07 19:21:16
Couldn't we just have PrerenderManager defriend Un
gavinp
2012/12/08 17:27:48
I think you're right, but it's out of scope for th
|
| class PrerenderTest : public testing::Test { |